From e76809097159e3a0e0e897bb5f42bdb00ea60380 Mon Sep 17 00:00:00 2001
From: Sandeep Yadav <sandyada@redhat.com>
Date: Fri, 29 Jan 2021 19:04:18 +0530
Subject: [PATCH] Unprovision composable networks command

Command to unprovision composable networks which are previously
created by provision command.

Depends-On: https://review.opendev.org/c/openstack/tripleo-ansible/+/773021
Change-Id: I3c72b623590d1f7902648e303ff975077e737d28
---
 setup.cfg                                     |  1 +
 .../test_overcloud_network.py                 | 38 ++++++++++++++
 tripleoclient/v2/overcloud_network.py         | 51 +++++++++++++++++++
 3 files changed, 90 insertions(+)

diff --git a/setup.cfg b/setup.cfg
index 0de4f0cc5..fcbf6887b 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -59,6 +59,7 @@ openstack.tripleoclient.v2 =
     overcloud_network_provision = tripleoclient.v2.overcloud_network:OvercloudNetworkProvision
     overcloud_network_vip_extract = tripleoclient.v2.overcloud_network:OvercloudVirtualIPsExtract
     overcloud_network_vip_provision = tripleoclient.v2.overcloud_network:OvercloudVirtualIPsProvision
+    overcloud_network_unprovision = tripleoclient.v2.overcloud_network:OvercloudNetworkUnprovision
     overcloud_node_configure = tripleoclient.v1.overcloud_node:ConfigureNode
     overcloud_node_delete = tripleoclient.v1.overcloud_node:DeleteNode
     overcloud_node_import = tripleoclient.v2.overcloud_node:ImportNode
diff --git a/tripleoclient/tests/v2/overcloud_network/test_overcloud_network.py b/tripleoclient/tests/v2/overcloud_network/test_overcloud_network.py
index 4c1011652..e5ef59819 100644
--- a/tripleoclient/tests/v2/overcloud_network/test_overcloud_network.py
+++ b/tripleoclient/tests/v2/overcloud_network/test_overcloud_network.py
@@ -119,3 +119,41 @@ class TestOvercloudNetworkProvision(fakes.FakePlaybookExecution):
 
         self.assertRaises(osc_lib_exc.CommandError,
                           self.cmd.take_action, parsed_args)
+
+
+class TestOvercloudNetworkUnprovision(fakes.FakePlaybookExecution):
+
+    def setUp(self):
+        super(TestOvercloudNetworkUnprovision, self).setUp()
+
+        # Get the command object to test
+        app_args = mock.Mock()
+        app_args.verbose_level = 1
+        self.app.options = fakes.FakeOptions()
+        self.cmd = overcloud_network.OvercloudNetworkUnprovision(self.app,
+                                                                 None)
+        self.cmd.app_args = mock.Mock(verbose_level=1)
+
+    @mock.patch('tripleoclient.utils.TempDirs', autospect=True)
+    @mock.patch('os.path.abspath', autospect=True)
+    @mock.patch('os.path.exists', autospect=True)
+    @mock.patch('tripleoclient.utils.run_ansible_playbook', autospec=True)
+    def test_overcloud_network_unprovision(self, mock_playbook,
+                                           mock_path_exists,
+                                           mock_abspath, mock_tempdirs):
+        arglist = ['--yes', 'network_data_v2.yaml']
+        parsed_args = self.check_parser(self.cmd, arglist, [])
+
+        mock_abspath.side_effect = ['/test/network_data_v2.yaml']
+        mock_path_exists.side_effect = [True]
+        self.cmd.take_action(parsed_args)
+        mock_playbook.assert_called_once_with(
+            workdir=mock.ANY,
+            playbook='cli-overcloud-network-unprovision.yaml',
+            inventory=mock.ANY,
+            playbook_dir=mock.ANY,
+            verbosity=3,
+            extra_vars={
+                "network_data_path": '/test/network_data_v2.yaml'
+            }
+        )
diff --git a/tripleoclient/v2/overcloud_network.py b/tripleoclient/v2/overcloud_network.py
index bfcb25e47..46c3cbde0 100644
--- a/tripleoclient/v2/overcloud_network.py
+++ b/tripleoclient/v2/overcloud_network.py
@@ -254,3 +254,54 @@ class OvercloudVirtualIPsProvision(command.Command):
                 verbosity=oooutils.playbook_verbosity(self=self),
                 extra_vars=extra_vars,
             )
+
+
+class OvercloudNetworkUnprovision(command.Command):
+
+    log = logging.getLogger(__name__ + ".OvercloudNetworkUnprovision")
+
+    def get_parser(self, prog_name):
+        parser = super(OvercloudNetworkUnprovision, self).get_parser(prog_name)
+
+        parser.add_argument('networks_file',
+                            metavar='<network_data.yaml>',
+                            help=_('Configuration file describing the network '
+                                   'deployment.'))
+        parser.add_argument('-y', '--yes',
+                            help=_('Skip yes/no prompt (assume yes).'),
+                            default=False,
+                            action="store_true")
+
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)" % parsed_args)
+
+        networks_file_path = os.path.abspath(parsed_args.networks_file)
+
+        if not parsed_args.yes:
+            confirm = oooutils.prompt_user_for_confirmation(
+                message=_("Are you sure you want to unprovision the networks "
+                          "mentioned in file %s [y/N]? " % networks_file_path),
+                logger=self.log)
+            if not confirm:
+                raise oscexc.CommandError("Action not confirmed, exiting.")
+
+        if not os.path.exists(networks_file_path):
+            raise oscexc.CommandError(
+                "Network configuration file does not exist:"
+                " %s" % parsed_args.networks_file)
+
+        extra_vars = {
+            "network_data_path": networks_file_path,
+        }
+
+        with oooutils.TempDirs() as tmp:
+            oooutils.run_ansible_playbook(
+                playbook='cli-overcloud-network-unprovision.yaml',
+                inventory='localhost,',
+                workdir=tmp,
+                playbook_dir=constants.ANSIBLE_TRIPLEO_PLAYBOOKS,
+                verbosity=oooutils.playbook_verbosity(self=self),
+                extra_vars=extra_vars,
+            )