9e5f03b6c4
With this commit, core/state's access to the underlying key/value database is mediated through an interface. Database errors are tracked in StateDB and returned by CommitTo or the new Error method. Motivation for this change: We can remove the light client's duplicated copy of core/state. The light client now supports node iteration, so tracing and storage enumeration can work with the light client (not implemented in this commit).
125 lines
3.0 KiB
Go
125 lines
3.0 KiB
Go
// Copyright 2015 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package state
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
)
|
|
|
|
var addr = common.BytesToAddress([]byte("test"))
|
|
|
|
func create() (*ManagedState, *account) {
|
|
db, _ := ethdb.NewMemDatabase()
|
|
statedb, _ := New(common.Hash{}, NewDatabase(db))
|
|
ms := ManageState(statedb)
|
|
ms.StateDB.SetNonce(addr, 100)
|
|
ms.accounts[addr] = newAccount(ms.StateDB.getStateObject(addr))
|
|
return ms, ms.accounts[addr]
|
|
}
|
|
|
|
func TestNewNonce(t *testing.T) {
|
|
ms, _ := create()
|
|
|
|
nonce := ms.NewNonce(addr)
|
|
if nonce != 100 {
|
|
t.Error("expected nonce 100. got", nonce)
|
|
}
|
|
|
|
nonce = ms.NewNonce(addr)
|
|
if nonce != 101 {
|
|
t.Error("expected nonce 101. got", nonce)
|
|
}
|
|
}
|
|
|
|
func TestRemove(t *testing.T) {
|
|
ms, account := create()
|
|
|
|
nn := make([]bool, 10)
|
|
for i := range nn {
|
|
nn[i] = true
|
|
}
|
|
account.nonces = append(account.nonces, nn...)
|
|
|
|
i := uint64(5)
|
|
ms.RemoveNonce(addr, account.nstart+i)
|
|
if len(account.nonces) != 5 {
|
|
t.Error("expected", i, "'th index to be false")
|
|
}
|
|
}
|
|
|
|
func TestReuse(t *testing.T) {
|
|
ms, account := create()
|
|
|
|
nn := make([]bool, 10)
|
|
for i := range nn {
|
|
nn[i] = true
|
|
}
|
|
account.nonces = append(account.nonces, nn...)
|
|
|
|
i := uint64(5)
|
|
ms.RemoveNonce(addr, account.nstart+i)
|
|
nonce := ms.NewNonce(addr)
|
|
if nonce != 105 {
|
|
t.Error("expected nonce to be 105. got", nonce)
|
|
}
|
|
}
|
|
|
|
func TestRemoteNonceChange(t *testing.T) {
|
|
ms, account := create()
|
|
nn := make([]bool, 10)
|
|
for i := range nn {
|
|
nn[i] = true
|
|
}
|
|
account.nonces = append(account.nonces, nn...)
|
|
ms.NewNonce(addr)
|
|
|
|
ms.StateDB.stateObjects[addr].data.Nonce = 200
|
|
nonce := ms.NewNonce(addr)
|
|
if nonce != 200 {
|
|
t.Error("expected nonce after remote update to be", 200, "got", nonce)
|
|
}
|
|
ms.NewNonce(addr)
|
|
ms.NewNonce(addr)
|
|
ms.NewNonce(addr)
|
|
ms.StateDB.stateObjects[addr].data.Nonce = 200
|
|
nonce = ms.NewNonce(addr)
|
|
if nonce != 204 {
|
|
t.Error("expected nonce after remote update to be", 204, "got", nonce)
|
|
}
|
|
}
|
|
|
|
func TestSetNonce(t *testing.T) {
|
|
ms, _ := create()
|
|
|
|
var addr common.Address
|
|
ms.SetNonce(addr, 10)
|
|
|
|
if ms.GetNonce(addr) != 10 {
|
|
t.Error("Expected nonce of 10, got", ms.GetNonce(addr))
|
|
}
|
|
|
|
addr[0] = 1
|
|
ms.StateDB.SetNonce(addr, 1)
|
|
|
|
if ms.GetNonce(addr) != 1 {
|
|
t.Error("Expected nonce of 1, got", ms.GetNonce(addr))
|
|
}
|
|
}
|