test: update unit and integration tests

aws · Oct 29, 2024 · cf3ba4e · cf3ba4e
1 parent 4f93e11
commit cf3ba4e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/pg/lib/client.ts b/pg/lib/client.ts
@@ -62,6 +62,9 @@ export class AwsPGClient extends AwsClient {
         throw new AwsWrapperError(Messages.get("HostInfo.noHostParameter"));
       }
       const result: ClientWrapper = await this.pluginManager.connect(hostInfo, this.properties, true);
+      result.client.on("error", (error: any) => {
+        this.emit("error", error);
+      });
       await this.pluginService.setCurrentClient(result, result.hostInfo);
       await this.internalPostConnect();
     });

diff --git a/pg/lib/dialect/node_postgres_driver_dialect.ts b/pg/lib/dialect/node_postgres_driver_dialect.ts
@@ -33,9 +33,6 @@ export class NodePostgresDriverDialect implements DriverDialect {
 
   async connect(hostInfo: HostInfo, props: Map<string, any>): Promise<ClientWrapper> {
     const targetClient = new Client(WrapperProperties.removeWrapperProperties(props));
-    targetClient.on("error", (error: any) => {
-      // targetClient.emit("error", error);
-    });
     await targetClient.connect();
     return Promise.resolve(new PgClientWrapper(targetClient, hostInfo, props));
   }

diff --git a/pg/lib/pg_pool_client.ts b/pg/lib/pg_pool_client.ts
@@ -18,6 +18,7 @@ import { Pool, PoolClient, PoolConfig } from "pg";
 import { AwsPoolClient } from "../../common/lib/aws_pool_client";
 import { Messages } from "../../common/lib/utils/messages";
 import { AwsWrapperError } from "../../common/lib/utils/errors";
+import { logger } from "../../common/logutils";
 
 export class AwsPgPoolClient implements AwsPoolClient {
   targetPool: Pool;
@@ -26,7 +27,7 @@ export class AwsPgPoolClient implements AwsPoolClient {
     this.targetPool = new Pool(props);
     this.targetPool.on("connect", (_client: PoolClient) => {
       _client.on("error", (err: Error) => {
-        console.log(err);
+        logger.debug(err);
       });
     });
   }