Function TestServer_Metrics

agent/boundarylogproxy/proxy_test.go:679–812

(t *testing.T)

}

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

	makeReq := func(n int) *agentproto.ReportBoundaryLogsRequest {
		logs := make([]*agentproto.BoundaryLog, n)
		for i := range n {
			logs[i] = &agentproto.BoundaryLog{
				Allowed: true,
				Time:    timestamppb.Now(),
				Resource: &agentproto.BoundaryLog_HttpRequest_{
					HttpRequest: &agentproto.BoundaryLog_HttpRequest{
						Method: "GET",
						Url:    "https://example.com",
					},
				},
			}
		}
		return &agentproto.ReportBoundaryLogsRequest{Logs: logs}
	}

	// BufferFull needs its own setup because it intentionally does not run
	// a forwarder so the channel fills up.
	t.Run("BufferFull", func(t *testing.T) {
		t.Parallel()

		reg := prometheus.NewRegistry()
		socketPath := filepath.Join(testutil.TempDirUnixSocket(t), "boundary.sock")
		srv := boundarylogproxy.NewServer(testutil.Logger(t), socketPath, reg)

		err := srv.Start()
		require.NoError(t, err)
		t.Cleanup(func() { require.NoError(t, srv.Close()) })

		conn, err := net.Dial("unix", socketPath)
		require.NoError(t, err)
		defer conn.Close()

		// Fill the buffer (size 100) without running a forwarder so nothing
		// drains. Then send one more to trigger the drop path.
		for range 101 {
			sendLogs(t, conn, makeReq(1))
		}

		require.Eventually(t, func() bool {
			return getCounterVecValue(t, reg, "agent_boundary_log_proxy_batches_dropped_total", "buffer_full") >= 1
		}, testutil.WaitShort, testutil.IntervalFast)
		require.GreaterOrEqual(t,
			getCounterVecValue(t, reg, "agent_boundary_log_proxy_logs_dropped_total", "buffer_full"),
			float64(1))
	})

	// The remaining metrics share one server, forwarder, and connection. The
	// phases run sequentially so metrics accumulate.
	t.Run("Forwarding", func(t *testing.T) {
		t.Parallel()

		reg := prometheus.NewRegistry()
		socketPath := filepath.Join(testutil.TempDirUnixSocket(t), "boundary.sock")