Mov (#518)

[bytom/vapor.git] / netsync / chainmgr / block_keeper_test.go

diff --git a/netsync/chainmgr/block_keeper_test.go b/netsync/chainmgr/block_keeper_test.go
index e34e48b..52b206c 100644 (file)
--- a/netsync/chainmgr/block_keeper_test.go
+++ b/netsync/chainmgr/block_keeper_test.go
@@ -448,7 +448,8 @@ func TestLocateBlocks(t *testing.T) {
                        want = append(want, blocks[i])
                }
 
-               got, err := bk.locateBlocks(locator, &c.stopHash)
+               mockTimeout := func() bool { return false }
+               got, err := bk.locateBlocks(locator, &c.stopHash, mockTimeout)
                if err != c.wantErr {
                        t.Errorf("case %d: got %v want err = %v", i, err, c.wantErr)
                }