django-embed-video/embed_video/tests/tests_custom_backend.py

from unittest import TestCase

from embed_video.backends import detect_backend

from .custom_backend import CustomBackend


class CustomBackendTestCase(TestCase):
    def setUp(self):
        self.backend = detect_backend('http://myvideo.com/1530')

    def test_detect_backend(self):
        self.assertIsInstance(self.backend, CustomBackend)

    def test_code(self):
        self.assertEqual(self.backend.code, '1530')

    def test_url(self):
        self.assertEqual(self.backend.get_url(),
                         'http://play.myvideo.com/c/1530/')

    def test_url_https(self):
        self.backend.is_secure = True
        self.assertEqual(self.backend.get_url(),
                         'https://play.myvideo.com/c/1530/')

    def test_thumbnail(self):
        self.assertEqual(self.backend.get_thumbnail_url(),
                         'http://thumb.myvideo.com/c/1530/')