mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-08-20 10:52:25 +00:00
87 lines
3.1 KiB
Go
87 lines
3.1 KiB
Go
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/rawdb"
|
|
"github.com/ethereum/go-ethereum/core/state"
|
|
"github.com/ethereum/go-ethereum/core/vm"
|
|
"github.com/ethereum/go-ethereum/core/vm/runtime"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
)
|
|
|
|
var chainConfig params.ChainConfig
|
|
|
|
func init() {
|
|
chainConfig = params.ChainConfig{
|
|
ChainID: big.NewInt(1),
|
|
HomesteadBlock: new(big.Int),
|
|
ByzantiumBlock: new(big.Int),
|
|
ConstantinopleBlock: new(big.Int),
|
|
DAOForkBlock: new(big.Int),
|
|
DAOForkSupport: false,
|
|
EIP150Block: new(big.Int),
|
|
EIP155Block: new(big.Int),
|
|
EIP158Block: new(big.Int),
|
|
}
|
|
}
|
|
|
|
func TestSloadAndStore(t *testing.T) {
|
|
rawStateManagerAbi, _ := ioutil.ReadFile("./StateManagerABI.json")
|
|
stateManagerAbi, _ := abi.JSON(strings.NewReader(string(rawStateManagerAbi)))
|
|
state := newState()
|
|
|
|
address := common.HexToAddress("9999999999999999999999999999999999999999")
|
|
key := [32]byte{}
|
|
value := [32]byte{}
|
|
copy(key[:], []byte("hello"))
|
|
copy(value[:], []byte("world"))
|
|
|
|
storeCalldata, _ := stateManagerAbi.Pack("setStorage", address, key, value)
|
|
getCalldata, _ := stateManagerAbi.Pack("getStorage", address, key)
|
|
|
|
call(t, state, vm.StateManagerAddress, storeCalldata)
|
|
getStorageReturnValue, _ := call(t, state, vm.StateManagerAddress, getCalldata)
|
|
|
|
if !bytes.Equal(value[:], getStorageReturnValue) {
|
|
t.Errorf("Expected %020x; got %020x", value[:], getStorageReturnValue)
|
|
}
|
|
}
|
|
|
|
func TestCreate(t *testing.T) {
|
|
initCode, _ := hex.DecodeString("6080604052348015600f57600080fd5b5060b28061001e6000396000f3fe6080604052348015600f57600080fd5b506004361060285760003560e01c80639b0b0fda14602d575b600080fd5b606060048036036040811015604157600080fd5b8101908080359060200190929190803590602001909291905050506062565b005b8060008084815260200190815260200160002081905550505056fea265627a7a7231582053ac32a8b70d1cf87fb4ebf5a538ea9d9e773351e6c8afbc4bf6a6c273187f4a64736f6c63430005110032")
|
|
rawStateManagerAbi, _ := ioutil.ReadFile("./StateManagerABI.json")
|
|
stateManagerAbi, _ := abi.JSON(strings.NewReader(string(rawStateManagerAbi)))
|
|
state := newState()
|
|
|
|
address := common.HexToAddress("9999999999999999999999999999999999999999")
|
|
callerAddress := common.HexToAddress("42")
|
|
|
|
deployContractCalldata, _ := stateManagerAbi.Pack("deployContract", address, initCode, true, callerAddress)
|
|
createdContractAddr, _ := call(t, state, vm.StateManagerAddress, deployContractCalldata)
|
|
if !bytes.Equal(createdContractAddr, []byte{}) {
|
|
t.Errorf("Expected %020x; got %020x", createdContractAddr, []byte{})
|
|
}
|
|
}
|
|
|
|
func newState() *state.StateDB {
|
|
db := state.NewDatabase(rawdb.NewMemoryDatabase())
|
|
state, _ := state.New(common.Hash{}, db, nil)
|
|
return state
|
|
}
|
|
|
|
func call(t *testing.T, state *state.StateDB, address common.Address, callData []byte) ([]byte, error) {
|
|
returnValue, _, err := runtime.Call(address, callData, &runtime.Config{
|
|
State: state,
|
|
ChainConfig: &chainConfig,
|
|
})
|
|
|
|
return returnValue, err
|
|
}
|