(b *testing.B)
| 119 | } |
| 120 | |
| 121 | func BenchmarkGrpcCalls(b *testing.B) { |
| 122 | cfg := server.Config{} |
| 123 | (&cfg).RegisterFlags(flag.NewFlagSet("fake", flag.ContinueOnError)) |
| 124 | register := func(s *grpc.Server) { |
| 125 | d := &mockGprcServer{} |
| 126 | ingester_client.RegisterIngesterServer(s, d) |
| 127 | } |
| 128 | run(b, cfg, register, func(t testing.TB, conn *grpc.ClientConn) { |
| 129 | ctx := context.Background() |
| 130 | ctx = metadata.NewOutgoingContext(ctx, metadata.MD{"i": []string{strconv.Itoa(0)}}) |
| 131 | client := ingester_client.NewIngesterClient(conn) |
| 132 | b.ReportAllocs() |
| 133 | b.ResetTimer() |
| 134 | for i := 0; i < b.N; i++ { |
| 135 | s, err := client.QueryStream(ctx, &ingester_client.QueryRequest{}) |
| 136 | require.NoError(t, err) |
| 137 | _, err = s.Recv() |
| 138 | require.NoError(t, err) |
| 139 | } |
| 140 | }) |
| 141 | } |
| 142 | |
| 143 | func TestConcurrentGrpcCalls(t *testing.T) { |
| 144 | cfg := server.Config{} |
nothing calls this directly
no test coverage detected