Required wait_strategy

Author: Tranquility2

modules/generic/testcontainers/generic/sql.py

@@ -6,8 +6,6 @@
 from testcontainers.core.exceptions import ContainerStartException
 from testcontainers.core.waiting_utils import WaitStrategy
 
-from .sql_utils import SqlConnectWaitStrategy
-
 logger = logging.getLogger(__name__)
 
 
@@ -20,17 +18,17 @@ class SqlContainer(DockerContainer):
     Database connection readiness is automatically handled by the provided wait strategy.
     """
 
-    def __init__(self, image: str, wait_strategy: Optional[WaitStrategy] = None, **kwargs):
+    def __init__(self, image: str, wait_strategy: WaitStrategy, **kwargs):
         """
         Initialize SqlContainer with optional wait strategy.
 
         Args:
             image: Docker image name
-            wait_strategy: Wait strategy for SQL database connectivity (defaults to SqlConnectWaitStrategy)
+            wait_strategy: Wait strategy for SQL database connectivity
             **kwargs: Additional arguments passed to DockerContainer
         """
         super().__init__(image, **kwargs)
-        self.wait_strategy = wait_strategy or SqlConnectWaitStrategy()
+        self.wait_strategy = wait_strategy
 
     def _create_connection_url(
         self,

modules/generic/tests/test_sql.py

@@ -1,13 +1,15 @@
 import pytest
+
 from testcontainers.core.exceptions import ContainerStartException
+from testcontainers.generic.sql_utils import SqlConnectWaitStrategy
 from testcontainers.generic.sql import SqlContainer
 
 
 class SimpleSqlContainer(SqlContainer):
     """Simple concrete implementation for testing."""
 
     def __init__(self, image: str = "postgres:13"):
-        super().__init__(image)
+        super().__init__(image, wait_strategy=SqlConnectWaitStrategy())
         self.username = "testuser"
         self.password = "testpass"
         self.dbname = "testdb"
@@ -27,7 +29,7 @@ def _configure(self) -> None:
 
 class TestSqlContainer:
     def test_abstract_methods_raise_not_implemented(self):
-        container = SqlContainer("test:latest")
+        container = SqlContainer("test:latest", SqlConnectWaitStrategy())
 
         with pytest.raises(NotImplementedError):
             container.get_connection_url()
@@ -36,7 +38,7 @@ def test_abstract_methods_raise_not_implemented(self):
             container._configure()
 
     def test_transfer_seed_default_behavior(self):
-        container = SqlContainer("test:latest")
+        container = SqlContainer("test:latest", SqlConnectWaitStrategy())
         # Should not raise an exception
         container._transfer_seed()