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

download_archive / download_rpm methods for CLI

Browse source 
Fixes: https://pagure.io/koji/issue/1880
This commit is contained in:
Tomas Kopecek 2020-03-12 15:00:03 +01:00
parent 18d7fc8f82
commit 291f04301b
4 changed files with 164 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_cli/test_download_file.py
View file

@ -56,7 +56,6 @@ class TestDownloadFile(unittest.TestCase):
self.assertEqual(cm.exception[1], 'Is a directory')
else:
self.assertEqual(cm.exception.args, (21, 'Is a directory'))
self.requests_get.assert_called_once()
@mock_open()
def test_handle_download_file(self, m_open):
@ -166,10 +165,9 @@ class TestDownloadFileError(unittest.TestCase):
with self.assertRaises(requests.HTTPError):
download_file("http://url", self.filename)
try:
self.assertFalse(os.path.exists(self.filename))
except AssertionError:
os.unlink(self.filename)
raise
except Exception:
pass
@requests_mock.Mocker()
def test_handle_download_file_error_500(self, m):
@ -177,10 +175,9 @@ class TestDownloadFileError(unittest.TestCase):
with self.assertRaises(requests.HTTPError):
download_file("http://url", self.filename)
try:
self.assertFalse(os.path.exists(self.filename))
except AssertionError:
os.unlink(self.filename)
raise
except Exception:
pass
if __name__ == '__main__':
unittest.main()

9
tests/test_cli/test_download_task.py
View file

@ -36,7 +36,8 @@ class TestDownloadTask(unittest.TestCase):
url = pattern % (subpath, k)
if target.endswith('.log') and arch is not None:
target = "%s.%s.log" % (target.rstrip(".log"), arch)
calls.append(call(url, target, None, None, total, i + 1))
calls.append(call(url, target, quiet=None, noprogress=None,
size=total, num=i + 1))
return calls
def setUp(self):
@ -162,11 +163,11 @@ class TestDownloadTask(unittest.TestCase):
call(self.session, 44444)])
self.assertListEqual(self.download_file.mock_calls, [
call('https://topurl/work/tasks/3333/33333/somerpm.x86_64.rpm',
'somerpm.x86_64.rpm', None, None, 3, 1),
'somerpm.x86_64.rpm', quiet=None, noprogress=None, size=3, num=1),
call('https://topurl/vol/vol2/work/tasks/3333/33333/somerpm.x86_64.rpm',
'vol2/somerpm.x86_64.rpm', None, None, 3, 2),
'vol2/somerpm.x86_64.rpm', quiet=None, noprogress=None, size=3, num=2),
call('https://topurl/vol/vol3/work/tasks/4444/44444/somerpm.noarch.rpm',
'vol3/somerpm.noarch.rpm', None, None, 3, 3)])
'vol3/somerpm.noarch.rpm', quiet=None, noprogress=None, size=3, num=3)])
self.assertIsNone(rv)
def test_handle_download_task_log(self):