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
This commit is contained in:
Tang Chen 2015-11-12 13:22:06 +08:00
parent b1cc7fb4f6
commit 5c0959c4fe

View File

@ -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):