def test014_add_files_error(self, m_validate_id, m_process_error):
db_mock = MagicMock()
exception = Exception("test")
m_validate_id.side_effect = exception
scanid = "whatever"
api_scans.add_files(scanid, db_mock)
m_process_error.assert_called_once_with(exception)
def test015_add_files_no_files(self, m_validate_id, m_scan_schema,
m_Scan, m_scan_ctrl, m_request):
db_mock = MagicMock()
expected = "No files uploaded"
scanid = "whatever"
m_request.files = {}
with self.assertRaises(HTTPError) as context:
api_scans.add_files(scanid, db_mock)
m_validate_id.assert_called_once_with(scanid)
m_Scan.load_from_ext_id.assert_called_once_with(scanid, db_mock)
self.assertEqual(context.exception.body.message, expected)
self.assertEqual(api_scans.response.content_type,
"application/json; charset=UTF-8")
m_scan_ctrl.add_files.assert_not_called()
def test013_add_files_ok(self, m_validate_id, m_scan_schema,
m_Scan, m_scan_ctrl, m_request):
db_mock = MagicMock()
m_file = MagicMock()
expected = m_scan_schema.dumps().data
scanid = "whatever"
m_file.raw_filename = "filename"
m_request.files = {'file': m_file}
result = api_scans.add_files(scanid, db_mock)
m_validate_id.assert_called_once_with(scanid)
m_Scan.load_from_ext_id.assert_called_once_with(scanid, db_mock)
self.assertEqual(result, expected)
self.assertEqual(api_scans.response.content_type,
"application/json; charset=UTF-8")
m_scan_ctrl.add_files.assert_called_once()