Function TestStreamingSends

pkg/ingester/client/cortex_util_test.go:21–145

(t *testing.T)

)

func TestStreamingSends(t *testing.T) {
	tests := map[string]struct {
		streamName string
		serverSend func(grpc.ServerStream) error
		clientRecv func(context.Context, IngesterClient) error
	}{
		"testing SendQueryStream": {
			streamName: "QueryStream",
			serverSend: func(stream grpc.ServerStream) error {
				return SendQueryStream(stream.(Ingester_QueryStreamServer), &QueryStreamResponse{})
			},
			clientRecv: func(ctx context.Context, client IngesterClient) error {
				stream, err := client.QueryStream(ctx, &QueryRequest{})
				require.NoError(t, err)

				// Try to receive the response and assert the error we get is the context.Canceled
				// wrapped within a gRPC error.
				_, err = stream.Recv()
				return err
			},
		},
		"testing SendMetricsForLabelMatchersStream": {
			streamName: "MetricsForLabelMatchersStream",
			serverSend: func(stream grpc.ServerStream) error {
				return SendMetricsForLabelMatchersStream(stream.(Ingester_MetricsForLabelMatchersStreamServer), &MetricsForLabelMatchersStreamResponse{})
			},
			clientRecv: func(ctx context.Context, client IngesterClient) error {
				stream, err := client.MetricsForLabelMatchersStream(ctx, &MetricsForLabelMatchersRequest{})
				require.NoError(t, err)

				// Try to receive the response and assert the error we get is the context.Canceled
				// wrapped within a gRPC error.
				_, err = stream.Recv()
				return err
			},
		},
		"testing LabelNamesStream": {
			streamName: "LabelNamesStream",
			serverSend: func(stream grpc.ServerStream) error {
				return SendLabelNamesStream(stream.(Ingester_LabelNamesStreamServer), &LabelNamesStreamResponse{})
			},
			clientRecv: func(ctx context.Context, client IngesterClient) error {
				stream, err := client.LabelNamesStream(ctx, &LabelNamesRequest{})
				require.NoError(t, err)

				// Try to receive the response and assert the error we get is the context.Canceled
				// wrapped within a gRPC error.
				_, err = stream.Recv()
				return err
			},
		},
		"testing LabelValuesStream": {
			streamName: "LabelValuesStream",
			serverSend: func(stream grpc.ServerStream) error {
				return SendLabelValuesStream(stream.(Ingester_LabelValuesStreamServer), &LabelValuesStreamResponse{})
			},
			clientRecv: func(ctx context.Context, client IngesterClient) error {
				stream, err := client.LabelValuesStream(ctx, &LabelValuesRequest{})