path: root/ptrie/trie_test.go

package ptrie

import (
    "bytes"
    "fmt"
    "testing"

    "github.com/ethereum/go-ethereum/ethutil"
)

func TestInsert(t *testing.T) {
    trie := NewEmpty()

    trie.UpdateString("doe", "reindeer")
    trie.UpdateString("dog", "puppy")
    trie.UpdateString("dogglesworth", "cat")

    exp := ethutil.Hex2Bytes("8aad789dff2f538bca5d8ea56e8abe10f4c7ba3a5dea95fea4cd6e7c3a1168d3")
    root := trie.Hash()
    if !bytes.Equal(root, exp) {
        t.Errorf("exp %x got %x", exp, root)
    }

    trie = NewEmpty()
    trie.UpdateString("A", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")

    exp = ethutil.Hex2Bytes("d23786fb4a010da3ce639d66d5e904a11dbc02746d1ce25029e53290cabf28ab")
    root = trie.Hash()
    if !bytes.Equal(root, exp) {
        t.Errorf("exp %x got %x", exp, root)
    }
}

func TestGet(t *testing.T) {
    trie := NewEmpty()

    trie.UpdateString("doe", "reindeer")
    trie.UpdateString("dog", "puppy")
    trie.UpdateString("dogglesworth", "cat")

    res := trie.GetString("dog")
    if !bytes.Equal(res, []byte("puppy")) {
        t.Errorf("expected puppy got %x", res)
    }

    unknown := trie.GetString("unknown")
    if unknown != nil {
        t.Errorf("expected nil got %x", unknown)
    }
}

func TestDelete(t *testing.T) {
    trie := NewEmpty()

    vals := []struct{ k, v string }{
        {"do", "verb"},
        {"ether", "wookiedoo"},
        {"horse", "stallion"},
        {"shaman", "horse"},
        {"doge", "coin"},
        {"ether", ""},
        {"dog", "puppy"},
        {"shaman", ""},
    }
    for _, val := range vals {
        trie.UpdateString(val.k, val.v)
    }

    hash := trie.Hash()
    exp := ethutil.Hex2Bytes("5991bb8c6514148a29db676a14ac506cd2cd5775ace63c30a4fe457715e9ac84")
    if !bytes.Equal(hash, exp) {
        t.Errorf("expected %x got %x", exp, hash)
    }
}

func TestReplication(t *testing.T) {
    trie := NewEmpty()
    vals := []struct{ k, v string }{
        {"do", "verb"},
        {"ether", "wookiedoo"},
        {"horse", "stallion"},
        {"shaman", "horse"},
        {"doge", "coin"},
        {"ether", ""},
        {"dog", "puppy"},
        {"shaman", ""},
        {"somethingveryoddindeedthis is", "myothernodedata"},
    }
    for _, val := range vals {
        trie.UpdateString(val.k, val.v)
    }
    trie.Hash()

    trie2 := New(trie.roothash, trie.backend)
    if string(trie2.GetString("horse")) != "stallion" {
        t.Error("expected to have harse => stallion")
    }

    hash := trie2.Hash()
    exp := trie.Hash()
    if !bytes.Equal(hash, exp) {
        t.Errorf("root failure. expected %x got %x", exp, hash)
    }

}

func BenchmarkGets(b *testing.B) {
    trie := NewEmpty()
    vals := []struct{ k, v string }{
        {"do", "verb"},
        {"ether", "wookiedoo"},
        {"horse", "stallion"},
        {"shaman", "horse"},
        {"doge", "coin"},
        {"ether", ""},
        {"dog", "puppy"},
        {"shaman", ""},
        {"somethingveryoddindeedthis is", "myothernodedata"},
    }
    for _, val := range vals {
        trie.UpdateString(val.k, val.v)
    }

    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        trie.Get([]byte("horse"))
    }
}

func BenchmarkUpdate(b *testing.B) {
    trie := NewEmpty()

    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        trie.UpdateString(fmt.Sprintf("aaaaaaaaa%d", base, i), "value")
    }
    trie.Hash()
}

// Not actual test
func TestOutput(t *testing.T) {
    base := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    trie := NewEmpty()
    for i := 0; i < 50; i++ {
        trie.UpdateString(fmt.Sprintf("%s%d", base, i), "valueeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee")
    }
    trie.Hash()
    fmt.Println("############################## FULL ################################")
    fmt.Println(trie.root)

    trie2 := New(trie.roothash, trie.backend)
    trie2.GetString(base + "20")
    fmt.Println("############################## SMALL ################################")
    fmt.Println(trie2.root)
}