Browse Source

Fix tests

Jakob Borg 10 years ago
parent
commit
aa853ac833
1 changed files with 5 additions and 4 deletions
  1. 5 4
      lib/scanner/walk_test.go

+ 5 - 4
lib/scanner/walk_test.go

@@ -153,9 +153,10 @@ func TestVerify(t *testing.T) {
 	// data should be an even multiple of blocksize long
 	data := []byte("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut e")
 	buf := bytes.NewBuffer(data)
-	var progress int64
+	progress := newByteCounter()
+	defer progress.Close()
 
-	blocks, err := Blocks(buf, blocksize, 0, &progress)
+	blocks, err := Blocks(buf, blocksize, 0, progress)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -163,8 +164,8 @@ func TestVerify(t *testing.T) {
 		t.Fatalf("Incorrect number of blocks %d != %d", len(blocks), exp)
 	}
 
-	if int64(len(data)) != progress {
-		t.Fatalf("Incorrect counter value %d  != %d", len(data), progress)
+	if int64(len(data)) != progress.Total() {
+		t.Fatalf("Incorrect counter value %d  != %d", len(data), progress.Total())
 	}
 
 	buf = bytes.NewBuffer(data)