Fix cm unit test failure

alan-turing-institute · May 16, 2024 · 3f27b5d · 3f27b5d
1 parent 2d0375e
commit 3f27b5d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/usage_function/costmanagement/__init__.py b/usage_function/costmanagement/__init__.py
@@ -8,13 +8,13 @@
 from azure.identity import DefaultAzureCredential
 from azure.mgmt.costmanagement import CostManagementClient
 from azure.mgmt.costmanagement.models import (
-    QueryDefinition,
-    QueryGrouping,
-    QueryDataset,
-    TimeframeType,
     ExportType,
     QueryAggregation,
+    QueryDataset,
+    QueryDefinition,
+    QueryGrouping,
     QueryTimePeriod,
+    TimeframeType,
 )
 
 import utils.settings

diff --git a/usage_function/tests/test_function_app.py b/usage_function/tests/test_function_app.py
@@ -255,19 +255,23 @@ def test_get_all_usage(self) -> None:
             )
             parameters2 = QueryDefinition(
                 time_period=QueryTimePeriod(
-                    from_property=start_datetime + timedelta(364),
+                    from_property=start_datetime + timedelta(365),
                     to=end_datetime,
                 ),
                 dataset=query_dataset,
                 type=query_type,
                 timeframe=query_timeframe,
             )
             scope = "/providers/Microsoft.Management/managementGroups/ea"
-            mock_list_func.assert_has_calls(
-                [
-                    call(scope=scope, parameters=parameters1),
-                    call(scope=scope, parameters=parameters2),
-                ]
+            self.assertEqual(mock_list_func.call_args_list[0].kwargs["scope"], scope)
+            self.assertEqual(
+                mock_list_func.call_args_list[0].kwargs["parameters"].serialize(),
+                parameters1.serialize(),
+            )
+            self.assertEqual(mock_list_func.call_args_list[1].kwargs["scope"], scope)
+            self.assertEqual(
+                mock_list_func.call_args_list[1].kwargs["parameters"].serialize(),
+                parameters2.serialize(),
             )
             self.assertEqual(mock_list_func.call_count, 2)