mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2025-01-12 21:37:32 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: Fix telemetry tests (#3804)

Browse source 
* 🧪 Set telemetry env vars

* 🔥 Remove logging

* 🎨 Format with Prettier
This commit is contained in:
Iván Ovejero 2022-08-01 22:37:59 +02:00 committed by GitHub
parent 3de062202d
commit 231cfaa24d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 299 additions and 295 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
packages/cli/test/unit/Telemetry.test.ts
View file

@ -1,4 +1,5 @@
import { Telemetry } from '../../src/telemetry'; import { Telemetry } from '../../src/telemetry';
import config from '../../config';
jest.spyOn(Telemetry.prototype as any, 'createTelemetryClient').mockImplementation(() => { jest.spyOn(Telemetry.prototype as any, 'createTelemetryClient').mockImplementation(() => {
return { return {
@ -18,9 +19,13 @@ describe('Telemetry', () => {
const testDateTime = new Date('2022-01-01 00:00:00'); const testDateTime = new Date('2022-01-01 00:00:00');
beforeAll(() => { beforeAll(() => {
startPulseSpy = jest.spyOn(Telemetry.prototype as any, 'startPulse').mockImplementation(() => {}); startPulseSpy = jest
.spyOn(Telemetry.prototype as any, 'startPulse')
.mockImplementation(() => {});
jest.useFakeTimers(); jest.useFakeTimers();
jest.setSystemTime(testDateTime); jest.setSystemTime(testDateTime);
config.set('diagnostics.enabled', true);
config.set('deployment.type', 'n8n-testing');
}); });
afterAll(() => { afterAll(() => {
@ -56,7 +61,7 @@ describe('Telemetry', () => {
workflow_id: '1', workflow_id: '1',
is_manual: true, is_manual: true,
success: true, success: true,
error_node_type: 'custom-nodes-base.node-type' error_node_type: 'custom-nodes-base.node-type',
}; };
payload.is_manual = true; payload.is_manual = true;
@ -106,7 +111,7 @@ describe('Telemetry', () => {
workflow_id: '1', workflow_id: '1',
is_manual: true, is_manual: true,
success: false, success: false,
error_node_type: 'custom-nodes-base.node-type' error_node_type: 'custom-nodes-base.node-type',
}; };
const execTime1 = fakeJestSystemTime('2022-01-01 12:00:00'); const execTime1 = fakeJestSystemTime('2022-01-01 12:00:00');
@ -141,7 +146,7 @@ describe('Telemetry', () => {
workflow_id: '1', workflow_id: '1',
is_manual: false, is_manual: false,
success: true, success: true,
error_node_type: 'node_type' error_node_type: 'node_type',
}; };
// successful execution // successful execution
@ -266,7 +271,7 @@ describe('Telemetry', () => {
afterEach(() => { afterEach(() => {
pulseSpy.mockClear(); pulseSpy.mockClear();
}) });
test('should trigger pulse in intervals', () => { test('should trigger pulse in intervals', () => {
expect(pulseSpy).toBeCalledTimes(0); expect(pulseSpy).toBeCalledTimes(0);
@ -296,7 +301,7 @@ describe('Telemetry', () => {
workflow_id: '1', workflow_id: '1',
is_manual: true, is_manual: true,
success: true, success: true,
error_node_type: 'custom-nodes-base.node-type' error_node_type: 'custom-nodes-base.node-type',
}; };
await telemetry.trackWorkflowExecution(payload); await telemetry.trackWorkflowExecution(payload);
@ -330,7 +335,6 @@ describe('Telemetry', () => {
expect(pulseSpy).toBeCalledTimes(1); expect(pulseSpy).toBeCalledTimes(1);
expect(spyTrack).toHaveBeenCalledTimes(3); expect(spyTrack).toHaveBeenCalledTimes(3);
console.log(spyTrack.getMockImplementation());
expect(spyTrack).toHaveBeenNthCalledWith(1, 'Workflow execution count', { expect(spyTrack).toHaveBeenNthCalledWith(1, 'Workflow execution count', {
event_version: '2', event_version: '2',
workflow_id: '1', workflow_id: '1',
@ -349,7 +353,7 @@ describe('Telemetry', () => {
prod_success: { prod_success: {
count: 2, count: 2,
first: testDateTime, first: testDateTime,
} },
}); });
expect(spyTrack).toHaveBeenNthCalledWith(2, 'Workflow execution count', { expect(spyTrack).toHaveBeenNthCalledWith(2, 'Workflow execution count', {
event_version: '2', event_version: '2',
@ -357,7 +361,7 @@ describe('Telemetry', () => {
prod_error: { prod_error: {
count: 2, count: 2,
first: testDateTime, first: testDateTime,
} },
}); });
expect(spyTrack).toHaveBeenNthCalledWith(3, 'pulse'); expect(spyTrack).toHaveBeenNthCalledWith(3, 'pulse');
expect(Object.keys(execBuffer).length).toBe(0); expect(Object.keys(execBuffer).length).toBe(0);
@ -378,4 +382,4 @@ const fakeJestSystemTime = (dateTime: string | Date): Date => {
const dt = new Date(dateTime); const dt = new Date(dateTime);
jest.setSystemTime(dt); jest.setSystemTime(dt);
return dt; return dt;
} };