Function TestGrpcStatsMaxStreamsDisabled

middleware/grpc_stats_test.go:266–307

(t *testing.T)

}

func TestGrpcStatsMaxStreamsDisabled(t *testing.T) {
	reg := prometheus.NewRegistry()

	received := promauto.With(reg).NewHistogramVec(prometheus.HistogramOpts{
		Name:    "received_payload_bytes",
		Help:    "Size of received gRPC messages",
		Buckets: BodySizeBuckets,
	}, []string{"method", "route"})

	sent := promauto.With(reg).NewHistogramVec(prometheus.HistogramOpts{
		Name:    "sent_payload_bytes",
		Help:    "Size of sent gRPC",
		Buckets: BodySizeBuckets,
	}, []string{"method", "route"})

	inflightRequests := promauto.With(reg).NewGaugeVec(prometheus.GaugeOpts{
		Name: "inflight_requests",
		Help: "Current number of inflight requests.",
	}, []string{"method", "route"})

	stats := NewStatsHandler(reg, received, sent, inflightRequests, false)

	serv := grpc.NewServer(grpc.StatsHandler(stats), grpc.MaxRecvMsgSize(10e6))
	defer serv.GracefulStop()

	listener, err := net.Listen("tcp", "localhost:0")
	require.NoError(t, err)

	middleware_test.RegisterEchoServerServer(serv, &halfEcho{log: t.Log})

	go func() {
		require.NoError(t, serv.Serve(listener))
	}()

	ctx, cancel1 := context.WithCancel(context.Background())
	defer cancel1()
	conn1 := launchConnWithStreams(t, ctx, listener, 10)
	defer conn1.Close()

	err = testutil.GatherAndCompare(reg, bytes.NewBufferString(``), "grpc_concurrent_streams_by_conn_max")
	assert.NoError(t, err)
}

func TestGrpcStatsMaxStreams(t *testing.T) {
	const (