def test_download_404(file_server, mktemp, client):
path = mktemp()
with pytest.raises(Exception) as error:
yield from download(
client, file_server.url('hello.txt'), path, request_id='?',
)
assert str(error.value) == 'Download failed.'
def test_download(file_server, mktemp, client):
file_server.provide('hello.txt', 'hello, world!')
path = mktemp()
content_type = yield from download(
client, file_server.url('hello.txt'), path, request_id='?',
)
assert content_type == 'text/plain'
with open(path, 'r') as stream:
assert stream.read() == 'hello, world!'
def test_download_reject(file_server, mktemp, client):
def check_ext(url, response):
return not url.endswith('.zip')
file_server.provide('hello.txt', 'hello, world!')
path = mktemp()
with pytest.raises(Exception) as error:
yield from download(
client, file_server.url('hello.txt'), path, reject=check_ext,
request_id='?',
)
assert str(error.value) == 'Download rejected.'
