Zoo-Code/packages/telemetry/src/__tests__/TelemetryService.task-completed.test.ts at 39ab05453072fc27a0fe121fbdf10c83aee9dfe1 · Zoo-Code-Org/Zoo-Code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// pnpm --filter @roo-code/telemetry test src/__tests__/TelemetryService.task-completed.test.ts

import { TelemetryEventName, type TelemetryClient } from "@roo-code/types"

import { TelemetryService } from "../TelemetryService"

describe("TelemetryService.captureTaskCompleted", () => {
	let mockClient: TelemetryClient

	beforeEach(() => {
		mockClient = {
			setProvider: vi.fn(),
			capture: vi.fn().mockResolvedValue(undefined),
			captureException: vi.fn().mockResolvedValue(undefined),
			updateTelemetryState: vi.fn(),
			isTelemetryEnabled: vi.fn().mockReturnValue(true),
			shutdown: vi.fn().mockResolvedValue(undefined),
		}
	})

	it("captures Task Completed with the taskId and a default 'attempt_completion' completionReason when no summary is provided", () => {
		const service = new TelemetryService([mockClient])

		service.captureTaskCompleted("task_1")

		expect(mockClient.capture).toHaveBeenCalledWith({
			event: TelemetryEventName.TASK_COMPLETED,
			properties: { taskId: "task_1", completionReason: "attempt_completion" },
		})
	})

	it("includes toolsUsed and messageCount summaries when provided", () => {
		const service = new TelemetryService([mockClient])

		service.captureTaskCompleted(
			"task_1",
			{ read_file: { attempts: 3, failures: 0 }, apply_diff: { attempts: 1, failures: 1 } },
			{ user: 4, assistant: 5 },
		)

		expect(mockClient.capture).toHaveBeenCalledWith({
			event: TelemetryEventName.TASK_COMPLETED,
			properties: {
				taskId: "task_1",
				completionReason: "attempt_completion",
				toolsUsed: { read_file: { attempts: 3, failures: 0 }, apply_diff: { attempts: 1, failures: 1 } },
				messageCount: { user: 4, assistant: 5 },
			},
		})
	})

	it("includes the given completionReason for idle/shutdown installments", () => {
		const service = new TelemetryService([mockClient])

		service.captureTaskCompleted("task_1", { read_file: { attempts: 1, failures: 0 } }, undefined, "idle")

		expect(mockClient.capture).toHaveBeenCalledWith({
			event: TelemetryEventName.TASK_COMPLETED,
			properties: {
				taskId: "task_1",
				completionReason: "idle",
				toolsUsed: { read_file: { attempts: 1, failures: 0 } },
			},
		})
	})
})