.

aws · Nov 15, 2024 · b162d5b · b162d5b
1 parent 218e585
commit b162d5b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/integration/container/tests/read_write_splitting.test.ts b/tests/integration/container/tests/read_write_splitting.test.ts
@@ -57,7 +57,7 @@ async function initDefaultConfig(host: string, port: number, connectToProxy: boo
   };
 
   if (connectToProxy) {
-    config["clusterInstanceHostPattern"] = "?." + env.proxyDatabaseInfo.instanceEndpointSuffix;
+    config["clusterInstanceHostPattern"] = "?." + env.proxyDatabaseInfo.instanceEndpointSuffix + ":" + env.proxyDatabaseInfo.instanceEndpointPort;
   }
   config = DriverHelper.addDriverSpecificConfiguration(config, env.engine);
   return config;
@@ -78,7 +78,7 @@ async function initConfigWithFailover(host: string, port: number, connectToProxy
   };
 
   if (connectToProxy) {
-    config["clusterInstanceHostPattern"] = "?." + env.proxyDatabaseInfo.instanceEndpointSuffix;
+    config["clusterInstanceHostPattern"] = "?." + env.proxyDatabaseInfo.instanceEndpointSuffix + ":" + env.proxyDatabaseInfo.instanceEndpointPort;
   }
   config = DriverHelper.addDriverSpecificConfiguration(config, env.engine);
   return config;

diff --git a/tests/integration/container/tests/utils/test_environment.ts b/tests/integration/container/tests/utils/test_environment.ts
@@ -114,6 +114,7 @@ export class TestEnvironment {
                 port: info?.databaseInfo.instanceEndpointPort ?? 5432,
                 user: info?.databaseInfo.username,
                 password: info?.databaseInfo.password,
+                database: info?.databaseInfo.default_db_name,
                 query_timeout: 3000,
                 connectionTimeoutMillis: 3000
               });