client: fix test

performancecopilot · Aug 6, 2016 · 990f035 · 990f035
1 parent 6409525
commit 990f035
Showing 1 changed file with 5 additions and 0 deletions.
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())