feat: fixing units tests.

project-koku · Jul 1, 2024 · 827a2b2 · 827a2b2
1 parent 4a4d050
commit 827a2b2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_azure_generator.py b/tests/test_azure_generator.py
@@ -432,8 +432,10 @@ def test_update_data_with_attributes(self):
             if generator.azure_columns == AZURE_COLUMNS_V2_SUBSCRIPTION:
                 self.assertEqual(row["SubscriptionId"], self.payer_account)
                 self.assertEqual(row["ResourceId"], self.instance_id)
+                row["BillingCurrency"] = self.currency
             else:
                 self.assertEqual(row["SubscriptionGuid"], self.payer_account)
+                row["BillingCurrencyCode"] = self.currency
 
 
 class TestDTGenerator(AzureGeneratorTestCase):