#1129 unit tests

Author: acamacho82

tests/CLI/modules/subnet_tests.py

@@ -4,12 +4,17 @@
 
     :license: MIT, see LICENSE for more details.
 """
+from SoftLayer.fixtures import SoftLayer_Product_Order
+from SoftLayer.fixtures import SoftLayer_Product_Package
 from SoftLayer import testing
 
 import json
+import mock
+import SoftLayer
 
 
 class SubnetTests(testing.TestCase):
+
     def test_detail(self):
         result = self.run_command(['subnet', 'detail', '1234'])
 
@@ -39,3 +44,55 @@ def test_detail(self):
     def test_list(self):
         result = self.run_command(['subnet',  'list'])
         self.assert_no_fail(result)
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    def test_create_subnet_ipv4(self, confirm_mock):
+        confirm_mock.return_value = True
+
+        item_mock = self.set_mock('SoftLayer_Product_Package', 'getItems')
+        item_mock.return_value = SoftLayer_Product_Package.getItems
+
+        place_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder')
+        place_mock.return_value = SoftLayer_Product_Order.placeOrder
+
+        result = self.run_command(['subnet', 'create', 'private', '8', '12346'])
+        self.assert_no_fail(result)
+
+        output = [
+            {'Item': 'this is a thing', 'cost': '2.00'},
+            {'Item': 'Total monthly cost', 'cost': '2.00'}
+        ]
+
+        self.assertEqual(output, json.loads(result.output))
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    def test_create_subnet_ipv6(self, confirm_mock):
+        confirm_mock.return_value = True
+
+        item_mock = self.set_mock('SoftLayer_Product_Package', 'getItems')
+        item_mock.return_value = SoftLayer_Product_Package.getItems
+
+        place_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder')
+        place_mock.return_value = SoftLayer_Product_Order.placeOrder
+
+        result = self.run_command(['subnet', 'create', '--v6', 'public', '64', '12346', '--test'])
+        self.assert_no_fail(result)
+
+        output = [
+            {'Item': 'this is a thing', 'cost': '2.00'},
+            {'Item': 'Total monthly cost', 'cost': '2.00'}
+        ]
+
+        self.assertEqual(output, json.loads(result.output))
+
+    def test_create_subnet_no_prices_found(self):
+        item_mock = self.set_mock('SoftLayer_Product_Package', 'getItems')
+        item_mock.return_value = SoftLayer_Product_Package.getItems
+
+        verify_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder')
+        verify_mock.side_effect = SoftLayer.SoftLayerAPIError('SoftLayer_Exception', 'Price not found')
+
+        result = self.run_command(['subnet', 'create', '--v6', 'public', '32', '12346', '--test'])
+
+        self.assertRaises(SoftLayer.SoftLayerAPIError, verify_mock)
+        self.assertEqual(result.exception.message, 'There is no price id for 32 public ipv6')

tests/managers/network_tests.py

@@ -24,9 +24,6 @@ def test_ip_lookup(self):
                                 'getByIpAddress',
                                 args=('10.0.1.37',))
 
-    def test_add_subnet_raises_exception_on_failure(self):
-        self.assertRaises(TypeError, self.network.add_subnet, ('bad'))
-
     def test_add_global_ip(self):
         # Test a global IPv4 order
         result = self.network.add_global_ip(test_order=True)