def test_execute_ng(self):
# use url which does not exist
url = "https://spam.com/get"
timeout = 5
retry_count = 1
d = Download(url, self.__dest_path, timeout, retry_count)
with pytest.raises(HTTPError) as execinfo:
d.execute()
assert "Http request failed." in str(execinfo.value)
def test_execute_ok(self):
# use Postman echo
url = "https://postman-echo.com/get?foo1=bar1&foo2=bar2"
timeout = 10
retry_cnt = 3
d = Download(url, self._dest_path, timeout, retry_cnt)
d.execute()
f = open("/tmp/test.result", "r")
result = f.read()
f.close()
os.remove(self._dest_path)
assert "postman-echo.com" in result
def execute(self, *args):
super().execute()
os.makedirs(self._dest_dir, exist_ok=True)
url = os.path.join(self._src_url, self._src_pattern)
dest_path = (os.path.join(self._dest_dir, self._dest_pattern)
if self._dest_pattern else os.path.join(
self._dest_dir, self._src_pattern))
d = Download(url, dest_path, self._timeout, self._retry_count)
d.execute()