From 08b0e5855be5dc12a2fd04c5c92027c8a85a0d00 Mon Sep 17 00:00:00 2001
From: jay <jayadityagupta11@gmail.com>
Date: Tue, 21 Apr 2020 14:23:58 +0200
Subject: Add 'server migration force complete' command

This is equivalent to novaclient's 'live-migration-force-complete'
command.

Change-Id: Ic4dc639afa16cdf8c5a46774895e850d92985292
Story: 2007513
Task: 39293
---
 .../tests/unit/compute/v2/test_server.py           | 51 ++++++++++++++++++++++
 1 file changed, 51 insertions(+)

(limited to 'openstackclient/tests/unit')

diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py
index 9ac29352..3b98c873 100644
--- a/openstackclient/tests/unit/compute/v2/test_server.py
+++ b/openstackclient/tests/unit/compute/v2/test_server.py
@@ -4263,6 +4263,57 @@ class TestServerMigrationAbort(TestServer):
             str(ex))
 
 
+class TestServerMigrationForceComplete(TestServer):
+
+    def setUp(self):
+        super(TestServerMigrationForceComplete, self).setUp()
+
+        self.server = compute_fakes.FakeServer.create_one_server()
+
+        # Return value for utils.find_resource for server.
+        self.servers_mock.get.return_value = self.server
+
+        # Get the command object to test
+        self.cmd = server.ForceCompleteMigration(self.app, None)
+
+    def test_migration_force_complete(self):
+        self.app.client_manager.compute.api_version = api_versions.APIVersion(
+            '2.22')
+
+        arglist = [
+            self.server.id,
+            '2',  # arbitrary migration ID
+        ]
+        verifylist = []
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        self.servers_mock.get.assert_called_with(self.server.id)
+        self.server_migrations_mock.live_migrate_force_complete\
+            .assert_called_with(self.server.id, '2',)
+        self.assertIsNone(result)
+
+    def test_migration_force_complete_pre_v222(self):
+        self.app.client_manager.compute.api_version = api_versions.APIVersion(
+            '2.21')
+
+        arglist = [
+            self.server.id,
+            '2',  # arbitrary migration ID
+        ]
+        verifylist = []
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        ex = self.assertRaises(
+            exceptions.CommandError,
+            self.cmd.take_action,
+            parsed_args)
+        self.assertIn(
+            '--os-compute-api-version 2.22 or greater is required',
+            str(ex))
+
+
 class TestServerPause(TestServer):
 
     def setUp(self):
-- 
cgit v1.2.1