fix log getting

lsst · Nov 15, 2023 · cff4fae · cff4fae
1 parent c725ba9
commit cff4fae
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/tests_integration/test_docker_container.py b/tests_integration/test_docker_container.py
@@ -33,14 +33,16 @@ def _run_server_docker():
             server_host = container.get_container_host_ip()
             server_port = container.get_exposed_port(port)
             server_url = f"http://{server_host}:{server_port}/api/butler"
-            yield server_url
-        (stdout, stderr) = container.get_logs()
-        if stdout:
-            print("STDOUT:")
-            print(stdout)
-        if stderr:
-            print("STDERR:")
-            print(stderr)
+            try:
+                yield server_url
+            finally:
+                (stdout, stderr) = container.get_logs()
+                if stdout:
+                    print("STDOUT:")
+                    print(stdout)
+                if stderr:
+                    print("STDERR:")
+                    print(stderr)
 
 
 class ButlerDockerTestCase(unittest.TestCase):