From a9bc84cece547d101069afd195d1500f67e3a44a Mon Sep 17 00:00:00 2001 From: Joffrey F Date: Wed, 26 Nov 2014 23:14:19 +0100 Subject: Enforce passing string as version param in ctor --- docker/client.py | 6 ++++++ tests/test.py | 11 +++++++++++ 2 files changed, 17 insertions(+) diff --git a/docker/client.py b/docker/client.py index de3bce6..06fa145 100644 --- a/docker/client.py +++ b/docker/client.py @@ -50,6 +50,12 @@ class Client(requests.Session): raise errors.TLSParameterError( 'If using TLS, the base_url argument must begin with ' '"https://".') + if not isinstance(version, six.string_types): + raise errors.DockerException( + 'version parameter must be a string. Found {0}'.format( + type(version).__name__ + ) + ) self.base_url = base_url self._version = version self._timeout = timeout diff --git a/tests/test.py b/tests/test.py index c9a1b05..38cbc2f 100644 --- a/tests/test.py +++ b/tests/test.py @@ -102,6 +102,17 @@ class DockerClientTest(Cleanup, unittest.TestCase): def tearDown(self): self.client.close() + def test_ctor(self): + try: + docker.Client(version=1.12) + except Exception as e: + self.assertTrue(isinstance(e, docker.errors.DockerException)) + if not six.PY3: + self.assertEqual( + e.message, + 'version parameter must be a string. Found float' + ) + ######################### # INFORMATION TESTS # ######################### -- cgit v1.2.1