1 files changed, 106 insertions, 0 deletions
diff --git a/openstackclient/tests/network/v2/test_subnet.py b/openstackclient/tests/network/v2/test_subnet.py
index 99b558c0..82813d6c 100644
--- a/openstackclient/tests/network/v2/test_subnet.py
+++ b/openstackclient/tests/network/v2/test_subnet.py
@@ -767,3 +767,109 @@ class TestShowSubnet(TestSubnet):
 
         self.assertEqual(self.columns, columns)
         self.assertEqual(self.data, data)
+
+
+class TestUnsetSubnet(TestSubnet):
+
+    def setUp(self):
+        super(TestUnsetSubnet, self).setUp()
+        self._testsubnet = network_fakes.FakeSubnet.create_one_subnet(
+            {'dns_nameservers': ['8.8.8.8',
+                                 '8.8.8.4'],
+             'host_routes': [{'destination': '10.20.20.0/24',
+                              'nexthop': '10.20.20.1'},
+                             {'destination': '10.30.30.30/24',
+                              'nexthop': '10.30.30.1'}],
+             'allocation_pools': [{'start': '8.8.8.100',
+                                   'end': '8.8.8.150'},
+                                  {'start': '8.8.8.160',
+                                   'end': '8.8.8.170'}], })
+        self.network.find_subnet = mock.Mock(return_value=self._testsubnet)
+        self.network.update_subnet = mock.Mock(return_value=None)
+        # Get the command object to test
+        self.cmd = subnet_v2.UnsetSubnet(self.app, self.namespace)
+
+    def test_unset_subnet_params(self):
+        arglist = [
+            '--dns-nameserver', '8.8.8.8',
+            '--host-route', 'destination=10.30.30.30/24,gateway=10.30.30.1',
+            '--allocation-pool', 'start=8.8.8.100,end=8.8.8.150',
+            self._testsubnet.name,
+        ]
+        verifylist = [
+            ('dns_nameservers', ['8.8.8.8']),
+            ('host_routes', [{
+                "destination": "10.30.30.30/24", "gateway": "10.30.30.1"}]),
+            ('allocation_pools', [{
+                'start': '8.8.8.100', 'end': '8.8.8.150'}]),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+        result = self.cmd.take_action(parsed_args)
+
+        attrs = {
+            'dns_nameservers': ['8.8.8.4'],
+            'host_routes': [{
+                "destination": "10.20.20.0/24", "nexthop": "10.20.20.1"}],
+            'allocation_pools': [{'start': '8.8.8.160', 'end': '8.8.8.170'}],
+        }
+        self.network.update_subnet.assert_called_once_with(
+            self._testsubnet, **attrs)
+        self.assertIsNone(result)
+
+    def test_unset_subnet_wrong_host_routes(self):
+        arglist = [
+            '--dns-nameserver', '8.8.8.8',
+            '--host-route', 'destination=10.30.30.30/24,gateway=10.30.30.2',
+            '--allocation-pool', 'start=8.8.8.100,end=8.8.8.150',
+            self._testsubnet.name,
+        ]
+        verifylist = [
+            ('dns_nameservers', ['8.8.8.8']),
+            ('host_routes', [{
+                "destination": "10.30.30.30/24", "gateway": "10.30.30.2"}]),
+            ('allocation_pools', [{
+                'start': '8.8.8.100', 'end': '8.8.8.150'}]),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+        self.assertRaises(exceptions.CommandError,
+                          self.cmd.take_action, parsed_args)
+
+    def test_unset_subnet_wrong_allocation_pool(self):
+        arglist = [
+            '--dns-nameserver', '8.8.8.8',
+            '--host-route', 'destination=10.30.30.30/24,gateway=10.30.30.1',
+            '--allocation-pool', 'start=8.8.8.100,end=8.8.8.156',
+            self._testsubnet.name,
+        ]
+        verifylist = [
+            ('dns_nameservers', ['8.8.8.8']),
+            ('host_routes', [{
+                "destination": "10.30.30.30/24", "gateway": "10.30.30.1"}]),
+            ('allocation_pools', [{
+                'start': '8.8.8.100', 'end': '8.8.8.156'}]),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+        self.assertRaises(exceptions.CommandError,
+                          self.cmd.take_action, parsed_args)
+
+    def test_unset_subnet_wrong_dns_nameservers(self):
+        arglist = [
+            '--dns-nameserver', '8.8.8.1',
+            '--host-route', 'destination=10.30.30.30/24,gateway=10.30.30.1',
+            '--allocation-pool', 'start=8.8.8.100,end=8.8.8.150',
+            self._testsubnet.name,
+        ]
+        verifylist = [
+            ('dns_nameservers', ['8.8.8.1']),
+            ('host_routes', [{
+                "destination": "10.30.30.30/24", "gateway": "10.30.30.1"}]),
+            ('allocation_pools', [{
+                'start': '8.8.8.100', 'end': '8.8.8.150'}]),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+        self.assertRaises(exceptions.CommandError,
+                          self.cmd.take_action, parsed_args)