Merge "Add unit tests for "server lock" command."

1 files changed, 20 insertions, 0 deletions

diff --git a/openstackclient/tests/compute/v2/test_server.py b/openstackclient/tests/compute/v2/test_server.py
index ea190fd0..981d9b2a 100644
--- a/openstackclient/tests/compute/v2/test_server.py
+++ b/openstackclient/tests/compute/v2/test_server.py
@@ -611,6 +611,26 @@ class TestServerImageCreate(TestServer):
         self.assertEqual(datalist, data)
 
 
+class TestServerLock(TestServer):
+
+    def setUp(self):
+        super(TestServerLock, self).setUp()
+
+        # Get the command object to test
+        self.cmd = server.LockServer(self.app, None)
+
+        # Set methods to be tested.
+        self.methods = {
+            'lock': None,
+        }
+
+    def test_server_lock_one_server(self):
+        self.run_method_with_servers('lock', 1)
+
+    def test_server_lock_multi_servers(self):
+        self.run_method_with_servers('lock', 3)
+
+
 class TestServerPause(TestServer):
 
     def setUp(self):