Creating fake_app_state test

vm/test/fake_app_state.go

@@ -0,0 +1,127 @@
+package main
+
+import (
+	"fmt"
+
+	. "github.com/tendermint/tendermint/common"
+	. "github.com/tendermint/tendermint/vm"
+)
+
+type FakeAppState struct {
+	accounts map[string]*Account
+	storage  map[string]Word
+	logs     []*Log
+}
+
+func (fas *FakeAppState) GetAccount(addr Word) (*Account, error) {
+	account := fas.accounts[addr.String()]
+	if account == nil {
+		return account, nil
+	} else {
+		return nil, Errorf("Invalid account addr: %v", addr)
+	}
+}
+
+func (fas *FakeAppState) UpdateAccount(account *Account) error {
+	_, ok := fas.accounts[account.Address.String()]
+	if !ok {
+		return Errorf("Invalid account addr: %v", account.Address.String())
+	} else {
+		// Nothing to do
+		return nil
+	}
+}
+
+func (fas *FakeAppState) DeleteAccount(account *Account) error {
+	_, ok := fas.accounts[account.Address.String()]
+	if !ok {
+		return Errorf("Invalid account addr: %v", account.Address.String())
+	} else {
+		// Delete account
+		delete(fas.accounts, account.Address.String())
+		return nil
+	}
+}
+
+func (fas *FakeAppState) CreateAccount(addr Word, balance uint64) (*Account, error) {
+	account := fas.accounts[addr.String()]
+	if account == nil {
+		return &Account{
+			Address:   addr,
+			Balance:   balance,
+			Code:      nil,
+			Nonce:     0,
+			StateRoot: Zero,
+		}, nil
+	} else {
+		return nil, Errorf("Invalid account addr: %v", addr)
+	}
+}
+
+func (fas *FakeAppState) GetStorage(addr Word, key Word) (Word, error) {
+	_, ok := fas.accounts[addr.String()]
+	if !ok {
+		return Zero, Errorf("Invalid account addr: %v", addr)
+	}
+
+	value, ok := fas.storage[addr.String()+key.String()]
+	if ok {
+		return value, nil
+	} else {
+		return Zero, nil
+	}
+}
+
+func (fas *FakeAppState) SetStorage(addr Word, key Word, value Word) (bool, error) {
+	_, ok := fas.accounts[addr.String()]
+	if !ok {
+		return false, Errorf("Invalid account addr: %v", addr)
+	}
+
+	_, ok = fas.storage[addr.String()+key.String()]
+	fas.storage[addr.String()+key.String()] = value
+	return ok, nil
+}
+
+func (fas *FakeAppState) RemoveStorage(addr Word, key Word) error {
+	_, ok := fas.accounts[addr.String()]
+	if !ok {
+		return Errorf("Invalid account addr: %v", addr)
+	}
+
+	delete(fas.storage, addr.String()+key.String())
+	return nil
+}
+
+func (fas *FakeAppState) AddLog(log *Log) {
+	fas.logs = append(fas.logs, log)
+}
+
+func main() {
+	appState := &FakeAppState{
+		accounts: make(map[string]*Account),
+		storage:  make(map[string]Word),
+		logs:     nil,
+	}
+	params := VMParams{
+		BlockHeight: 0,
+		BlockHash:   Zero,
+		BlockTime:   0,
+		GasLimit:    0,
+	}
+	ourVm := NewVM(appState, params)
+
+	// Create accounts
+	account1, err := appState.CreateAccount(Uint64ToWord(100), 0)
+	if err != nil {
+		panic(err)
+	}
+	account2, err := appState.CreateAccount(Uint64ToWord(101), 0)
+	if err != nil {
+		panic(err)
+	}
+
+	var gas uint64 = 1000
+	output, err := ourVm.Call(account1, account2, []byte{0x60, 0x01, 0x60, 0x01}, []byte{}, 0, &gas)
+	fmt.Printf("Output: %v Error: %v\n", output, err)
+}