def test_get_thumbnail_url(url, mock_bigthumbhd, mock_bigthumb, mock_thumb, expected): from youtube_thumbnail.api import get_thumbnail_url mock_pafy = mock.MagicMock( bigthumbhd=mock_bigthumbhd, bigthumb=mock_bigthumb, thumb=mock_thumb, ) with mock.patch('youtube_thumbnail.api.get_pafy', return_value=mock_pafy): assert get_thumbnail_url(url) == expected
def test_get_thumbnail_url_cache(): from youtube_thumbnail.api import get_thumbnail_url mock_pafy = mock.MagicMock( bigthumbhd='http://test.com/test.png', bigthumb='', thumb='', ) url = '//www.youtube.com/watch?v=dQw4w9WgXcQ' cache.set('youtube_thumbnail:{}'.format(url), 'http://test-cache.com/test2.png') with mock.patch('youtube_thumbnail.api.get_pafy', return_value=mock_pafy): assert get_thumbnail_url(url) == '//test-cache.com/test2.png' cache.clear()
def test_get_thumbnail_url_error(): from youtube_thumbnail.api import get_thumbnail_url with mock.patch('pafy.new', side_effect=Exception("test exception")): assert get_thumbnail_url("//www.youtube.com/watch?v=dQw4w9WgXcQ") == ''
def test_get_thumbnail_url_unknown(): from youtube_thumbnail.api import get_thumbnail_url with mock.patch('youtube_thumbnail.api.get_pafy', return_value=None): assert get_thumbnail_url("//www.youtube.com/watch?v=dQw4w9WgXcQ") == ''