91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package state
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
checker "gopkg.in/check.v1"
|
|
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
)
|
|
|
|
type StateSuite struct {
|
|
state *StateDB
|
|
}
|
|
|
|
var _ = checker.Suite(&StateSuite{})
|
|
|
|
// var ZeroHash256 = make([]byte, 32)
|
|
|
|
func (s *StateSuite) TestDump(c *checker.C) {
|
|
// generate a few entries
|
|
obj1 := s.state.GetOrNewStateObject([]byte{0x01})
|
|
obj1.AddBalance(big.NewInt(22))
|
|
obj2 := s.state.GetOrNewStateObject([]byte{0x01, 0x02})
|
|
obj2.SetCode([]byte{3, 3, 3, 3, 3, 3, 3})
|
|
obj3 := s.state.GetOrNewStateObject([]byte{0x02})
|
|
obj3.SetBalance(big.NewInt(44))
|
|
|
|
// write some of them to the trie
|
|
s.state.UpdateStateObject(obj1)
|
|
s.state.UpdateStateObject(obj2)
|
|
|
|
// check that dump contains the state objects that are in trie
|
|
got := string(s.state.Dump())
|
|
want := `{
|
|
"root": "6e277ae8357d013e50f74eedb66a991f6922f93ae03714de58b3d0c5e9eee53f",
|
|
"accounts": {
|
|
"1468288056310c82aa4c01a7e12a10f8111a0560e72b700555479031b86c357d": {
|
|
"balance": "22",
|
|
"nonce": 0,
|
|
"root": "56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421",
|
|
"codeHash": "c5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470",
|
|
"storage": {}
|
|
},
|
|
"a17eacbc25cda025e81db9c5c62868822c73ce097cee2a63e33a2e41268358a1": {
|
|
"balance": "0",
|
|
"nonce": 0,
|
|
"root": "56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421",
|
|
"codeHash": "87874902497a5bb968da31a2998d8f22e949d1ef6214bcdedd8bae24cca4b9e3",
|
|
"storage": {}
|
|
}
|
|
}
|
|
}`
|
|
if got != want {
|
|
c.Errorf("dump mismatch:\ngot: %s\nwant: %s\n", got, want)
|
|
}
|
|
}
|
|
|
|
func (s *StateSuite) SetUpTest(c *checker.C) {
|
|
db, _ := ethdb.NewMemDatabase()
|
|
s.state = New(nil, db)
|
|
}
|
|
|
|
func (s *StateSuite) TestSnapshot(c *checker.C) {
|
|
stateobjaddr := []byte("aa")
|
|
storageaddr := common.Big("0")
|
|
data1 := common.NewValue(42)
|
|
data2 := common.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)
|
|
}
|