Some minor cleanup from yelp/fig

Signed-off-by: Daniel Nephin <dnephin@yelp.com>

1 files changed, 6 insertions, 0 deletions

diff --git a/tests/unit/cli/docker_client_test.py b/tests/unit/cli/docker_client_test.py
index c8821073..67575ee0 100644
--- a/tests/unit/cli/docker_client_test.py
+++ b/tests/unit/cli/docker_client_test.py
@@ -14,3 +14,9 @@ class DockerClientTestCase(unittest.TestCase):
         with mock.patch.dict(os.environ):
             del os.environ['HOME']
             docker_client.docker_client()
+
+    def test_docker_client_with_custom_timeout(self):
+        with mock.patch.dict(os.environ):
+            os.environ['DOCKER_CLIENT_TIMEOUT'] = timeout = "300"
+            client = docker_client.docker_client()
+        self.assertEqual(client._timeout, int(timeout))