Init unit tests for telemetry

packages/cli/src/InternalHooks.ts

@@ -9,7 +9,7 @@ import {
 	IWorkflowBase,
 	IWorkflowDb,
 } from '.';
-import { Telemetry } from './telemetry';
+import Telemetry from './telemetry';
 
 export class InternalHooksClass implements IInternalHooksClass {
 	private versionCli: string;

packages/cli/src/InternalHooksManager.ts

@@ -1,7 +1,7 @@
 /* eslint-disable import/no-cycle */
 import { INodeTypes } from 'n8n-workflow';
 import { InternalHooksClass } from './InternalHooks';
-import { Telemetry } from './telemetry';
+import Telemetry from './telemetry';
 
 export class InternalHooksManager {
 	private static internalHooksInstance: InternalHooksClass;

packages/cli/src/telemetry/index.ts

@@ -35,7 +35,8 @@ interface IExecutionsBuffer {
 	firstExecutions: IFirstExecutions;
 }
 
-export class Telemetry {
+// eslint-disable-next-line import/no-default-export
+export default class Telemetry {
 	private client?: TelemetryClient;
 
 	private instanceId: string;
@@ -71,14 +72,26 @@ export class Telemetry {
 				return;
 			}
 
-			this.client = new TelemetryClient(key, url, { logLevel });
+			this.client = this.createTelemetryClient(key, url, logLevel);
 
-			this.pulseIntervalReference = setInterval(async () => {
-				void this.pulse();
-			}, 6 * 60 * 60 * 1000); // every 6 hours
+			this.startPulse();
 		}
 	}
 
+	private createTelemetryClient(
+		key: string,
+		url: string,
+		logLevel: string,
+	): TelemetryClient | undefined {
+		return new TelemetryClient(key, url, { logLevel });
+	}
+
+	private startPulse() {
+		this.pulseIntervalReference = setInterval(async () => {
+			void this.pulse();
+		}, 6 * 60 * 60 * 1000); // every 6 hours
+	}
+
 	private async pulse(): Promise<unknown> {
 		if (!this.client) {
 			return Promise.resolve();
@@ -104,6 +117,10 @@ export class Telemetry {
 		return Promise.all(allPromises);
 	}
 
+	getTelemetryClient(): unknown {
+		return this.client;
+	}
+
 	async trackWorkflowExecution(properties: IDataObject): Promise<void> {
 		if (this.client) {
 			const workflowId = properties.workflow_id as string;

packages/cli/test/integration/shared/utils.ts

@@ -45,7 +45,6 @@ import { credentialsController } from '../../../src/api/credentials.api';
 import { loadPublicApiVersions } from '../../../src/PublicApi/';
 import type { User } from '../../../src/databases/entities/User';
 import type { ApiPath, EndpointGroup, PostgresSchemaSection, SmtpTestAccount } from './types';
-import { Telemetry } from '../../../src/telemetry';
 import type { N8nApp } from '../../../src/UserManagement/Interfaces';
 import { set } from 'lodash';
 interface TriggerTime {

packages/cli/test/unit/Telemetry.test.ts

@@ -0,0 +1,24 @@
+import Telemetry from '../../src/telemetry';
+
+describe('Telemetry', () => {
+    jest.spyOn(Telemetry.prototype as any, 'createTelemetryClient').mockImplementation(() => {
+        return {
+            flush: () => {},
+            identify: () => {},
+            track: () => {},
+        };
+    });
+
+    jest.spyOn(Telemetry.prototype as any, 'startPulse').mockImplementation(() => {});
+
+    const spyTrack = jest.spyOn(Telemetry.prototype, 'track');
+
+	const telemetry = new Telemetry('Telemetry unit test', '0.0.0');
+
+	describe('trackN8nStop', () => {
+		test('should call track method', () => {
+			telemetry.trackN8nStop();
+            expect(spyTrack).toHaveBeenCalledTimes(1);
+		});
+	});
+});