Add unlock methods to Swarm modelunlock_swarm_support

Signed-off-by: Joffrey F <joffrey@docker.com>

1 files changed, 6 insertions, 5 deletions

diff --git a/tests/integration/api_swarm_test.py b/tests/integration/api_swarm_test.py
index 1a945c5..56b0129 100644
--- a/tests/integration/api_swarm_test.py
+++ b/tests/integration/api_swarm_test.py
@@ -10,13 +10,13 @@ class SwarmTest(BaseAPIIntegrationTest):
     def setUp(self):
         super(SwarmTest, self).setUp()
         force_leave_swarm(self.client)
+        self._unlock_key = None
 
     def tearDown(self):
         super(SwarmTest, self).tearDown()
         try:
-            unlock_key = self.client.get_unlock_key()
-            if unlock_key.get('UnlockKey'):
-                self.unlock_swarm(unlock_key)
+            if self._unlock_key:
+                self.client.unlock_swarm(self._unlock_key)
         except docker.errors.APIError:
             pass
 
@@ -71,14 +71,15 @@ class SwarmTest(BaseAPIIntegrationTest):
     def test_init_swarm_with_autolock_managers(self):
         spec = self.client.create_swarm_spec(autolock_managers=True)
         assert self.init_swarm(swarm_spec=spec)
+        # save unlock key for tearDown
+        self._unlock_key = self.client.get_unlock_key()
         swarm_info = self.client.inspect_swarm()
 
         assert (
             swarm_info['Spec']['EncryptionConfig']['AutoLockManagers'] is True
         )
 
-        unlock_key = self.get_unlock_key()
-        assert unlock_key.get('UnlockKey')
+        assert self._unlock_key.get('UnlockKey')
 
     @requires_api_version('1.25')
     @pytest.mark.xfail(