update testcontainer

HaoXuAI · HaoXuAI · commit 33b46bd779df · 2024-04-11T21:42:52.000-07:00
diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py b/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py
@@ -17,7 +17,7 @@
 import time
 from typing import Dict
 
-from testcontainers.core.container import DockerContainer
+from testcontainers.cassandra import CassandraContainer
 from testcontainers.core.waiting_utils import wait_for_logs
 
 from tests.integration.feature_repos.universal.online_store_creator import (
@@ -28,8 +28,8 @@
 class CassandraOnlineStoreCreator(OnlineStoreCreator):
     def __init__(self, project_name: str, **kwargs):
         super().__init__(project_name)
-        self.container = DockerContainer("library/cassandra:4.0.4").with_exposed_ports(
-            "9042"
+        self.container = CassandraContainer("cassandra:4.1.4").with_exposed_ports(
+            9042
         )
 
     def create_online_store(self) -> Dict[str, object]:
diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/postgres.py b/sdk/python/tests/integration/feature_repos/universal/online_store/postgres.py
@@ -11,7 +11,7 @@ class PostgresOnlieStoreCreator(OnlineStoreCreator):
     def __init__(self, project_name: str, **kwargs):
         super().__init__(project_name)
         self.container = (
-            PostgresContainer("postgres:latest", platform="linux/amd64")
+            PostgresContainer("postgres:16", platform="linux/amd64")
             .with_exposed_ports(5432)
             .with_env("POSTGRES_USER", "root")
             .with_env("POSTGRES_PASSWORD", "test")
diff --git a/setup.py b/setup.py
@@ -177,7 +177,7 @@
         "pytest-mock==1.10.4",
         "pytest-env",
         "Sphinx>4.0.0,<7",
-        "testcontainers>=3.5,<4",
+        "testcontainers==4.3.3",
         "firebase-admin>=5.2.0,<6",
         "pre-commit<3.3.2",
         "assertpy==1.1",
