ToolJet/server/plugins/datasources/postgresql/index.service.spec.ts

import { Test, TestingModule } from '@nestjs/testing';
import PostgresQueryService from '.';

describe('PostgresQueryService', () => {
  let service: PostgresQueryService;

  beforeAll(async () => {
    const module: TestingModule = await Test.createTestingModule({
      providers: [PostgresQueryService],
    }).compile();

    service = module.get<PostgresQueryService>(PostgresQueryService);
  });

  it('should generate the query for bulk update operation', async () => {
    const queryOptions = { 
        table: 'customers',
        primary_key_column: 'id',
        records: [
          { 
            id: 1,
            name:  'sam',
            email: 'sam@example.com'
          },
          { 
            id: 2,
            name:  'jon',
            email: 'jon@example.com'
          }
        ]

     };

    const builtQuery = await service.buildBulkUpdateQuery(queryOptions);
    const expectedQuery = "UPDATE customers SET name = 'sam',  email = 'sam@example.com',  WHERE id = 1; UPDATE customers SET name = 'jon',  email = 'jon@example.com',  WHERE id = 2;";

    expect(builtQuery).toBe(expectedQuery);
  });

});