fix postgres integration test

HaoXuAI · HaoXuAI · commit 061837868345 · 2024-04-12T14:22:34.000-07:00
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py b/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py
@@ -1,12 +1,12 @@
-from feast.infra.offline_stores.contrib.postgres_offline_store.tests.data_source import (
-    PostgreSQLDataSourceCreator,
+from tests.integration.feature_repos.universal.online_store.postgres import (
+    PostgresOnlieStoreCreator,
 )
 from tests.integration.feature_repos.integration_test_repo_config import (
     IntegrationTestRepoConfig,
 )
 
 FULL_REPO_CONFIGS = [
     IntegrationTestRepoConfig(
-        online_store="postgres", online_store_creator=PostgreSQLDataSourceCreator
+        online_store="postgres", online_store_creator=PostgresOnlieStoreCreator
     ),
 ]
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,22 +11,24 @@ class PostgresOnlieStoreCreator(OnlineStoreCreator):
     def __init__(self, project_name: str, **kwargs):
         super().__init__(project_name)
         self.container = (
-            PostgresContainer("postgres:16", platform="linux/amd64")
+            PostgresContainer(
+                "postgres:16",
+                username="root",
+                password="test",
+                dbname="test",
+                )
             .with_exposed_ports(5432)
-            .with_env("POSTGRES_USER", "root")
-            .with_env("POSTGRES_PASSWORD", "test")
-            .with_env("POSTGRES_DB", "test")
         )
 
     def create_online_store(self) -> Dict[str, str]:
         self.container.start()
-        exposed_port = self.container.get_exposed_port(5432)
         return {
+            "host": "localhost",
             "type": "postgres",
             "user": "root",
             "password": "test",
             "database": "test",
-            "port": exposed_port,
+            "port": self.container.get_exposed_port(5432),
         }
 
     def teardown(self):

Original file line number	Diff line number	Diff line change
`@@ -1,12 +1,12 @@`
`1`		`-from feast.infra.offline_stores.contrib.postgres_offline_store.tests.data_source import (`
`2`		`- PostgreSQLDataSourceCreator,`
	`1`	`+from tests.integration.feature_repos.universal.online_store.postgres import (`
	`2`	`+ PostgresOnlieStoreCreator,`
`3`	`3`	`)`
`4`	`4`	`from tests.integration.feature_repos.integration_test_repo_config import (`
`5`	`5`	`IntegrationTestRepoConfig,`
`6`	`6`	`)`
`7`	`7`
`8`	`8`	`FULL_REPO_CONFIGS = [`
`9`	`9`	`IntegrationTestRepoConfig(`
`10`		`- online_store="postgres", online_store_creator=PostgreSQLDataSourceCreator`
	`10`	`+ online_store="postgres", online_store_creator=PostgresOnlieStoreCreator`
`11`	`11`	`),`
`12`	`12`	`]`