particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lib: retry openRemoteFile and check size

Browse source
This commit is contained in:
Tomas Kopecek 2020-05-18 14:44:36 +02:00
parent 360ccd248a
commit 8ea4a7245b
2 changed files with 50 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/test_lib/test_utils.py
View file

@ -161,6 +161,38 @@ class MiscFunctionTestCase(unittest.TestCase):
for m in mocks:
m.assert_not_called()
for m in mocks:
m.reset_mock()
# downloaded size is larger than content-length
with requests_mock.Mocker() as m_requests:
text = 'random content'
m_requests.register_uri('GET', url, text=text,
headers = {'Content-Length': "3"})
m_TemporaryFile.return_value.tell.return_value = len(text)
# using neither
with self.assertRaises(koji.GenericError):
koji.openRemoteFile(path, topurl=topurl)
m_TemporaryFile.assert_called_once()
m_TemporaryFile.return_value.tell.assert_called()
m_open.assert_not_called()
for m in mocks:
m.reset_mock()
# downloaded size is shorter than content-length
with requests_mock.Mocker() as m_requests:
text = 'random content'
m_requests.register_uri('GET', url, text=text,
headers = {'Content-Length': "100"})
m_TemporaryFile.return_value.tell.return_value = len(text)
# using neither
with self.assertRaises(koji.GenericError):
koji.openRemoteFile(path, topurl=topurl)
m_TemporaryFile.assert_called_once()
m_TemporaryFile.return_value.tell.assert_called()
m_open.assert_not_called()
def test_joinpath_bad(self):
bad_joins = [
['/foo', '../bar'],