def test_add_files_no_files(self, m_Scan, m_scan_ctrl, m_IrmaScanStatus):
scan_id = "whatever"
m_request = MagicMock()
m_request.files = {}
expected = "The \"files\" parameter is invalid. Empty list"
with self.assertRaises(HTTPInvalidParam) as context:
api_scans.add_files(m_request, scan_id=scan_id)
m_Scan.load_from_ext_id.assert_called_once_with(scan_id, self.session)
self.assertEqual(context.exception.description, expected)
m_scan_ctrl.add_files.assert_not_called()
def test_add_files_ok(self, m_Scan, m_IrmaScanStatus, m_FileExt, m_File):
m_file = MagicMock()
m_request = MagicMock()
scan_id = "whatever"
data = b"DATA"
filename = "filename"
m_file.filename = filename
m_file.file = io.BytesIO(data)
m_request._params = {'files': m_file}
result = api_scans.add_files(m_request, scan_id)
m_Scan.load_from_ext_id.assert_called_once_with(scan_id, self.session)
self.assertIsScan(result)