client: fix test

performancecopilot · 990f03559e8cac861e8f1698d20391196373739d · suyash committed Aug 6, 2016 · Aug 6, 2016 · 990f035
1 parent 6409525
commit 990f03559e8cac861e8f1698d20391196373739d
Unified Split

Showing with 5 additions and 0 deletions.

+5 −0 client_test.go
diff --git a/client_test.go b/client_test.go
@@ -61,6 +61,11 @@ func TestTocCountAndLength(t *testing.T) {
 		t.Errorf("expected tocCount to be 3, got %v", c.tocCount())
 	}
 
+	var MetricLength = Metric1Length
+	if c.r.version2 {
+		MetricLength = Metric2Length
+	}
+
 	expectedLength = HeaderLength + 3*TocLength + 1*MetricLength + 1*ValueLength + 1*StringLength
 	if c.Length() != expectedLength {
 		t.Errorf("expected Length to be %v, got %v", expectedLength, c.Length())