36 lines
851 B
Go
36 lines
851 B
Go
|
package ethstate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/ethereum/eth-go/ethdb"
|
||
|
|
"github.com/ethereum/eth-go/ethtrie"
|
||
|
|
"github.com/ethereum/eth-go/ethutil"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var ZeroHash256 = make([]byte, 32)
|
||
|
|
|
||
|
|
func TestSnapshot(t *testing.T) {
|
||
|
|
db, _ := ethdb.NewMemDatabase()
|
||
|
|
ethutil.ReadConfig(".ethtest", "/tmp/ethtest", "")
|
||
|
|
ethutil.Config.Db = db
|
||
|
|
|
||
|
|
state := NewState(ethtrie.NewTrie(db, ""))
|
||
|
|
|
||
|
|
stateObject := state.GetOrNewStateObject([]byte("aa"))
|
||
|
|
|
||
|
|
stateObject.SetStorage(ethutil.Big("0"), ethutil.NewValue(42))
|
||
|
|
|
||
|
|
snapshot := state.Copy()
|
||
|
|
|
||
|
|
stateObject = state.GetStateObject([]byte("aa"))
|
||
|
|
stateObject.SetStorage(ethutil.Big("0"), ethutil.NewValue(43))
|
||
|
|
|
||
|
|
state.Set(snapshot)
|
||
|
|
|
||
|
|
stateObject = state.GetStateObject([]byte("aa"))
|
||
|
|
res := stateObject.GetStorage(ethutil.Big("0"))
|
||
|
|
if !res.Cmp(ethutil.NewValue(42)) {
|
||
|
|
t.Error("Expected storage 0 to be 42", res)
|
||
|
|
}
|
||
|
|
}
|