test(core): Move unit tests closer to testable components (no-changelog) (#10287)

packages/cli/src/posthog/__tests__/PostHog.test.ts

@@ -0,0 +1,91 @@
+import { PostHog } from 'posthog-node';
+import { InstanceSettings } from 'n8n-core';
+import { PostHogClient } from '@/posthog';
+import config from '@/config';
+import { mockInstance } from '@test/mocking';
+
+jest.mock('posthog-node');
+
+describe('PostHog', () => {
+	const instanceId = 'test-id';
+	const userId = 'distinct-id';
+	const apiKey = 'api-key';
+	const apiHost = 'api-host';
+
+	const instanceSettings = mockInstance(InstanceSettings, { instanceId });
+
+	beforeAll(() => {
+		config.set('diagnostics.config.posthog.apiKey', apiKey);
+		config.set('diagnostics.config.posthog.apiHost', apiHost);
+	});
+
+	beforeEach(() => {
+		config.set('diagnostics.enabled', true);
+		jest.resetAllMocks();
+	});
+
+	it('inits PostHog correctly', async () => {
+		const ph = new PostHogClient(instanceSettings);
+		await ph.init();
+
+		expect(PostHog.prototype.constructor).toHaveBeenCalledWith(apiKey, { host: apiHost });
+	});
+
+	it('does not initialize or track if diagnostics are not enabled', async () => {
+		config.set('diagnostics.enabled', false);
+
+		const ph = new PostHogClient(instanceSettings);
+		await ph.init();
+
+		ph.track({
+			userId: 'test',
+			event: 'test',
+			properties: {},
+		});
+
+		expect(PostHog.prototype.constructor).not.toHaveBeenCalled();
+		expect(PostHog.prototype.capture).not.toHaveBeenCalled();
+	});
+
+	it('captures PostHog events', async () => {
+		const event = 'test event';
+		const properties = {
+			user_id: 'test',
+			test: true,
+		};
+
+		const ph = new PostHogClient(instanceSettings);
+		await ph.init();
+
+		ph.track({
+			userId,
+			event,
+			properties,
+		});
+
+		expect(PostHog.prototype.capture).toHaveBeenCalledWith({
+			distinctId: userId,
+			event,
+			userId,
+			properties,
+			sendFeatureFlags: true,
+		});
+	});
+
+	it('gets feature flags', async () => {
+		const createdAt = new Date();
+		const ph = new PostHogClient(instanceSettings);
+		await ph.init();
+
+		await ph.getFeatureFlags({
+			id: userId,
+			createdAt,
+		});
+
+		expect(PostHog.prototype.getAllFlags).toHaveBeenCalledWith(`${instanceId}#${userId}`, {
+			personProperties: {
+				created_at_timestamp: createdAt.getTime().toString(),
+			},
+		});
+	});
+});