nestjs-hot-shots/test/metrics/collectors/gauge.collector.spec.ts

Hot-shots Module for Nest.js Framework

import { Test } from '@nestjs/testing';
import { StatsD } from 'hot-shots';

import { GaugeCollector, HotShotsModule, MetricsService } from '../../../src';

describe('GaugeCollector', () => {
	let metricsService: MetricsService, statsD: StatsD;

	beforeEach(async () => {
		const moduleRef = await Test.createTestingModule({
			imports: [HotShotsModule.forRoot({ mock: true })]
		}).compile();

		statsD = moduleRef.get(StatsD);
		metricsService = moduleRef.get(MetricsService);
	});

	it('should be defined', () => {
		expect(statsD).toBeDefined();
		expect(metricsService).toBeDefined();
	});

	it('should create GaugeCollector', () => {
		const instance = metricsService.getGauge('test_metric');
		expect(instance).toBeDefined();
		expect(instance).toBeInstanceOf(GaugeCollector);
	});

	it('should set gauge value', () => {
		const instance = metricsService.getGauge('test_metric');
		instance.set(10);

		expect(statsD.mockBuffer[0]).toBe('test_metric:10|g');
	});

	it('should set gauge value with tags', () => {
		const instance = metricsService.getGauge('test_metric');
		instance.set(10, { tag1: 'value1', tag2: 'value2' });

		expect(statsD.mockBuffer[0]).toBe('test_metric:10|g|#tag1:value1,tag2:value2');
	});

	it('should set gauge value with merge tags', () => {
		const instance = metricsService.getGauge('test_metric', { tags: { tag1: 'value1' } });
		instance.set(10, { tag2: 'value2' });

		expect(statsD.mockBuffer[0]).toBe('test_metric:10|g|#tag1:value1,tag2:value2');
	});

	afterEach(() => {
		jest.clearAllMocks();
	});
});