70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
|
package tracker_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"math/rand"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/cerc-io/eth-iterator-utils/internal"
|
||
|
|
"github.com/cerc-io/eth-iterator-utils/tracker"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTracker(t *testing.T) {
|
||
|
|
NumIters := uint(1)
|
||
|
|
recoveryFile := filepath.Join(t.TempDir(), "tracker_test.csv")
|
||
|
|
|
||
|
|
tree, edb := internal.OpenFixtureTrie(t, 1)
|
||
|
|
t.Cleanup(func() { edb.Close() })
|
||
|
|
|
||
|
|
// traverse trie and trigger error at some intermediate point
|
||
|
|
N := len(internal.FixtureNodePaths)
|
||
|
|
interrupt := rand.Intn(N/2) + N/4
|
||
|
|
failedTraverse := func() []byte {
|
||
|
|
tr := tracker.New(recoveryFile, NumIters)
|
||
|
|
defer tr.CloseAndSave()
|
||
|
|
|
||
|
|
var prevPath []byte
|
||
|
|
count := 0
|
||
|
|
for it := tr.Tracked(tree.NodeIterator(nil)); it.Next(true); {
|
||
|
|
if count == interrupt {
|
||
|
|
return prevPath // tracker rewinds one node to prevent gaps
|
||
|
|
}
|
||
|
|
prevPath = it.Path()
|
||
|
|
count++
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
failedAt := failedTraverse()
|
||
|
|
if failedAt == nil {
|
||
|
|
t.Fatal("traversal wasn't interrupted")
|
||
|
|
}
|
||
|
|
|
||
|
|
if !fileExists(recoveryFile) {
|
||
|
|
t.Fatal("recovery file wasn't created")
|
||
|
|
}
|
||
|
|
|
||
|
|
tr := tracker.New(recoveryFile, NumIters)
|
||
|
|
its, err := tr.Restore(tree.NodeIterator)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if uint(len(its)) != NumIters {
|
||
|
|
t.Fatalf("expected to restore %d iterators, got %d", NumIters, len(its))
|
||
|
|
}
|
||
|
|
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) {
|
||
|
|
t.Fatal("recovery file wasn't removed")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func fileExists(file string) bool {
|
||
|
|
_, err := os.Stat(file)
|
||
|
|
return !os.IsNotExist(err)
|
||
|
|
}
|