Function TestPushUTF8

handler/handler_test.go:440–627

(t *testing.T)

}

func TestPushUTF8(t *testing.T) {
	ValidationScheme = model.UTF8Validation
	EscapingScheme = model.ValueEncodingEscaping
	mms := MockMetricStore{}
	handler := Push(&mms, false, true, false, logger)
	handlerBase64 := Push(&mms, false, true, true, logger)

	// With job name, instance name, UTF-8 escaped label name in params, UTF-8 metric name and text content.
	mms.lastWriteRequest = storage.WriteRequest{}
	req, err := http.NewRequest(
		"POST", "http://example.org/",
		bytes.NewBufferString("some_metric 3.14\n{\"another.metric\",instance=\"testinstance\",job=\"testjob\",\"dotted.label.name\"=\"mylabelvalue\"} 42\n"),
	)
	if err != nil {
		t.Fatal(err)
	}
	w := httptest.NewRecorder()

	params := map[string]string{
		"job":    "testjob",
		"labels": "/instance/testinstance/U__dotted_2e_label_2e_name/mylabelvalue",
	}

	handler(w, req.WithContext(ctxWithParams(params, req)))
	if expected, got := http.StatusOK, w.Code; expected != got {
		t.Errorf("Wanted status code %v, got %v.", expected, got)
	}
	if mms.lastWriteRequest.Timestamp.IsZero() {
		t.Errorf("Write request timestamp not set: %#v", mms.lastWriteRequest)
	}
	if expected, got := "testjob", mms.lastWriteRequest.Labels["job"]; expected != got {
		t.Errorf("Wanted job %v, got %v.", expected, got)
	}
	if expected, got := "testinstance", mms.lastWriteRequest.Labels["instance"]; expected != got {
		t.Errorf("Wanted instance %v, got %v.", expected, got)
	}
	if expected, got := "mylabelvalue", mms.lastWriteRequest.Labels["dotted.label.name"]; expected != got {
		t.Errorf("Wanted dotted.label.name %v, got %v.", expected, got)
	}
	// Note that sanitation hasn't happened yet, grouping labels not in request.
	verifyMetricFamily(t, `name:"some_metric" type:UNTYPED metric:{untyped:{value:3.14}}`, mms.lastWriteRequest.MetricFamilies["some_metric"])
	verifyMetricFamily(t, `name:"another.metric" type:UNTYPED metric:{label:{name:"instance" value:"testinstance"} label:{name:"job" value:"testjob"} label:{name:"dotted.label.name" value:"mylabelvalue"} untyped:{value:42}}`, mms.lastWriteRequest.MetricFamilies["another.metric"])

	// With base64-encoded label values, UTF-8 escaped label name in params, UTF-8 metric name and text content.
	mms.lastWriteRequest = storage.WriteRequest{}
	req, err = http.NewRequest(
		"POST", "http://example.org/",
		bytes.NewBufferString("some_metric 3.14\n{\"another.metric\",instance=\"testinstance\",job=\"testjob\",\"dotted.label.name\"=\"mylabelvalue\"} 42\n"),
	)
	if err != nil {
		t.Fatal(err)
	}
	w = httptest.NewRecorder()
	params = map[string]string{
		"job":    "dGVzdC9qb2I=",                                                                         // job="test/job"
		"labels": "/instance@base64/dGVzdGluc3RhbmNl/U__dotted_2e_label_2e_name@base64/bXlsYWJlbHZhbHVl", // instance="testinstance", dotted.label.name="mylabelvalue"
	}
	handlerBase64(w, req.WithContext(ctxWithParams(params, req)))