undeexcepturi/tests/features/upsert/GH3787-2.test.ts

TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.

import { Entity, PrimaryKey, Property, SimpleLogger } from '@mikro-orm/core';
import { MikroORM } from '@mikro-orm/sqlite';
import { mockLogger } from '../../helpers';

@Entity()
class MyEntity1 {

  @PrimaryKey({ type: 'number' })
  id!: number;

  @Property({ type: 'json' })
  field!: object;

}

@Entity()
class MyEntity2 {

  @PrimaryKey({ type: 'number' })
  id!: number;

  @Property({ type: 'json', columnType: 'json' })
  field!: object;

}

let orm: MikroORM;

beforeAll(async () => {
  orm = await MikroORM.init({
    dbName: ':memory:',
    loggerFactory: options => new SimpleLogger(options),
    entities: [MyEntity1, MyEntity2],
  });
  await orm.schema.refreshDatabase();
});

afterAll(() => orm.close(true));
beforeEach(() => orm.schema.clearDatabase());

test('JSON serialization with upsert', async () => {
  const mock = mockLogger(orm, ['query', 'query-params']);
  const entity1 = orm.em.create(MyEntity1, {
    id: 1,
    field: {
      firstName: 'John',
      lastName: 'Doe',
    },
  });
  await orm.em.upsert(entity1);
  const entity2 = orm.em.create(MyEntity2, {
    id: 1,
    field: {
      firstName: 'Albert',
      lastName: 'Doe',
    },
  });
  await orm.em.upsert(entity2);
  expect(mock.mock.calls).toEqual([
    ['[query] insert into `my_entity1` (`field`, `id`) values (\'{"firstName":"John","lastName":"Doe"}\', 1) on conflict (`id`) do update set `field` = excluded.`field`'],
    ['[query] insert into `my_entity2` (`field`, `id`) values (\'{"firstName":"Albert","lastName":"Doe"}\', 1) on conflict (`id`) do update set `field` = excluded.`field`'],
  ]);
});

test('JSON serialization with upsertMany', async () => {
  const mock = mockLogger(orm, ['query', 'query-params']);
  const entity1 = orm.em.create(MyEntity1, {
    id: 1,
    field: {
      firstName: 'John',
      lastName: 'Doe',
    },
  });
  await orm.em.upsertMany([entity1]);
  const entity2 = orm.em.create(MyEntity2, {
    id: 1,
    field: {
      firstName: 'Albert',
      lastName: 'Doe',
    },
  });
  await orm.em.upsertMany([entity2]);
  expect(mock.mock.calls).toEqual([
    ['[query] insert into `my_entity1` (`field`, `id`) values (\'{"firstName":"John","lastName":"Doe"}\', 1) on conflict (`id`) do update set `field` = excluded.`field` returning `id`'],
    ['[query] insert into `my_entity2` (`field`, `id`) values (\'{"firstName":"Albert","lastName":"Doe"}\', 1) on conflict (`id`) do update set `field` = excluded.`field` returning `id`'],
  ]);
});

test('upsertMany with managed entity calls assign', async () => {
  const mock = mockLogger(orm, ['query', 'query-params']);
  const entity1 = orm.em.create(MyEntity1, {
    id: 1,
    field: {
      firstName: 'John',
      lastName: 'Doe',
    },
  });
  await orm.em.flush();
  (entity1.field as any).firstName = '123';
  await orm.em.upsertMany([entity1]);
  await orm.em.flush();
  expect(mock.mock.calls).toEqual([
    ['[query] begin'],
    [`[query] insert into \`my_entity1\` (\`id\`, \`field\`) values (1, '{"firstName":"John","lastName":"Doe"}')`],
    ['[query] commit'],
    ['[query] begin'],
    [`[query] update \`my_entity1\` set \`field\` = '{"firstName":"123","lastName":"Doe"}' where \`id\` = 1`],
    ['[query] commit'],
  ]);
});

test('validation', async () => {
  await expect(orm.em.upsert(MyEntity1, {})).rejects.toThrow('Unique property value required for upsert, provide one of: id');
  await expect(orm.em.upsertMany(MyEntity1, [{}])).rejects.toThrow('Unique property value required for upsert, provide one of: id');
});