From 4bff5d29bf1fa151c4d72caa79a90082809b322c Mon Sep 17 00:00:00 2001
From: Fabian Reinartz <fab.reinartz@gmail.com>
Date: Fri, 27 Feb 2015 14:41:43 +0100
Subject: [PATCH] Add tests for retrieving fingerprints for label matchers.

This checks for the basic behaviour of GetFingerprintsForLabelMatchers, that is, whether the different matcher types filter the correct fingerprints and intersections are correct.
---
 storage/local/storage_test.go | 107 ++++++++++++++++++++++++++++++++++
 1 file changed, 107 insertions(+)

diff --git a/storage/local/storage_test.go b/storage/local/storage_test.go
index a30a0a7fe..5c21fbf8e 100644
--- a/storage/local/storage_test.go
+++ b/storage/local/storage_test.go
@@ -30,7 +30,114 @@ import (
 )
 
 func TestGetFingerprintsForLabelMatchers(t *testing.T) {
+	storage, closer := NewTestStorage(t)
+	defer closer.Close()
+
+	samples := make([]*clientmodel.Sample, 100)
+	fingerprints := make(clientmodel.Fingerprints, 100)
+
+	for i := range samples {
+		metric := clientmodel.Metric{
+			clientmodel.MetricNameLabel: clientmodel.LabelValue(fmt.Sprintf("test_metric_%d", i)),
+			"label1":                    clientmodel.LabelValue(fmt.Sprintf("test_%d", i/10)),
+			"label2":                    clientmodel.LabelValue(fmt.Sprintf("test_%d", (i+5)/10)),
+		}
+		samples[i] = &clientmodel.Sample{
+			Metric:    metric,
+			Timestamp: clientmodel.Timestamp(i),
+			Value:     clientmodel.SampleValue(i),
+		}
+		fingerprints[i] = metric.Fingerprint()
+	}
 
+	storage.AppendSamples(samples)
+	storage.WaitForIndexing()
+
+	newMatcher := func(matchType metric.MatchType, name clientmodel.LabelName, value clientmodel.LabelValue) *metric.LabelMatcher {
+		lm, err := metric.NewLabelMatcher(matchType, name, value)
+		if err != nil {
+			t.Fatalf("error creating label matcher: %s", err)
+		}
+		return lm
+	}
+
+	var matcherTests = []struct {
+		matchers metric.LabelMatchers
+		expected clientmodel.Fingerprints
+	}{
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.Equal, "label1", "x")},
+			expected: fingerprints[:0],
+		},
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.Equal, "label1", "test_0")},
+			expected: fingerprints[:10],
+		},
+		{
+			matchers: metric.LabelMatchers{
+				newMatcher(metric.Equal, "label1", "test_0"),
+				newMatcher(metric.Equal, "label2", "test_1"),
+			},
+			expected: fingerprints[5:10],
+		},
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.NotEqual, "label1", "x")},
+			expected: fingerprints,
+		},
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.NotEqual, "label1", "test_0")},
+			expected: fingerprints[10:],
+		},
+		{
+			matchers: metric.LabelMatchers{
+				newMatcher(metric.NotEqual, "label1", "test_0"),
+				newMatcher(metric.NotEqual, "label1", "test_1"),
+				newMatcher(metric.NotEqual, "label1", "test_2"),
+			},
+			expected: fingerprints[30:],
+		},
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.RegexMatch, "label1", `test_[3-5]`)},
+			expected: fingerprints[30:60],
+		},
+		{
+			matchers: metric.LabelMatchers{newMatcher(metric.RegexNoMatch, "label1", `test_[3-5]`)},
+			expected: append(append(clientmodel.Fingerprints{}, fingerprints[:30]...), fingerprints[60:]...),
+		},
+		{
+			matchers: metric.LabelMatchers{
+				newMatcher(metric.RegexMatch, "label1", `test_[3-5]`),
+				newMatcher(metric.RegexMatch, "label2", `test_[4-6]`),
+			},
+			expected: fingerprints[35:60],
+		},
+		{
+			matchers: metric.LabelMatchers{
+				newMatcher(metric.RegexMatch, "label1", `test_[3-5]`),
+				newMatcher(metric.NotEqual, "label2", `test_4`),
+			},
+			expected: append(append(clientmodel.Fingerprints{}, fingerprints[30:35]...), fingerprints[45:60]...),
+		},
+	}
+
+	for _, mt := range matcherTests {
+		resfps := storage.GetFingerprintsForLabelMatchers(mt.matchers)
+		if len(mt.expected) != len(resfps) {
+			t.Fatalf("expected %d matches for %q, found %d", len(mt.expected), mt.matchers, len(resfps))
+		}
+		for _, fp1 := range resfps {
+			found := false
+			for _, fp2 := range mt.expected {
+				if fp1 == fp2 {
+					found = true
+					break
+				}
+			}
+			if !found {
+				t.Errorf("expected fingerprint %s for %q not in result", fp1, mt.matchers)
+			}
+		}
+	}
 }
 
 // TestLoop is just a smoke test for the loop method, if we can switch it on and