def test__process_response_bad_status(self):
download = _download.Download(EXAMPLE_URL)
_fix_up_virtual(download)
# Make sure **not finished** before.
assert not download.finished
response = mock.Mock(status_code=int(http_client.NOT_FOUND),
spec=['status_code'])
with pytest.raises(common.InvalidResponse) as exc_info:
download._process_response(response)
error = exc_info.value
assert error.response is response
assert len(error.args) == 5
assert error.args[1] == response.status_code
assert error.args[3] == http_client.OK
assert error.args[4] == http_client.PARTIAL_CONTENT
# Make sure **finished** even after a failure.
assert download.finished
def test__prepare_request_already_finished(self):
download = _download.Download(EXAMPLE_URL)
download._finished = True
with pytest.raises(ValueError):
download._prepare_request()
def test_consume(self):
download = _download.Download(EXAMPLE_URL)
with pytest.raises(NotImplementedError) as exc_info:
download.consume(None)
exc_info.match(u"virtual")