024d41d0c2
Removed exported statedb object accessors, reducing the chance for nasty bugs to creep in. It's also ugly and unnecessary to have these methods.
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{}, 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))
|
|
}
|
|
}
|