diff --git a/tests/utils/tracked_container.py b/tests/utils/tracked_container.py index a912da5f..5ae1bbd0 100644 --- a/tests/utils/tracked_container.py +++ b/tests/utils/tracked_container.py @@ -57,8 +57,8 @@ class TrackedContainer: def get_health(self) -> str: assert self.container is not None - inspect_results = self.docker_client.api.inspect_container(self.container.name) - return inspect_results["State"]["Health"]["Status"] # type: ignore + self.container.reload() + return self.container.health # type: ignore def exec_cmd(self, cmd: str, **kwargs: Any) -> str: assert self.container is not None