mirror of
https://github.com/Hopiu/django-embed-video.git
synced 2026-03-16 21:30:23 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import re
|
|
from unittest import TestCase
|
|
|
|
from embed_video.backends import VideoBackend, detect_backend
|
|
|
|
|
|
class CustomBackend(VideoBackend):
|
|
re_detect = re.compile(r'http://myvideo\.com/[0-9]+')
|
|
re_code = re.compile(r'http://myvideo\.com/(?P<code>[0-9]+)')
|
|
|
|
pattern_url = '{protocol}://play.myvideo.com/c/{code}/'
|
|
pattern_thumbnail_url = '{protocol}://thumb.myvideo.com/c/{code}/'
|
|
|
|
|
|
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/')
|