Function TestHTTPRequestVerification

pkg/telemetry/telemetry_test.go:587–685

TestHTTPRequestVerification tests that HTTP requests are made correctly when telemetry is enabled

(t *testing.T)

// TestHTTPRequestVerification tests that HTTP requests are made correctly when telemetry is enabled
func TestHTTPRequestVerification(t *testing.T) {
	t.Parallel()
	logger := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: slog.LevelDebug}))
	mockHTTP := NewMockHTTPClient()

	client := newClient(t.Context(), logger, true, true, "test-version", mockHTTP.Client)

	client.endpoint = "https://test-telemetry.example.com/api/events"
	client.apiKey = "test-api-key"
	client.header = "test-header"

	ctx := t.Context()

	t.Run("CommandEventHTTPRequest", func(t *testing.T) {
		// Reset mock before test
		mockHTTP = NewMockHTTPClient()
		client.httpClient = mockHTTP

		event := &CommandEvent{
			Action:  "run",
			Args:    []string{"config.yaml"},
			Success: true,
		}

		assert.NotEmpty(t, client.endpoint, "Client endpoint should be set for this test")
		assert.NotEmpty(t, client.apiKey, "Client API key should be set for this test")
		assert.True(t, client.enabled, "Client should be enabled for this test")

		t.Logf("Before Track: endpoint=%s, apiKey len=%d, enabled=%t", client.endpoint, len(client.apiKey), client.enabled)

		client.Track(ctx, event)

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

		t.Logf("HTTP requests captured: %d", mockHTTP.GetRequestCount())

		requests := mockHTTP.GetRequests()
		req := requests[0]

		assert.Equal(t, http.MethodPost, req.Method, "Expected POST request")
		assert.Equal(t, "https://test-telemetry.example.com/api/events", req.URL.String(), "Expected correct URL")

		assert.Equal(t, "application/json", req.Header.Get("Content-Type"), "Expected Content-Type application/json")
		assert.Equal(t, "cagent/test-version", req.Header.Get("User-Agent"), "Expected User-Agent cagent/test-version")
		assert.Equal(t, "test-api-key", req.Header.Get("test-header"), "Expected test-header test-api-key")

		bodies := mockHTTP.GetBodies()
		assert.NotEmpty(t, bodies, "Expected request body to be captured")

		var requestBody map[string]any
		require.NoError(t, json.Unmarshal(bodies[0], &requestBody), "Failed to unmarshal request body")

		records, ok := requestBody["records"].([]any)
		require.True(t, ok, "Expected 'records' array in request body")
		assert.Len(t, records, 1, "Expected 1 record")