Support multiple hash sizes in one build

1 files changed, 8 insertions, 8 deletions

diff --git a/hash_test.go b/hash_test.go
index e09eadfb..4b359c4a 100644
--- a/hash_test.go
+++ b/hash_test.go
@@ -7,24 +7,24 @@ import (
 
 func TestParseHashValidAndInvalid(t *testing.T) {
 	pattern := "0123456789abcdef"
-	repeats := (HashSize*2 + len(pattern) - 1) / len(pattern)
-	hexStr := strings.Repeat(pattern, repeats)[:HashSize*2]
+	repeats := (testHashSize*2 + len(pattern) - 1) / len(pattern)
+	hexStr := strings.Repeat(pattern, repeats)[:testHashSize*2]
 
-	id, err := ParseHash(hexStr)
+	id, err := ParseHashWithSize(hexStr, testHashSize)
 	if err != nil {
 		t.Fatalf("ParseHash returned error: %v", err)
 	}
 
-	if got := id.String(); got != hexStr {
+	if got := id.StringWithSize(testHashSize); got != hexStr {
 		t.Fatalf("unexpected String result: %q", got)
 	}
 
-	if _, err := ParseHash("abcd"); err == nil {
+	if _, err := ParseHashWithSize("abcd", testHashSize); err == nil {
 		t.Fatal("expected error for short hash")
 	}
 
-	badHex := strings.Repeat("z", HashSize*2)
-	if _, err := ParseHash(badHex); err == nil {
+	badHex := strings.Repeat("z", testHashSize*2)
+	if _, err := ParseHashWithSize(badHex, testHashSize); err == nil {
 		t.Fatal("expected error for non-hex input")
 	}
 }
@@ -34,7 +34,7 @@ func TestHashBytesCopiesUnderlyingData(t *testing.T) {
 	for i := range id {
 		id[i] = byte(i)
 	}
-	orig := id.Bytes()
+	orig := id.BytesWithSize(testHashSize)
 	orig[0] ^= 0xff
 	if id[0] == orig[0] {
 		t.Fatal("Bytes should return a copy")