Function TestSessionTracking

pkg/telemetry/telemetry_test.go:109–146

(t *testing.T)

}

func TestSessionTracking(t *testing.T) {
	t.Parallel()
	logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
	mockHTTP := NewMockHTTPClient()
	client := newClient(t.Context(), logger, true, true, "test-version", mockHTTP.Client)

	client.endpoint = "https://test-session-tracking.com/api"
	client.apiKey = "test-session-key"
	client.header = "test-header"

	ctx := t.Context()

	sessionID := client.RecordSessionStart(ctx, "test-agent", "test-session-id")
	assert.NotEmpty(t, sessionID)

	// Record some activity
	client.RecordToolCall(ctx, "test-tool", "session-id", "agent-name", time.Millisecond, nil)
	client.RecordTokenUsage(ctx, "test-model", 100, 50, 0.5)

	// End session
	client.RecordSessionEnd(ctx)

	// Multiple ends should be safe
	client.RecordSessionEnd(ctx)

	require.Eventually(t, func() bool {
		return mockHTTP.GetRequestCount() > 0
	}, time.Second, 5*time.Millisecond, "Expected HTTP requests to be made for session tracking events")

	requestCount := mockHTTP.GetRequestCount()
	t.Logf("Session tracking HTTP requests captured: %d", requestCount)

	requests := mockHTTP.GetRequests()
	for i, req := range requests {
		assert.Equal(t, http.MethodPost, req.Method, "Request %d: Expected POST method", i)
		assert.Equal(t, "test-session-key", req.Header.Get("test-header"), "Request %d: Expected test-header test-session-key", i)
	}
}

func TestCommandTracking(t *testing.T) {
	t.Parallel()