aboutsummaryrefslogtreecommitdiffstats
path: root/state/state_test.go
blob: 825d21fcc5504d70500018b73de34c74b6e16518 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package state

import (
    checker "gopkg.in/check.v1"

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

type StateSuite struct {
    state *State
}

var _ = checker.Suite(&StateSuite{})

// var ZeroHash256 = make([]byte, 32)

func (s *StateSuite) TestDump(c *checker.C) {
    key := []byte{0x01}
    value := "foo"
    node := []interface{}{key, value}
    s.state.Trie.Put(node)
    dump := s.state.Dump()
    c.Assert(dump, checker.NotNil)
}

func (s *StateSuite) SetUpTest(c *checker.C) {
    db, _ := ethdb.NewMemDatabase()
    ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "")
    ethutil.Config.Db = db
    s.state = New(trie.New(db, ""))
}

func (s *StateSuite) TestSnapshot(c *checker.C) {
    stateobjaddr := []byte("aa")
    storageaddr := ethutil.Big("0")
    data1 := ethutil.NewValue(42)
    data2 := ethutil.NewValue(43)

    // get state object
    stateObject := s.state.GetOrNewStateObject(stateobjaddr)
    // set inital state object value
    stateObject.SetStorage(storageaddr, data1)
    // get snapshot of current state
    snapshot := s.state.Copy()

    // get state object. is this strictly necessary?
    stateObject = s.state.GetStateObject(stateobjaddr)
    // set new state object value
    stateObject.SetStorage(storageaddr, data2)
    // restore snapshot
    s.state.Set(snapshot)

    // get state object
    stateObject = s.state.GetStateObject(stateobjaddr)
    // get state storage value
    res := stateObject.GetStorage(storageaddr)

    c.Assert(data1, checker.DeepEquals, res)
}