From 5c0959c4fedf541fa40d543f17b9e97dedabab9d Mon Sep 17 00:00:00 2001
From: Tang Chen <tangchen@cn.fujitsu.com>
Date: Thu, 12 Nov 2015 13:22:06 +0800
Subject: [PATCH] Add unit tests for "server pause" command.

This patch adds unit tests for "server pause" command,
including one and multiple servers.

Change-Id: If5551e77d7dd4f7f48c6ee4a7f80f8313817f492
Implements: blueprint osc-unit-test-framework-improvement
---
 .../tests/compute/v2/test_server.py           | 57 +++++++++++++++++++
 1 file changed, 57 insertions(+)

diff --git a/openstackclient/tests/compute/v2/test_server.py b/openstackclient/tests/compute/v2/test_server.py
index f6f9797acf..791a90abd1 100644
--- a/openstackclient/tests/compute/v2/test_server.py
+++ b/openstackclient/tests/compute/v2/test_server.py
@@ -528,6 +528,63 @@ class TestServerImageCreate(TestServer):
         self.assertEqual(datalist, data)
 
 
+class TestServerPause(TestServer):
+
+    def setUp(self):
+        super(TestServerPause, self).setUp()
+
+        # Get the command object to test
+        self.cmd = server.PauseServer(self.app, None)
+
+        # Set methods to be tested.
+        self.methods = {
+            'pause': None,
+        }
+
+    def setup_servers_mock(self, count=1):
+        servers = fakes.FakeServer.create_servers(methods=self.methods,
+                                                  count=count)
+
+        # This is the return value for utils.find_resource()
+        self.servers_mock.get = fakes.FakeServer.get_servers(servers, 1)
+
+        return servers
+
+    def test_server_pause_one_server(self):
+        servers = self.setup_servers_mock(1)
+
+        arglist = [
+            servers[0].id,
+        ]
+        verifylist = [
+            ('server', [servers[0].id]),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        servers[0].pause.assert_called_with()
+
+    def test_server_pause_multi_servers(self):
+        servers = self.setup_servers_mock(3)
+        arglist = []
+        verifylist = []
+
+        for i in range(0, len(servers)):
+            arglist.append(servers[i].id)
+        verifylist = [
+            ('server', arglist),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        for i in range(0, len(servers)):
+            servers[i].pause.assert_called_with()
+
+
 class TestServerResize(TestServer):
 
     def setUp(self):