Function TestReplace

storage/diskmetricstore_test.go:1262–1392

(t *testing.T)

}

func TestReplace(t *testing.T) {
	dms := NewDiskMetricStore("", 100*time.Millisecond, nil, logger)

	// First do an invalid push to set pushFailedTimestamp and to later
	// verify that it is retained and not replaced.
	ts1 := time.Now()
	grouping1 := map[string]string{
		"job": "job1",
	}
	errCh := make(chan error, 1)
	dms.SubmitWriteRequest(WriteRequest{
		Labels:         grouping1,
		Timestamp:      ts1,
		MetricFamilies: testutil.MetricFamiliesMap(mf1ts),
		Done:           errCh,
	})
	var err error
	for err = range errCh {
		if err != errTimestamp {
			t.Errorf("Expected error %q, got %q.", errTimestamp, err)
		}
	}
	if err == nil {
		t.Error("Expected error on pushing metric with timestamp.")
	}
	pushTimestamp := newPushTimestampGauge(grouping1, time.Time{})
	pushFailedTimestamp := newPushFailedTimestampGauge(grouping1, ts1)
	if err := checkMetricFamilies(
		dms,
		pushTimestamp, pushFailedTimestamp,
	); err != nil {
		t.Error(err)
	}

	// Now a valid update in replace mode. It doesn't replace anything, but
	// it already tests that the push-failed timestamp is retained.
	ts2 := ts1.Add(time.Second)
	errCh = make(chan error, 1)
	dms.SubmitWriteRequest(WriteRequest{
		Labels:         grouping1,
		Timestamp:      ts2,
		MetricFamilies: testutil.MetricFamiliesMap(mf1a),
		Done:           errCh,
		Replace:        true,
	})
	for err := range errCh {
		t.Fatal("Unexpected error:", err)
	}
	pushTimestamp = newPushTimestampGauge(grouping1, ts2)
	if err := checkMetricFamilies(
		dms, mf1a,
		pushTimestamp, pushFailedTimestamp,
	); err != nil {
		t.Error(err)
	}

	// Now push something else in replace mode that should replace mf1.
	ts3 := ts2.Add(time.Second)