Update arguments for several Swarm API methods

Add Client.update_swarm method
Add test for Client.update_swarm

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

1 files changed, 31 insertions, 0 deletions

diff --git a/tests/integration/swarm_test.py b/tests/integration/swarm_test.py
index 969e05e..226689b 100644
--- a/tests/integration/swarm_test.py
+++ b/tests/integration/swarm_test.py
@@ -65,3 +65,34 @@ class SwarmTest(helpers.BaseTestCase):
         with pytest.raises(docker.errors.APIError) as exc_info:
             self.client.inspect_swarm()
         exc_info.value.response.status_code == 406
+
+    @requires_api_version('1.24')
+    def test_update_swarm(self):
+        assert self.client.init_swarm('eth0')
+        swarm_info_1 = self.client.inspect_swarm()
+        spec = self.client.create_swarm_spec(
+            snapshot_interval=5000, log_entries_for_slow_followers=1200,
+            node_cert_expiry=7776000000000000
+        )
+        assert self.client.update_swarm(
+            version=swarm_info_1['Version']['Index'],
+            swarm_spec=spec, rotate_worker_token=True
+        )
+        swarm_info_2 = self.client.inspect_swarm()
+
+        assert (
+            swarm_info_1['Version']['Index'] !=
+            swarm_info_2['Version']['Index']
+        )
+        assert swarm_info_2['Spec']['Raft']['SnapshotInterval'] == 5000
+        assert (
+            swarm_info_2['Spec']['Raft']['LogEntriesForSlowFollowers'] == 1200
+        )
+        assert (
+            swarm_info_1['JoinTokens']['Manager'] ==
+            swarm_info_2['JoinTokens']['Manager']
+        )
+        assert (
+            swarm_info_1['JoinTokens']['Worker'] !=
+            swarm_info_2['JoinTokens']['Worker']
+        )