From 14a3664c917bd82891e3a144f2da552d6c84108d Mon Sep 17 00:00:00 2001
From: myersCody <cmyers@redhat.com>
Date: Thu, 15 Aug 2024 10:56:01 -0400
Subject: [PATCH] Fix tests

---
 tests/test_aws_generator.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_aws_generator.py b/tests/test_aws_generator.py
index 5c8dbf16..fbc5f105 100644
--- a/tests/test_aws_generator.py
+++ b/tests/test_aws_generator.py
@@ -209,6 +209,7 @@ def setUp(self):
         self.amount = 1
         self.rate = 0.1
         self.saving = 0.1
+        self.disk_size = 10
         self.currency = "USD"
         self.legal_entity = "Red Hat"
         self.attributes = {
@@ -219,6 +220,7 @@ def setUp(self):
             "product_name": self.product_name,
             "resource_id": self.resource_id,
             "amount": self.amount,
+            "disk_size": self.disk_size,
             "rate": self.rate,
             "saving": self.saving,
             "product_family": self.product_family,
@@ -354,7 +356,7 @@ def test_init_with_attributes(self):
         self.assertEqual(generator._product_sku, self.product_sku)
         self.assertEqual(generator._tags, self.tags)
         self.assertEqual(generator._resource_id, "vol-" + self.resource_id)
-        self.assertEqual(generator._amount, self.amount)
+        self.assertEqual(generator._disk_size, self.disk_size)
         self.assertEqual(generator._rate, self.rate)
 
     def test_update_data(self):