[COST-5627] Fix Azure default extension issue

koku/masu/external/downloader/azure/azure_report_downloader.py

@@ -332,9 +332,7 @@ def _get_manifest(self, date_time):  # noqa: C901
                 manifest["reportKeys"] = [blob["blobName"] for blob in manifest_json["blobs"]]
             else:
                 try:
-                    blob = self._azure_client.get_latest_cost_export_for_path(
-                        report_path, self.container_name, compression_mode
-                    )
+                    blob = self._azure_client.get_latest_cost_export_for_path(report_path, self.container_name)
                 except AzureCostReportNotFound as ex:
                     msg = f"Unable to find manifest. Error: {ex}"
                     LOG.info(log_json(self.tracing_id, msg=msg, context=self.context))

koku/masu/external/downloader/azure/azure_service.py

@@ -64,24 +64,31 @@ def __init__(
             raise AzureServiceError("Azure Service credentials are not configured.")
 
     def _get_latest_blob(
-        self, report_path: str, blobs: list[BlobProperties], extension: str
+        self, report_path: str, blobs: list[BlobProperties], extension: t.Optional[str] = None
     ) -> t.Optional[BlobProperties]:
+        valid_extensions = [
+            AzureBlobExtension.csv.value,
+            AzureBlobExtension.gzip.value,
+            AzureBlobExtension.manifest.value,
+        ]
         latest_blob = None
         for blob in blobs:
-            if not blob.name.endswith(extension):
+            if extension:
+                if not blob.name.endswith(extension):
+                    continue
+            elif not any(blob.name.endswith(ext) for ext in valid_extensions):
                 continue
 
-            if report_path in blob.name and not latest_blob:
-                latest_blob = blob
-            elif report_path in blob.name and blob.last_modified > latest_blob.last_modified:
-                latest_blob = blob
+            if report_path in blob.name:
+                if not latest_blob or blob.last_modified > latest_blob.last_modified:
+                    latest_blob = blob
         return latest_blob
 
     def _get_latest_blob_for_path(
         self,
         report_path: str,
         container_name: str,
-        extension: str,
+        extension: t.Optional[str] = None,
     ) -> BlobProperties:
         """Get the latest file with the specified extension from given storage account container."""
 
@@ -120,10 +127,7 @@ def _get_latest_blob_for_path(
 
         latest_report = self._get_latest_blob(report_path, blobs, extension)
         if not latest_report:
-            message = (
-                f"No file with extension '{extension}' found in container "
-                f"'{container_name}' for path '{report_path}'."
-            )
+            message = f"No file found in container " f"'{container_name}' for path '{report_path}'."
             raise AzureCostReportNotFound(message)
 
         return latest_report
@@ -158,9 +162,7 @@ def get_file_for_key(self, key: str, container_name: str) -> BlobProperties:
 
         return report
 
-    def get_latest_cost_export_for_path(
-        self, report_path: str, container_name: str, compression: str
-    ) -> BlobProperties:
+    def get_latest_cost_export_for_path(self, report_path: str, container_name: str) -> BlobProperties:
         """
         Get the latest cost export for a given path and container based on the compression type.
 
@@ -176,11 +178,7 @@ def get_latest_cost_export_for_path(
             ValueError: If the compression type is not 'gzip' or 'csv'.
             AzureCostReportNotFound: If no blob is found for the given path and container.
         """
-        valid_compressions = [AzureBlobExtension.gzip.value, AzureBlobExtension.csv.value]
-        if compression not in valid_compressions:
-            raise ValueError(f"Invalid compression type: {compression}. Expected one of: {valid_compressions}.")
-
-        return self._get_latest_blob_for_path(report_path, container_name, compression)
+        return self._get_latest_blob_for_path(report_path, container_name)
 
     def get_latest_manifest_for_path(self, report_path: str, container_name: str) -> BlobProperties:
         return self._get_latest_blob_for_path(report_path, container_name, AzureBlobExtension.manifest.value)

koku/masu/test/external/downloader/azure/test_azure_services.py

@@ -210,15 +210,15 @@ def test_get_latest_cost_export_for_path(self):
         type(mock_blob).name = name_attr  # kludge to set name attribute on Mock
 
         svc = self.get_mock_client(blob_list=[mock_blob])
-        cost_export = svc.get_latest_cost_export_for_path(report_path, self.container_name, ".csv.gz")
+        cost_export = svc.get_latest_cost_export_for_path(report_path, self.container_name)
         self.assertEqual(cost_export.last_modified.date(), self.current_date_time.date())
 
     def test_get_latest_cost_export_for_path_missing(self):
         """Test that the no cost export is returned for a missing path."""
         report_path = FAKE.word()
         svc = self.get_mock_client()
         with self.assertRaises(AzureCostReportNotFound):
-            svc.get_latest_cost_export_for_path(report_path, self.container_name, ".csv.gz")
+            svc.get_latest_cost_export_for_path(report_path, self.container_name)
 
     def test_describe_cost_management_exports(self):
         """Test that cost management exports are returned for the account."""
@@ -259,7 +259,7 @@ def test_get_latest_cost_export_http_error(self):
         svc = self.get_mock_client(blob_list=[mock_blob])
         svc._cloud_storage_account.get_container_client.side_effect = throw_azure_http_error
         with self.assertRaises(AzureCostReportNotFound):
-            svc.get_latest_cost_export_for_path(report_path, self.container_name, ".csv.gz")
+            svc.get_latest_cost_export_for_path(report_path, self.container_name)
 
     def test_get_latest_cost_export_http_error_403(self):
         """Test that the latest cost export catches the error for Azure HttpError 403."""
@@ -272,7 +272,7 @@ def test_get_latest_cost_export_http_error_403(self):
         svc = self.get_mock_client(blob_list=[mock_blob])
         svc._cloud_storage_account.get_container_client.side_effect = throw_azure_http_error_403
         with self.assertRaises(AzureCostReportNotFound):
-            svc.get_latest_cost_export_for_path(report_path, self.container_name, ".csv.gz")
+            svc.get_latest_cost_export_for_path(report_path, self.container_name)
 
     def test_get_latest_cost_export_no_container(self):
         """Test that the latest cost export catches the error for no container."""
@@ -285,7 +285,7 @@ def test_get_latest_cost_export_no_container(self):
 
         svc = self.get_mock_client(blob_list=[mock_blob])
         with self.assertRaises(AzureCostReportNotFound):
-            svc.get_latest_cost_export_for_path(report_path, container_name, ".csv.gz")
+            svc.get_latest_cost_export_for_path(report_path, container_name)
 
     def test_get_latest_manifest_for_path(self):
         """Given a list of blobs with multiple manifests, ensure the latest one is returned"""
@@ -422,9 +422,7 @@ def test_get_latest_cost_export_for_path_exception(self, mock_factory):
             service = AzureService(
                 self.tenant_id, self.client_id, self.client_secret, self.resource_group_name, self.storage_account_name
             )
-            service.get_latest_cost_export_for_path(
-                report_path=FAKE.word(), container_name=FAKE.word(), compression=".csv.gz"
-            )
+            service.get_latest_cost_export_for_path(report_path=FAKE.word(), container_name=FAKE.word())
 
     def test_describe_cost_management_exports_with_scope_and_name(self):
         """Test that cost management exports using scope and name are returned for the account."""
@@ -487,7 +485,7 @@ def test_get_latest_cost_export_missing_container(self):
         svc = self.get_mock_client(blob_list=[mock_blob])
         svc._cloud_storage_account.get_container_client.side_effect = ResourceNotFoundError("Oops!")
         with self.assertRaises(AzureCostReportNotFound):
-            svc.get_latest_cost_export_for_path(report_path, self.container_name, ".csv.gz")
+            svc.get_latest_cost_export_for_path(report_path, self.container_name)
 
     @patch("masu.external.downloader.azure.azure_service.AzureClientFactory")
     def test_azure_service_missing_credentials(self, mock_factory):
@@ -507,27 +505,6 @@ def test_azure_service_missing_credentials(self, mock_factory):
 
         self.assertIn("Azure Service credentials are not configured.", str(context.exception))
 
-    @patch("masu.external.downloader.azure.azure_service.AzureClientFactory")
-    @patch.object(AzureService, "_get_latest_blob_for_path")
-    def test_get_latest_cost_export_for_path_invalid_compression(self, mock_get_latest_blob, mock_factory):
-        """Test when an invalid compression type is provided and ValueError is raised."""
-        mock_get_latest_blob.return_value = None
-        mock_factory.return_value.credentials = Mock()
-
-        service = AzureService(
-            tenant_id="fake_tenant_id",
-            client_id="fake_client_id",
-            client_secret="fake_client_secret",
-            resource_group_name="fake_resource_group",
-            storage_account_name="fake_storage_account",
-            subscription_id="fake_subscription_id",
-        )
-
-        with self.assertRaises(ValueError) as context:
-            service.get_latest_cost_export_for_path("fake_report_path", "fake_container_name", "invalid_compression")
-
-        self.assertIn("Invalid compression type", str(context.exception))
-
     @patch("masu.external.downloader.azure.azure_service.AzureService._list_blobs")
     @patch("masu.external.downloader.azure.azure_service.AzureClientFactory")
     @patch("masu.external.downloader.azure.azure_service.NamedTemporaryFile")
@@ -627,24 +604,6 @@ def test_download_file_raises_exception(self, mock_tempfile, mock_client_factory
 
         self.assertIn("Failed to download cost export", str(context.exception))
 
-    @patch("masu.external.downloader.azure.azure_service.AzureClientFactory")
-    def test_invalid_compression_type(self, mock_factory):
-        """Test that an invalid compression type raises an exception."""
-
-        service = AzureService(
-            tenant_id="fake_tenant_id",
-            client_id="fake_client_id",
-            client_secret="fake_client_secret",
-            resource_group_name="fake_resource_group",
-            storage_account_name="fake_storage_account",
-            subscription_id="fake_subscription_id",
-        )
-
-        with self.assertRaises(ValueError) as context:
-            service.get_latest_cost_export_for_path("fake_report_path", "fake_container_name", "invalid_compression")
-
-        self.assertIn("Invalid compression type", str(context.exception))
-
     @patch("masu.external.downloader.azure.azure_service.AzureService._list_blobs")
     @patch("masu.external.downloader.azure.azure_service.AzureClientFactory")
     @patch("masu.external.downloader.azure.azure_service.NamedTemporaryFile")
@@ -677,3 +636,18 @@ def test_download_file_with_compression(self, mock_tempfile, mock_client_factory
         self.assertTrue(result.endswith(".gz"))
         mock_cloud_storage_account.get_blob_client.assert_called_with("fake_container", "fake_key.csv.gz")
         mock_blob_client.download_blob.assert_called_once()
+
+    def test_get_latest_blob_no_extension_invalid_files(self):
+        """Test that blobs with invalid extensions are ignored when no extension is specified."""
+        report_path = "/container/report/path"
+        blobs = (
+            FakeBlob(f"{report_path}/file01.csv", datetime(2022, 12, 16)),
+            FakeBlob(f"{report_path}/file02.txt", datetime(2022, 12, 17)),  # Invalid extension
+            FakeBlob(f"{report_path}/file03.log", datetime(2022, 12, 18)),  # Invalid extension
+            FakeBlob(f"{report_path}/file04.csv.gz", datetime(2022, 12, 19)),  # Valid extension
+        )
+        azure_service = self.get_mock_client()
+        latest_blob = azure_service._get_latest_blob(f"{report_path}", blobs, None)
+
+        # The latest valid blob is file04.csv.gz
+        self.assertEqual(latest_blob.name, f"{report_path}/file04.csv.gz")