Fix tests

This commit is contained in:
Jakob Borg 2015-11-17 21:20:46 +01:00
parent 8b759d0e1e
commit aa853ac833

View File

@ -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)