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()