python-markdown-oembed/tests.py

# -*- coding: utf-8 -*-
import re
import unittest
import markdown
from mock import patch
from mdx_oembed.extension import OEMBED_LINK_RE


class OEmbedPatternRegexTestCase(unittest.TestCase):
    def setUp(self):
        self.re = re.compile(OEMBED_LINK_RE)

    def test_ignore_relative_image_link(self):
        text = '![image](/image.png)'
        match = self.re.match(text)
        self.assertIsNone(match)

    def test_ignore_absolute_image_link(self):
        text = '![Mumbo Jumbo](http://tannern.com/mumbo-jumbo.jpg)'
        match = self.re.match(text)
        self.assertIsNone(match)

    def test_ignore_png_image_link(self):
        text = '![Mumbo Jumbo](http://tannern.com/mumbo-jumbo.png)'
        match = self.re.match(text)
        self.assertIsNone(match)

    def test_ignore_jpg_image_link(self):
        text = '![Mumbo Jumbo](http://tannern.com/mumbo-jumbo.jpg)'
        match = self.re.match(text)
        self.assertIsNone(match)

    def test_ignore_gif_image_link(self):
        text = '![Mumbo Jumbo](http://tannern.com/mumbo-jumbo.gif)'
        match = self.re.match(text)
        self.assertIsNone(match)

    def test_find_youtube_link(self):
        text = '![video](http://www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        match = self.re.match(text)
        self.assertIsNotNone(match)

    def test_find_youtube_short_link(self):
        text = '![video](http://youtu.be/7XzdZ4KcI8Y)'
        match = self.re.match(text)
        self.assertIsNotNone(match)

    def test_find_youtube_http(self):
        text = '![video](http://youtu.be/7XzdZ4KcI8Y)'
        match = self.re.match(text)
        self.assertIsNotNone(match)

    def test_find_youtube_https(self):
        text = '![video](https://youtu.be/7XzdZ4KcI8Y)'
        match = self.re.match(text)
        self.assertIsNotNone(match)

    def test_find_youtube_auto(self):
        text = '![video](//youtu.be/7XzdZ4KcI8Y)'
        match = self.re.match(text)
        self.assertIsNotNone(match)


class OEmbedExtensionTestCase(unittest.TestCase):
    def setUp(self):
        self.markdown = markdown.Markdown(extensions=['oembed'])

    def assert_convert(self, text, expected):
        with patch('oembed.OEmbedEndpoint') as MockOEmbedEndpoint:
            MockOEmbedEndpoint.get.return_value = expected
            output = self.markdown.convert(text)
        self.assertEqual(output, expected)


class IgnoredTestCase(OEmbedExtensionTestCase):
    """
    The OEmbedExtension should ignore these tags allowing markdown's image
    processor to find and handle them.
    """

    def test_relative(self):
        text = '![alt](image.png)'
        expected = '<p><img alt="alt" src="image.png" /></p>'
        output = self.markdown.convert(text)
        self.assertEqual(output, expected)

    def test_slash_relative(self):
        text = '![alt](/image.png)'
        expected = '<p><img alt="alt" src="/image.png" /></p>'
        output = self.markdown.convert(text)
        self.assertEqual(output, expected)

    def test_absolute(self):
        text = '![Mumbo Jumbo](http://tannern.com/mumbo-jumbo.jpg)'
        expected = '<p><img alt="Mumbo Jumbo" src="http://tannern.com/mumbo-jumbo.jpg" /></p>'
        output = self.markdown.convert(text)
        self.assertEqual(output, expected)


class ProtocolVarietyTestCase(OEmbedExtensionTestCase):

    def test_http(self):
        text = '![video](http://www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)

    def test_https(self):
        text = '![video](https://www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)

    def test_auto(self):
        text = '![video](//www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)


class YoutubeTestCase(OEmbedExtensionTestCase):
    """
    The OEmbedExtension should handle embedding for these cases.
    """

    def test_youtube_link(self):
        """
        YouTube video link.
        """
        text = '![video](http://www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)

    def test_youtube_short_link(self):
        """
        Short format YouTube video link.
        """
        text = '![video](http://youtu.be/7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)


class VimeoTestCase(OEmbedExtensionTestCase):

    def test_vimeo_link(self):
        """
        Vimeo video link.
        """
        text = '![link](http://vimeo.com/52970271)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)


class LimitedOEmbedExtensionTestCase(OEmbedExtensionTestCase):
    def setUp(self):
        self.markdown = markdown.Markdown(
            extensions=['oembed'],
            extension_configs={
                'oembed': {
                    'allowed_endpoints': ['youtube',],
                }
            })

    def test_youtube_link(self):
        """
        YouTube video link.
        """
        text = '![video](http://www.youtube.com/watch?v=7XzdZ4KcI8Y)'
        output = self.markdown.convert(text)
        self.assertIn('<iframe', output)

    def test_vimeo_link(self):
        """
        Vimeo video link.
        """
        text = '![link](http://vimeo.com/52970271)'
        output = self.markdown.convert(text)
        self.assertNotIn('<iframe', output)


if __name__ == "__main__":
    unittest.main()