@aimee-blue/ab-service-kit/src/__tests__/setupSockets.test.js

Aimee Blue Service Template

import { setupSockets } from '../setup/sockets';
import { EventEmitter } from 'events';
import { createNoOpBasicLogger } from '../shared';

function mockServer() {
  const server = new EventEmitter();
  return server;
}

function prepareTestArgsAndMocks() {
  const registry = {
    initialize: jest.fn(),
    destroy: jest.fn(),
  };
  return {
    registry,
    server: mockServer(),
    config: {
      shouldUseDefaultEndpoints: false,
    },
    sharedDeps: {
      logger: createNoOpBasicLogger(),
    },
    deps: {
      getRegistry: jest.fn(() => registry),
    },
  };
}

describe('setupSockets', () => {
  it('should initialize registry', async () => {
    const data = prepareTestArgsAndMocks();
    await setupSockets(data.server, data.config, data.sharedDeps, data.deps);

    expect(data.deps.getRegistry).toBeCalledTimes(1);
    expect(data.deps.getRegistry.mock.calls[0][0]).toBe(data.server);
    expect(data.deps.getRegistry.mock.calls[0][1]).toEqual(new Map());
    expect(data.registry.initialize).toBeCalledTimes(1);
  });

  it('should dispose correctly', async () => {
    const data = prepareTestArgsAndMocks();
    const result = await setupSockets(
      data.server,
      data.config,
      data.sharedDeps,
      data.deps
    );

    await result('destroy');

    expect(data.registry.destroy).toBeCalledTimes(1);
  });
});