def test_two_userlinks(self): foo_userlink = Mock(title='foo_title', url='foo_userlink') media = _get_media([foo_userlink] * 2) self.assertItemsEqual(media, [{ 'title': 'foo_title', 'url': 'foo_hotlink', 'thumbnail_url': 'foo_thumbnail', }] * 2)
def test_hotlink_without_thumbnail(self, addon): spam_userlink = Mock(title='spam_title', url='spam_userlink') media = _get_media([spam_userlink]) self.assertItemsEqual(media, [{ 'title': 'spam_title', 'url': 'spam_hotlink', 'thumbnail_url': addon.icon, }])
def test_two_hotlinks(self): bar_userlink = Mock(title='bar_title', url='bar_userlink') media = _get_media([bar_userlink]) self.assertItemsEqual(media, [{ 'title': 'bar_title', 'url': 'bar_hotlink1', 'thumbnail_url': 'bar_thumbnail1', }, { 'title': 'bar_title', 'url': 'bar_hotlink2', 'thumbnail_url': 'bar_thumbnail2', }])
def test_unmanagable_userlink(self): media = _get_media([Mock()]) self.assertItemsEqual(media, [])
def test_no_userlinks(self): media = _get_media([]) self.assertItemsEqual(media, [])