add subnet edit tests

Author: ATGE

SoftLayer/fixtures/SoftLayer_Network_Subnet.py

@@ -25,5 +25,16 @@
         }
     ],
     'hardware': [],
-    'usableIpAddressCount': 22
+    'usableIpAddressCount': 22,
+    'note': 'test note',
+    'tagReferences': [
+        {'id': 1000123,
+         'resourceTableId': 1234,
+         'tag': {'id': 100123,
+                 'name': 'subnet: test tag'},
+         }
+    ]
 }
+
+editNote = True
+setTags = True

tests/CLI/modules/subnet_tests.py

@@ -36,8 +36,12 @@ def test_detail(self):
                         'private_ip': '10.0.1.2'
                     }
                 ],
-                'hardware': 'none',
-                'usable ips': 22
+                'hardware': None,
+                'usable ips': 22,
+                'note': 'test note',
+                'tags': [
+                    'subnet: test tag'
+                ],
             },
             json.loads(result.output))
 
@@ -134,3 +138,35 @@ def test_create_subnet_static_ipv6(self, confirm_mock):
         ]
 
         self.assertEqual(output, json.loads(result.output))
+
+    @mock.patch('SoftLayer.CLI.subnet.edit.click')
+    def test_subnet_set_tags(self, click):
+        result = self.run_command(['subnet', 'edit', '1234', '--tags=tag1,tag2'])
+        click.secho.assert_called_with('Set tags successfully', fg='green')
+        self.assert_no_fail(result)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'setTags', identifier=1234, args=("tag1,tag2",))
+
+    @mock.patch('SoftLayer.CLI.subnet.edit.click')
+    def test_subnet_edit_note(self, click):
+        result = self.run_command(['subnet', 'edit', '1234', '--note=test'])
+        click.secho.assert_called_with('Edit note successfully', fg='green')
+        self.assert_no_fail(result)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'editNote', identifier=1234, args=("test",))
+
+    @mock.patch('SoftLayer.CLI.subnet.edit.click')
+    def test_subnet_set_tags_failure(self, click):
+        mock = self.set_mock('SoftLayer_Network_Subnet', 'setTags')
+        mock.return_value = False
+        result = self.run_command(['subnet', 'edit', '1234', '--tags=tag1,tag2'])
+        click.secho.assert_called_with('Failed to set tags', fg='red')
+        self.assert_no_fail(result)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'setTags', identifier=1234, args=("tag1,tag2",))
+
+    @mock.patch('SoftLayer.CLI.subnet.edit.click')
+    def test_edit_note_failure(self, click):
+        mock = self.set_mock('SoftLayer_Network_Subnet', 'editNote')
+        mock.return_value = False
+        result = self.run_command(['subnet', 'edit', '1234', '--note=test'])
+        click.secho.assert_called_with('Failed to edit note', fg='red')
+        self.assert_no_fail(result)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'editNote', identifier=1234, args=("test",))

tests/managers/network_tests.py

@@ -158,6 +158,26 @@ def test_cancel_subnet(self):
         self.assert_called_with('SoftLayer_Billing_Item', 'cancelService',
                                 identifier=1056)
 
+    def test_set_tags_subnet(self):
+        subnet_id = 1234
+        tags = 'tags1,tag2'
+        result = self.network.set_tags_subnet(subnet_id, tags)
+
+        self.assertEqual(result, True)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'setTags',
+                                identifier=subnet_id,
+                                args=(tags,))
+
+    def test_edit_note_subnet(self):
+        subnet_id = 1234
+        note = 'test note'
+        result = self.network.edit_note_subnet(subnet_id, note)
+
+        self.assertEqual(result, True)
+        self.assert_called_with('SoftLayer_Network_Subnet', 'editNote',
+                                identifier=subnet_id,
+                                args=(note,))
+
     def test_create_securitygroup(self):
         result = self.network.create_securitygroup(name='foo',
                                                    description='bar')