path: root/ethchain/state_test.go

package ethchain

import (
    "github.com/ethereum/eth-go/ethdb"
    "github.com/ethereum/eth-go/ethutil"
    "testing"
)

func TestSnapshot(t *testing.T) {
    ethutil.ReadConfig("", ethutil.LogStd, "")

    db, _ := ethdb.NewMemDatabase()
    state := NewState(ethutil.NewTrie(db, ""))

    stateObject := NewContract([]byte("aa"), ethutil.Big1, ZeroHash256)
    state.UpdateStateObject(stateObject)
    stateObject.SetStorage(ethutil.Big("0"), ethutil.NewValue(42))

    snapshot := state.Snapshot()

    stateObject = state.GetStateObject([]byte("aa"))
    stateObject.SetStorage(ethutil.Big("0"), ethutil.NewValue(43))

    state.Revert(snapshot)

    stateObject = state.GetStateObject([]byte("aa"))
    if !stateObject.GetStorage(ethutil.Big("0")).Cmp(ethutil.NewValue(42)) {
        t.Error("Expected storage 0 to be 42")
    }
}