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)
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)
def test_unknown_quality():
with assert_raises(yturl.UnknownQualityError):
yturl.run(['-q', '123456', FAKE_URL], force_return=True)