mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-21 14:14:30 +00:00
b2c38ce397
Allow to register body extras in consumers of libevm. Co-authored-by: Arran Schlosberg <519948+ARR4N@users.noreply.github.com>
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
// Copyright 2024 the libevm authors.
|
|
//
|
|
// The libevm additions to go-ethereum are free software: you can redistribute
|
|
// them and/or modify them 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 libevm additions are distributed in the hope that they 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/stretchr/testify/require"
|
|
|
|
"github.com/ava-labs/libevm/common"
|
|
"github.com/ava-labs/libevm/core/types"
|
|
)
|
|
|
|
func TestStateObjectEmpty(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
registerAndSet func(*types.StateAccount)
|
|
wantEmpty bool
|
|
}{
|
|
{
|
|
name: "no registered types.StateAccount extra payload",
|
|
registerAndSet: func(*types.StateAccount) {},
|
|
wantEmpty: true,
|
|
},
|
|
{
|
|
name: "erroneously non-nil types.StateAccountExtra when no registered payload",
|
|
registerAndSet: func(acc *types.StateAccount) {
|
|
acc.Extra = &types.StateAccountExtra{}
|
|
},
|
|
wantEmpty: true,
|
|
},
|
|
{
|
|
name: "explicit false bool",
|
|
registerAndSet: func(acc *types.StateAccount) {
|
|
types.RegisterExtras[
|
|
types.NOOPHeaderHooks, *types.NOOPHeaderHooks,
|
|
types.NOOPBodyHooks, *types.NOOPBodyHooks,
|
|
bool,
|
|
]().StateAccount.Set(acc, false)
|
|
},
|
|
wantEmpty: true,
|
|
},
|
|
{
|
|
name: "implicit false bool",
|
|
registerAndSet: func(*types.StateAccount) {
|
|
types.RegisterExtras[
|
|
types.NOOPHeaderHooks, *types.NOOPHeaderHooks,
|
|
types.NOOPBodyHooks, *types.NOOPBodyHooks,
|
|
bool,
|
|
]()
|
|
},
|
|
wantEmpty: true,
|
|
},
|
|
{
|
|
name: "true bool",
|
|
registerAndSet: func(acc *types.StateAccount) {
|
|
types.RegisterExtras[
|
|
types.NOOPHeaderHooks, *types.NOOPHeaderHooks,
|
|
types.NOOPBodyHooks, *types.NOOPBodyHooks,
|
|
bool,
|
|
]().StateAccount.Set(acc, true)
|
|
},
|
|
wantEmpty: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
types.TestOnlyClearRegisteredExtras()
|
|
t.Cleanup(types.TestOnlyClearRegisteredExtras)
|
|
|
|
obj := newObject(nil, common.Address{}, nil)
|
|
tt.registerAndSet(&obj.data)
|
|
require.Equalf(t, tt.wantEmpty, obj.empty(), "%T.empty()", obj)
|
|
})
|
|
}
|
|
}
|