test: fix pathdb unit test

bnb-chain · Sep 11, 2024 · d5a0b88 · d5a0b88
1 parent e381d88
commit d5a0b88
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/triedb/pathdb/nodebufferlist.go b/triedb/pathdb/nodebufferlist.go
@@ -529,6 +529,10 @@ func (nf *nodebufferlist) getAllNodes() map[common.Hash]map[string]*trienode.Nod
 	defer nf.mux.Unlock()
 	defer nf.baseMux.Unlock()
 
+	if nf.useBase.Load() {
+		return nf.base.nodes
+	}
+
 	nc := newMultiDifflayer(nf.limit, 0, common.Hash{}, make(map[common.Hash]map[string]*trienode.Node), 0)
 	if err := nc.commit(nf.base.root, nf.base.id, nf.base.block, nf.layers, nf.base.nodes); err != nil {
 		log.Crit("failed to commit nodes to node buffer", "error", err)