Ejemplo n.º 1
0
def test_youtube_api_error_exit():
    mock_filename = os.path.join(SCRIPT_DIR, 'files/embed_restricted')
    with open(mock_filename, 'rb') as mock_f:
        fake_api_output = mock_f.read()

    httpretty.register_uri(
        httpretty.GET, yturl.GVI_BASE_URL + VIDEO_ID,
        body=fake_api_output, content_type='application/x-www-form-urlencoded',
    )

    with assert_raises(yturl.YouTubeAPIError):
        yturl.run([FAKE_URL], force_return=True)
Ejemplo n.º 2
0
def test_quality_as_word_ok():
    with open(os.path.join(SCRIPT_DIR, 'files/success_output')) as output_f:
        expected = dict(json.load(output_f))[43]

    with open(os.path.join(SCRIPT_DIR, 'files/success_input'), 'rb') as mock_f:
        fake_api_output = mock_f.read()

    httpretty.register_uri(
        httpretty.GET, yturl.GVI_BASE_URL + VIDEO_ID,
        body=fake_api_output, content_type='application/x-www-form-urlencoded',
    )

    chosen_uri = yturl.run(['-q', 'high', FAKE_URL], force_return=True)
    eq(chosen_uri, expected)
Ejemplo n.º 3
0
def test_unknown_quality():
    with assert_raises(yturl.UnknownQualityError):
        yturl.run(['-q', '123456', FAKE_URL], force_return=True)