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