dummy-consumer.controller.mock.ts

import { Controller, Inject } from '@nestjs/common/decorators';
import { Ctx, MessagePattern, Payload, RmqContext } from '@nestjs/microservices';
import { Channel } from 'amqp-connection-manager';
import { Message } from 'amqplib';

import {
  DUMMY_TOPIC_ACK,
  DUMMY_TOPIC_NOACK,
  DUMMY_WORKER_ID_ACK,
  DUMMY_WORKER_ID_NOACK,
  Resources,
} from './dummy.constants';

@Controller(Resources.DUMMY)
export class DummyConsumerController {
  constructor(@Inject('WORKER_ID') private readonly workerId = 0) {}

  // eslint-disable-next-line @typescript-eslint/no-unused-vars
  emptySpy(args: { data: unknown; workerId?: number }) {
    // leave empty we spy on this to check if controller's method is being called
  }

  @MessagePattern(DUMMY_TOPIC_ACK)
  identityWithAck(@Payload() data: unknown, @Ctx() context: RmqContext) {
    this.emptySpy({ data, workerId: this.workerId });
    const channel = context.getChannelRef() as Channel;
    const originalMessage = context.getMessage() as Message;
    channel.ack(originalMessage);
    return data;
  }

  @MessagePattern(DUMMY_TOPIC_NOACK)
  identityWithNoAck(@Payload() data: unknown) {
    this.emptySpy({ data, workerId: this.workerId });
    return data;
  }

  @MessagePattern(DUMMY_WORKER_ID_ACK)
  getWorkerIdAck(@Payload() data: unknown, @Ctx() context: RmqContext) {
    this.emptySpy({ data, workerId: this.workerId });
    const channel = context.getChannelRef() as Channel;
    const originalMessage = context.getMessage() as Message;
    channel.ack(originalMessage);
    return { workerId: this.workerId };
  }

  @MessagePattern(DUMMY_WORKER_ID_NOACK)
  workerIdNoAck(@Payload() data: unknown) {
    this.emptySpy({ data, workerId: this.workerId });
    return { workerId: this.workerId };
  }
}