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);
  });
});