1 files changed, 135 insertions, 0 deletions
diff --git a/openstackclient/tests/compute/v2/test_server.py b/openstackclient/tests/compute/v2/test_server.py
index efe4c58b..a98cd156 100644
--- a/openstackclient/tests/compute/v2/test_server.py
+++ b/openstackclient/tests/compute/v2/test_server.py
@@ -30,6 +30,10 @@ class TestServer(compute_fakes.TestComputev2):
         self.servers_mock = self.app.client_manager.compute.servers
         self.servers_mock.reset_mock()
 
+        # Get a shortcut to the FlavorManager Mock
+        self.flavors_mock = self.app.client_manager.compute.flavors
+        self.flavors_mock.reset_mock()
+
         # Get a shortcut to the ImageManager Mock
         self.images_mock = self.app.client_manager.image.images
         self.images_mock.reset_mock()
@@ -148,3 +152,134 @@ class TestServerImageCreate(TestServer):
             image_fakes.image_owner,
         )
         self.assertEqual(data, datalist)
+
+
+class TestServerResize(TestServer):
+
+    def setUp(self):
+        super(TestServerResize, self).setUp()
+
+        # This is the return value for utils.find_resource()
+        self.servers_get_return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(compute_fakes.SERVER),
+            loaded=True,
+        )
+        self.servers_mock.get.return_value = self.servers_get_return_value
+
+        self.servers_mock.resize.return_value = None
+        self.servers_mock.confirm_resize.return_value = None
+        self.servers_mock.revert_resize.return_value = None
+
+        # This is the return value for utils.find_resource()
+        self.flavors_get_return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(compute_fakes.FLAVOR),
+            loaded=True,
+        )
+        self.flavors_mock.get.return_value = self.flavors_get_return_value
+
+        # Get the command object to test
+        self.cmd = server.ResizeServer(self.app, None)
+
+    def test_server_resize_no_options(self):
+        arglist = [
+            compute_fakes.server_id,
+        ]
+        verifylist = [
+            ('verify', False),
+            ('revert', False),
+            ('server', compute_fakes.server_id),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.servers_mock.get.assert_called_with(
+            compute_fakes.server_id,
+        )
+
+        self.assertNotCalled(self.servers_mock.resize)
+        self.assertNotCalled(self.servers_mock.confirm_resize)
+        self.assertNotCalled(self.servers_mock.revert_resize)
+
+    def test_server_resize(self):
+        arglist = [
+            '--flavor', compute_fakes.flavor_id,
+            compute_fakes.server_id,
+        ]
+        verifylist = [
+            ('flavor', compute_fakes.flavor_id),
+            ('verify', False),
+            ('revert', False),
+            ('server', compute_fakes.server_id),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.servers_mock.get.assert_called_with(
+            compute_fakes.server_id,
+        )
+        self.flavors_mock.get.assert_called_with(
+            compute_fakes.flavor_id,
+        )
+
+        self.servers_mock.resize.assert_called_with(
+            self.servers_get_return_value,
+            self.flavors_get_return_value,
+        )
+        self.assertNotCalled(self.servers_mock.confirm_resize)
+        self.assertNotCalled(self.servers_mock.revert_resize)
+
+    def test_server_resize_confirm(self):
+        arglist = [
+            '--verify',
+            compute_fakes.server_id,
+        ]
+        verifylist = [
+            ('verify', True),
+            ('revert', False),
+            ('server', compute_fakes.server_id),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.servers_mock.get.assert_called_with(
+            compute_fakes.server_id,
+        )
+
+        self.assertNotCalled(self.servers_mock.resize)
+        self.servers_mock.confirm_resize.assert_called_with(
+            self.servers_get_return_value,
+        )
+        self.assertNotCalled(self.servers_mock.revert_resize)
+
+    def test_server_resize_revert(self):
+        arglist = [
+            '--revert',
+            compute_fakes.server_id,
+        ]
+        verifylist = [
+            ('verify', False),
+            ('revert', True),
+            ('server', compute_fakes.server_id),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.servers_mock.get.assert_called_with(
+            compute_fakes.server_id,
+        )
+
+        self.assertNotCalled(self.servers_mock.resize)
+        self.assertNotCalled(self.servers_mock.confirm_resize)
+        self.servers_mock.revert_resize.assert_called_with(
+            self.servers_get_return_value,
+        )