Function TestHandler_remoteWrite

pkg/util/push/push_test.go:720–855

(t *testing.T)

}

func TestHandler_remoteWrite(t *testing.T) {
	var limits validation.Limits
	flagext.DefaultValues(&limits)
	overrides := validation.NewOverrides(limits, nil)

	tests := []struct {
		name           string
		createBody     func() ([]byte, bool) // returns (bodyBytes, isV2)
		expectedStatus int
		expectedBody   string
		verifyResponse func(resp *httptest.ResponseRecorder)
	}{
		{
			name: "remote write v1",
			createBody: func() ([]byte, bool) {
				return createPrometheusRemoteWriteProtobuf(t), false
			},
			expectedStatus: http.StatusOK,
		},
		{
			name: "remote write v2",
			createBody: func() ([]byte, bool) {
				return createPrometheusRemoteWriteV2Protobuf(t), true
			},
			expectedStatus: http.StatusNoContent,
			verifyResponse: func(resp *httptest.ResponseRecorder) {
				respHeader := resp.Header()
				assert.Equal(t, "1", respHeader[rw20WrittenSamplesHeader][0])
				assert.Equal(t, "1", respHeader[rw20WrittenHistogramsHeader][0])
				assert.Equal(t, "1", respHeader[rw20WrittenExemplarsHeader][0])
			},
		},
		{
			name: "remote write v2 with empty samples and histograms should return 400",
			createBody: func() ([]byte, bool) {
				// Create a request with a TimeSeries that has no samples and no histograms
				reqProto := writev2.Request{
					Symbols: []string{"", "__name__", "foo"},
					Timeseries: []writev2.TimeSeries{
						{
							LabelsRefs: []uint32{1, 2},
							Exemplars: []writev2.Exemplar{
								{
									LabelsRefs: []uint32{},
									Value:      1.0,
									Timestamp:  time.Now().UnixMilli(),
								},
							},
						},
					},
				}
				reqBytes, err := reqProto.Marshal()
				require.NoError(t, err)
				return reqBytes, true
			},
			expectedStatus: http.StatusBadRequest,
			expectedBody:   "TimeSeries must contain at least one sample or histogram for series {__name__=\"foo\"}",
		},