def test_process_report(self): """Test report processing.""" report_meta = { "schema_name": "test_schema", "manifest_id": "1", "provider_uuid": uuid.uuid4(), "provider_type": "OCP", "compression": "UNCOMPRESSED", "file": "/path/to/file.csv", } with patch("masu.external.kafka_msg_handler._process_report_file" ) as mock_process: msg_handler.process_report("request_id", report_meta) mock_process.assert_called()
def test_process_report(self, mock_get_reports, mock_summarize): """Test processing a report for an unknown cluster_id.""" mock_download_process_value = [{ 'schema_name': self.test_schema, 'provider_type': OPENSHIFT_CONTAINER_PLATFORM, 'provider_uuid': self.ocp_test_provider_uuid, 'start_date': DateAccessor().today(), }] mock_get_reports.return_value = mock_download_process_value cluster_id = self.ocp_provider_resource_name sample_report = {'cluster_id': cluster_id} msg_handler.process_report(sample_report) mock_summarize.delay.assert_called_with(mock_download_process_value)
def test_process_report_unknown_cluster_id(self, mock_get_reports, mock_summarize): """Test processing a report for an unknown cluster_id.""" mock_download_process_value = [{ "schema_name": self.schema, "provider_type": Provider.PROVIDER_OCP, "provider_uuid": self.ocp_test_provider_uuid, "start_date": DateAccessor().today(), }] mock_get_reports.return_value = mock_download_process_value sample_report = {"cluster_id": "missing_cluster_id"} msg_handler.process_report(sample_report) mock_summarize.delay.assert_not_called()
def test_process_report_not_implemented_error(self, _): """Test report processing.""" report_meta = { "schema_name": "test_schema", "manifest_id": "1", "provider_uuid": uuid.uuid4(), "provider_type": "OCP", "compression": "UNCOMPRESSED", "file": "/path/to/file.csv", } self.assertTrue(msg_handler.process_report("request_id", report_meta))
def test_process_report_inactive_provider(self, mock_get_reports, mock_summarize): """Test processing a report for an inactive provider.""" self.ocp_provider.active = False self.ocp_provider.save() mock_download_process_value = [{ "schema_name": self.schema, "provider_type": Provider.PROVIDER_OCP, "provider_uuid": self.ocp_test_provider_uuid, "start_date": DateAccessor().today(), }] mock_get_reports.return_value = mock_download_process_value cluster_id = self.ocp_provider_resource_name sample_report = { "cluster_id": cluster_id, "date": DateAccessor().today() } msg_handler.process_report(sample_report) mock_summarize.delay.assert_not_called()
def test_process_report_inactive_provider(self, mock_get_reports, mock_summarize): """Test processing a report for an inactive provider.""" self.ocp_provider.active = False self.ocp_provider.save() mock_download_process_value = [{ 'schema_name': self.schema, 'provider_type': OPENSHIFT_CONTAINER_PLATFORM, 'provider_uuid': self.ocp_test_provider_uuid, 'start_date': DateAccessor().today(), }] mock_get_reports.return_value = mock_download_process_value cluster_id = self.ocp_provider_resource_name sample_report = { 'cluster_id': cluster_id, 'date': DateAccessor().today() } msg_handler.process_report(sample_report) mock_summarize.delay.assert_not_called()