106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package trie
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
)
|
|
|
|
// Internal utility functions to facilitate the tests.
|
|
|
|
type strValue string
|
|
|
|
func (s strValue) Hash() chainhash.Hash {
|
|
return chainhash.DoubleHashH([]byte(s))
|
|
}
|
|
|
|
func dump(prefix Key, value Value) error {
|
|
if value == nil {
|
|
fmt.Printf("[%-8s]\n", prefix)
|
|
return nil
|
|
}
|
|
fmt.Printf("[%-8s] %v\n", prefix, value)
|
|
return nil
|
|
}
|
|
|
|
func buildNode(n *node, pairs []pair) *node {
|
|
for _, val := range pairs {
|
|
update(n, Key(val.k), val.v)
|
|
}
|
|
return n
|
|
}
|
|
|
|
func buildTrie(mt *MerkleTrie, pairs []pair) *MerkleTrie {
|
|
for _, val := range pairs {
|
|
mt.Update(Key(val.k), val.v)
|
|
}
|
|
return mt
|
|
}
|
|
|
|
func buildMap(m map[string]Value, pairs []pair) map[string]Value {
|
|
for _, p := range pairs {
|
|
if p.v == nil {
|
|
delete(m, p.k)
|
|
} else {
|
|
m[p.k] = p.v
|
|
}
|
|
}
|
|
return m
|
|
}
|
|
|
|
func newMap() map[string]Value {
|
|
return map[string]Value{}
|
|
}
|
|
|
|
type pair struct {
|
|
k string
|
|
v Value
|
|
}
|
|
|
|
func pairs1() []pair {
|
|
return []pair{
|
|
{"alex", strValue("lion")},
|
|
{"al", strValue("tiger")},
|
|
{"tess", strValue("dolphin")},
|
|
{"bob", strValue("pig")},
|
|
{"ted", strValue("dog")},
|
|
{"teddy", strValue("bear")},
|
|
{"al", nil},
|
|
{"alex", nil},
|
|
{"bob", strValue("cat")},
|
|
}
|
|
}
|
|
|
|
func prunedNode() *node {
|
|
n := newNode(nil)
|
|
n.links['b'] = newNode(nil)
|
|
n.links['b'].links['o'] = newNode(nil)
|
|
n.links['b'].links['o'].links['b'] = newNode(strValue("cat"))
|
|
n.links['t'] = newNode(nil)
|
|
n.links['t'].links['e'] = newNode(nil)
|
|
n.links['t'].links['e'].links['d'] = newNode(strValue("dog"))
|
|
n.links['t'].links['e'].links['d'].links['d'] = newNode(nil)
|
|
n.links['t'].links['e'].links['d'].links['d'].links['y'] = newNode(strValue("bear"))
|
|
n.links['t'].links['e'].links['s'] = newNode(nil)
|
|
n.links['t'].links['e'].links['s'].links['s'] = newNode(strValue("dolphin"))
|
|
return n
|
|
}
|
|
|
|
func unprunedNode() *node {
|
|
n := newNode(nil)
|
|
n.links['a'] = newNode(nil)
|
|
n.links['a'].links['l'] = newNode(nil)
|
|
n.links['a'].links['l'].links['e'] = newNode(nil)
|
|
n.links['a'].links['l'].links['e'].links['x'] = newNode(nil)
|
|
n.links['b'] = newNode(nil)
|
|
n.links['b'].links['o'] = newNode(nil)
|
|
n.links['b'].links['o'].links['b'] = newNode(strValue("cat"))
|
|
n.links['t'] = newNode(nil)
|
|
n.links['t'].links['e'] = newNode(nil)
|
|
n.links['t'].links['e'].links['d'] = newNode(strValue("dog"))
|
|
n.links['t'].links['e'].links['d'].links['d'] = newNode(nil)
|
|
n.links['t'].links['e'].links['d'].links['d'].links['y'] = newNode(strValue("bear"))
|
|
n.links['t'].links['e'].links['s'] = newNode(nil)
|
|
n.links['t'].links['e'].links['s'].links['s'] = newNode(strValue("dolphin"))
|
|
return n
|
|
}
|