Adding missing unit test coverage

Nathan Beittenmiller · Nathan Beittenmiller · commit 76013446c765 · 2013-07-08T12:08:12.000-05:00
diff --git a/SoftLayer/tests/managers/hardware_tests.py b/SoftLayer/tests/managers/hardware_tests.py
@@ -151,9 +151,53 @@ def test_get_bare_metal_create_options(self):
     def test_generate_create_dict_with_all_bare_metal_options(self):
         package_id = 50
 
+        prices = [{
+            'id': 888,
+            'price_id': 1888,
+            'sort': 0,
+            'setupFee': 0,
+            'recurringFee': 0,
+            'hourlyRecurringFee': 0,
+            'oneTimeFee': 0,
+            'laborFee': 0,
+        }]
+        
         self.client['Product_Package'].getAllObjects.return_value = [
             {'name': 'Bare Metal Instance', 'id': package_id}]
 
+        self.client['Product_Package'].getRegions.return_value = [{
+            'location': {
+                'locationPackageDetails': [{
+                    'deliveryTimeInformation': 'Typically 2-4 hours',
+                }],
+            },
+            'keyname': 'RANDOM_LOCATION',
+            'description': 'Random unit testing location',
+        }]
+
+        self.client['Product_Package'].getConfiguration.return_value = [{
+            'itemCategory': {
+                'categoryCode': 'random',
+                'name': 'Random Category',
+            },
+            'sort': 0,
+            'orderStepId': 1,
+            'isRequired': 1,
+            'prices': prices,
+        }]
+
+        self.client['Product_Package'].getItems.return_value = [{
+            'itemCategory': {
+                'categoryCode': 'random',
+                'name': 'Random Category',
+            },
+            'id': 1000,
+            'description': 'Astronaut Sloths',
+            'prices': prices,
+            'capacity': 0,
+            'isRequired': 1,
+        }]
+
         args = {
             'server': 100,
             'hourly': False,
@@ -164,6 +208,7 @@ def test_generate_create_dict_with_all_bare_metal_options(self):
             'os': 200,
             'port_speed': 600,
             'bare_metal': True,
+            'hourly': True,
         }
 
         assert_data = {
@@ -174,11 +219,13 @@ def test_generate_create_dict_with_all_bare_metal_options(self):
             }],
             'location': args['location'],
             'packageId': package_id,
+            'hourlyBillingFlag': True,
             'prices': [
                 {'id': args['server']},
                 {'id': args['disks'][0]},
                 {'id': args['os']},
                 {'id': args['port_speed']},
+                {'id': prices[0]['price_id']},
             ],
         }
 
