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) }