Fix tracker.Restore test

This commit is contained in:
Roy Crihfield 2024-07-06 22:34:52 +08:00
parent 523a745cf3
commit eac1d8b517
2 changed files with 15 additions and 5 deletions

View File

@ -189,7 +189,7 @@ func (tr *TrackerImpl) Restore(makeIterator iter.IteratorConstructor) (
}
}
// force the lower bound path to an even length (required by geth API/HexToKeyBytes)
// force the lower bound path to an even length (required by NodeIterator constructor)
if len(recoveredPath)&1 == 1 {
// to avoid skipped nodes, we must rewind by one index
recoveredPath = rewindPath(recoveredPath)
@ -246,6 +246,7 @@ func (it *Iterator) Next(descend bool) bool {
return ret
}
// Bounds returns the bounds of the underlying PrefixBoundIterator, if any
func (it *Iterator) Bounds() ([]byte, []byte) {
if impl, ok := it.NodeIterator.(*iter.PrefixBoundIterator); ok {
return impl.Bounds()

View File

@ -25,7 +25,6 @@ func TestTracker(t *testing.T) {
tr := tracker.New(recoveryFile, NumIters)
defer tr.CloseAndSave()
var prevPath []byte
count := 0
nodeit, err := tree.NodeIterator(nil)
if err != nil {
@ -33,9 +32,9 @@ func TestTracker(t *testing.T) {
}
for it := tr.Tracked(nodeit); it.Next(true); {
if count == interrupt {
return prevPath // tracker rewinds one node to prevent gaps
t.Logf("interrupting at: i=%d path=%v", count, it.Path())
return it.Path()
}
prevPath = it.Path()
count++
}
return nil
@ -58,8 +57,18 @@ func TestTracker(t *testing.T) {
if uint(len(its)) != NumIters {
t.Fatalf("expected to restore %d iterators, got %d", NumIters, len(its))
}
if !its[0].Next(true) {
t.Fatal("iterator ends prematurely after restore")
}
if !bytes.Equal(failedAt, its[0].Path()) {
t.Fatalf("iterator restored to wrong position: expected %v, got %v", failedAt, its[0].Path())
// Due to the constraint that NodeIterator can only be initialized with an even-length path,
// we sometimes rewind an extra node when restoring (e.g. [1 2 0] => [1 2]).
if !its[0].Next(true) {
t.Fatal("iterator ends prematurely after restore")
}
if !bytes.Equal(failedAt, its[0].Path()) {
t.Fatalf("iterator restored to wrong position: expected %v, got %v", failedAt, its[0].Path())
}
}
if fileExists(recoveryFile) {