예제 #1
0
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
예제 #2
0
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
예제 #3
0
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()
예제 #4
0
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()
예제 #5
0
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") == ''
예제 #6
0
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") == ''
예제 #7
0
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") == ''
예제 #8
0
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") == ''