github.com/SmartMeshFoundation/Spectrum@v0.0.0-20220621030607-452a266fee1e/core/state/iterator_test.go (about)

     1  // Copyright 2016 The Spectrum Authors
     2  // This file is part of the Spectrum library.
     3  //
     4  // The Spectrum library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The Spectrum library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the Spectrum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package state
    18  
    19  import (
    20  	"bytes"
    21  	"testing"
    22  
    23  	"github.com/SmartMeshFoundation/Spectrum/common"
    24  )
    25  
    26  // Tests that the node iterator indeed walks over the entire database contents.
    27  func TestNodeIteratorCoverage(t *testing.T) {
    28  	// Create some arbitrary test state to iterate
    29  	db, mem, root, _ := makeTestState()
    30  
    31  	state, err := New(root, db)
    32  	if err != nil {
    33  		t.Fatalf("failed to create state trie at %x: %v", root, err)
    34  	}
    35  	// Gather all the node hashes found by the iterator
    36  	hashes := make(map[common.Hash]struct{})
    37  	for it := NewNodeIterator(state); it.Next(); {
    38  		if it.Hash != (common.Hash{}) {
    39  			hashes[it.Hash] = struct{}{}
    40  		}
    41  	}
    42  
    43  	// Cross check the hashes and the database itself
    44  	for hash := range hashes {
    45  		if _, err := mem.Get(hash.Bytes()); err != nil {
    46  			t.Errorf("failed to retrieve reported node %x: %v", hash, err)
    47  		}
    48  	}
    49  	for _, key := range mem.Keys() {
    50  		if bytes.HasPrefix(key, []byte("secure-key-")) {
    51  			continue
    52  		}
    53  		if _, ok := hashes[common.BytesToHash(key)]; !ok {
    54  			t.Errorf("state entry not reported %x", key)
    55  		}
    56  	}
    57  }