test tweaks

Author: alexanderankin

core/testcontainers/core/container.py

@@ -251,7 +251,9 @@ def get_exposed_port(self, port: int) -> int:
         from testcontainers.core.wait_strategies import ContainerStatusWaitStrategy as C
 
         C().wait_until_ready(self)
+        return self._get_exposed_port(port)
 
+    def _get_exposed_port(self, port: int) -> int:
         if self.get_docker_client().get_connection_mode().use_mapped_port:
             c = self._container
             assert c is not None

core/tests/test_container.py

@@ -67,14 +67,14 @@ def fake_mapped(container_id: str, port: int) -> int:
     monkeypatch.setattr(client, "port", fake_mapped)
     monkeypatch.setattr(client, "get_connection_mode", lambda: mode)
 
-    assert container.get_exposed_port(8080) == 45678
+    assert container._get_exposed_port(8080) == 45678
 
 
 def test_get_exposed_port_original(container: DockerContainer, monkeypatch: pytest.MonkeyPatch) -> None:
     client = container._docker
     monkeypatch.setattr(client, "get_connection_mode", lambda: ConnectionMode.bridge_ip)
 
-    assert container.get_exposed_port(8080) == 8080
+    assert container._get_exposed_port(8080) == 8080
 
 
 @pytest.mark.parametrize(

core/tests/test_wait_strategies.py

@@ -484,7 +484,7 @@ def test_wait_until_ready(self, mock_sleep, mock_time, mock_socket, connection_s
             strategy.wait_until_ready(mock_container)
             mock_socket_instance.connect.assert_called_once_with(("localhost", 8080))
         else:
-            with pytest.raises(TimeoutError, match="Port 8080 not available within 1 seconds"):
+            with pytest.raises(TimeoutError, match="Port 8080 not available within 1.0 seconds"):
                 strategy.wait_until_ready(mock_container)