Clean up testing code

near · Nov 18, 2023 · a9d42a8 · a9d42a8
1 parent 9b1c397
commit a9d42a8
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/runner/src/dml-handler/dml-handler.ts b/runner/src/dml-handler/dml-handler.ts
@@ -14,7 +14,6 @@ export default class DmlHandler {
     hasuraClient: HasuraClient = new HasuraClient(),
     PgClient = PgClientModule
   ): Promise<DmlHandler> {
-    console.log('CALL MADE');
     const connectionParameters = await hasuraClient.getDbConnectionParameters(account);
     const pgClient = new PgClient({
       user: connectionParameters.username,

diff --git a/runner/src/indexer/indexer.test.ts b/runner/src/indexer/indexer.test.ts
@@ -470,7 +470,7 @@ CREATE TABLE
     expect(result.length).toEqual(2);
   });
 
-  test('indexer builds context and does simultaneous inserts', async () => {
+  test('indexer builds context and does simultaneous upserts', async () => {
     const mockDmlHandler: any = {
       create: jest.fn().mockImplementation(() => {
         return { upsert: jest.fn().mockReturnValue([{ colA: 'valA' }, { colA: 'valA' }]) };
@@ -500,7 +500,7 @@ CREATE TABLE
     }
     await Promise.all(promises);
     expect(mockDmlHandler.create).toHaveBeenCalledTimes(1);
-  }, 100000);
+  });
 
   test('indexer builds context and selects objects from existing table', async () => {
     const selectFn = jest.fn();

diff --git a/runner/src/indexer/indexer.ts b/runner/src/indexer/indexer.ts
@@ -294,7 +294,7 @@ export default class Indexer {
 
         return {
           ...prev,
-          ...funcForTable,
+          ...funcForTable
         };
       }, {});
       return result;

diff --git a/runner/src/pg-client.ts b/runner/src/pg-client.ts
@@ -12,11 +12,10 @@ interface ConnectionParams {
 export default class PgClient {
   private readonly pgPool: Pool;
   public format: typeof pgFormatModule;
-  sleep = async (ms: number): Promise<void> => { await new Promise((resolve) => setTimeout(resolve, ms)); };
 
   constructor (
     connectionParams: ConnectionParams,
-    poolConfig: PoolConfig = { max: 2, idleTimeoutMillis: 30000 },
+    poolConfig: PoolConfig = { max: 10, idleTimeoutMillis: 30000 },
     PgPool: typeof Pool = Pool,
     pgFormat: typeof pgFormatModule = pgFormatModule
   ) {
@@ -32,6 +31,7 @@ export default class PgClient {
   }
 
   async query<R extends QueryResultRow = any>(query: string, params: any[] = []): Promise<QueryResult<R>> {
+    // Automatically manages client connections to pool
     return await this.pgPool.query<R>(query, params);
   }
 }