huridocs/uwazi

import { testingEnvironment } from 'api/utils/testingEnvironment';
import { DefaultTestingQueueAdapter } from 'api/queue.v2/configuration/factories';
import { NamespacedDispatcher } from 'api/queue.v2/infrastructure/NamespacedDispatcher';
import { UpdateTemplateRelationshipPropertiesJob } from '../UpdateTemplateRelationshipPropertiesJob';
import { UpdateRelationshipPropertiesJob } from '../UpdateRelationshipPropertiesJob';
import { QueuedRelationshipPropertyUpdateStrategy } from '../QueuedRelationshipPropertyUpdateStrategy';

beforeEach(async () => {
  await testingEnvironment.setUp({});
});

afterAll(async () => {
  await testingEnvironment.tearDown();
});

it('should enqueue a job per entity', async () => {
  const adapter = DefaultTestingQueueAdapter();
  const dispatcher = new NamespacedDispatcher('namespace', 'jobs', adapter);
  const strategy = new QueuedRelationshipPropertyUpdateStrategy(dispatcher);

  await strategy.update(['sharedId1', 'sharedId2']);
  const enqueued1 = await adapter.pickJob('jobs');
  const enqueued2 = await adapter.pickJob('jobs');
  const enqueued3 = await adapter.pickJob('jobs');

  expect(enqueued1!.name).toBe(UpdateRelationshipPropertiesJob.name);
  expect(enqueued1!.params.entityIds).toEqual(['sharedId1']);

  expect(enqueued2!.name).toBe(UpdateRelationshipPropertiesJob.name);
  expect(enqueued2!.params.entityIds).toEqual(['sharedId2']);

  expect(enqueued3).toBe(null);
});

it('should enqueue a job for the template', async () => {
  const adapter = DefaultTestingQueueAdapter();
  const dispatcher = new NamespacedDispatcher('namespace', 'jobs', adapter);
  const strategy = new QueuedRelationshipPropertyUpdateStrategy(dispatcher);

  await strategy.updateByTemplate('template1');
  const enqueued1 = await adapter.pickJob('jobs');
  const enqueued2 = await adapter.pickJob('jobs');

  expect(enqueued1!.name).toBe(UpdateTemplateRelationshipPropertiesJob.name);
  expect(enqueued1!.params.templateId).toBe('template1');

  expect(enqueued2).toBe(null);
});