61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
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/ptrie"
|
|
)
|
|
|
|
type StateSuite struct {
|
|
state *StateDB
|
|
}
|
|
|
|
var _ = checker.Suite(&StateSuite{})
|
|
|
|
// var ZeroHash256 = make([]byte, 32)
|
|
|
|
func (s *StateSuite) TestDump(c *checker.C) {
|
|
key := []byte{0x01}
|
|
value := []byte("foo")
|
|
s.state.trie.Update(key, value)
|
|
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(ptrie.New(nil, 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)
|
|
}
|