aboutsummaryrefslogtreecommitdiffstats
path: root/tests/helper/trie.go
diff options
context:
space:
mode:
Diffstat (limited to 'tests/helper/trie.go')
-rw-r--r--tests/helper/trie.go31
1 files changed, 31 insertions, 0 deletions
diff --git a/tests/helper/trie.go b/tests/helper/trie.go
new file mode 100644
index 000000000..119bac713
--- /dev/null
+++ b/tests/helper/trie.go
@@ -0,0 +1,31 @@
+package helper
+
+import "github.com/ethereum/go-ethereum/ethtrie"
+
+type MemDatabase struct {
+ db map[string][]byte
+}
+
+func NewMemDatabase() (*MemDatabase, error) {
+ db := &MemDatabase{db: make(map[string][]byte)}
+ return db, nil
+}
+func (db *MemDatabase) Put(key []byte, value []byte) {
+ db.db[string(key)] = value
+}
+func (db *MemDatabase) Get(key []byte) ([]byte, error) {
+ return db.db[string(key)], nil
+}
+func (db *MemDatabase) Delete(key []byte) error {
+ delete(db.db, string(key))
+ return nil
+}
+func (db *MemDatabase) Print() {}
+func (db *MemDatabase) Close() {}
+func (db *MemDatabase) LastKnownTD() []byte { return nil }
+
+func NewTrie() *ethtrie.Trie {
+ db, _ := NewMemDatabase()
+
+ return ethtrie.New(db, "")
+}