From 1fd5813c0eebed9cb65a49a47ff1a5a09d510448 Mon Sep 17 00:00:00 2001 From: Torsten Kilias Date: Thu, 25 Jan 2024 15:30:44 +0100 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Christoph Kuhnke --- test/unit/test_current_container_finder.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/unit/test_current_container_finder.py b/test/unit/test_current_container_finder.py index 5090e07..c6d270e 100644 --- a/test/unit/test_current_container_finder.py +++ b/test/unit/test_current_container_finder.py @@ -30,8 +30,12 @@ def create_ip_retriever_mock(ip_addresses: List[str]) -> Union[MagicMock, IPRetr return ip_retriever_mock +from typing import NewType +Network = NewType("Network", Dict[str, str]) + + def create_docker_client_mock( - containers: List[Dict[str, str]] + containers: List[Network] ) -> Union[docker.DockerClient, MagicMock]: docker_client_mock: Union[MagicMock, docker.DockerClient] = create_autospec( docker.DockerClient