Răsfoiți Sursa

Fix stacktrace when handling timeout error

Signed-off-by: Aanand Prasad <[email protected]>
Aanand Prasad 9 ani în urmă
părinte
comite
5aeeecb6f2
2 a modificat fișierele cu 9 adăugiri și 1 ștergeri
  1. 1 1
      compose/cli/errors.py
  2. 8 0
      tests/unit/cli/docker_client_test.py

+ 1 - 1
compose/cli/errors.py

@@ -53,7 +53,7 @@ def handle_connection_errors(client):
         log_api_error(e, client.api_version)
         raise ConnectionError()
     except (ReadTimeout, socket.timeout) as e:
-        log_timeout_error()
+        log_timeout_error(client.timeout)
         raise ConnectionError()
 
 

+ 8 - 0
tests/unit/cli/docker_client_test.py

@@ -42,6 +42,14 @@ class DockerClientTestCase(unittest.TestCase):
         assert fake_log.error.call_count == 1
         assert '123' in fake_log.error.call_args[0][0]
 
+        with mock.patch('compose.cli.errors.log') as fake_log:
+            with pytest.raises(errors.ConnectionError):
+                with errors.handle_connection_errors(client):
+                    raise errors.ReadTimeout()
+
+        assert fake_log.error.call_count == 1
+        assert '123' in fake_log.error.call_args[0][0]
+
     def test_user_agent(self):
         client = docker_client(os.environ)
         expected = "docker-compose/{0} docker-py/{1} {2}/{3}".format(