def test_datafile_download():
urls = ["https://google.com", "https://bing.com", "https://amazon.com"]
attempt = 0
for url in urls:
DataFile.download_to_file(url, "datafile_download_temp")
r = requests.get(urls[attempt])
if r.status_code == 200:
break
assert os.path.exists("datafile_download_temp")
os.remove("datafile_download_temp")
def test_datafile_download_bad_url(display):
display.change_status = mock.MagicMock(name="change_status")
mydownloadqueue = DownloadQueue()
url = "https://bad"
DataFile.download_to_file(url,
"datafile_download_temp",
"datafile download name",
mydownloadqueue,
display=display)
while mydownloadqueue.length > 0:
pass
assert display.change_status.call_count > 0
assert not os.path.exists("datafile_download_temp")
