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, [])
