test: all new hooks

Author: ARR4N

libevm/hookstest/stub.go

@@ -14,11 +14,11 @@ import (
 // Register clears any registered [params.Extras] and then registers `extras`
 // for the lifetime of the current test, clearing them via tb's
 // [testing.TB.Cleanup].
-func Register[C params.ChainConfigHooks, R params.RulesHooks](tb testing.TB, extras params.Extras[C, R]) {
+func Register[C params.ChainConfigHooks, R params.RulesHooks](tb testing.TB, extras params.Extras[C, R]) params.ExtraPayloads[C, R] {
 	tb.Helper()
 	params.TestOnlyClearRegisteredExtras()
 	tb.Cleanup(params.TestOnlyClearRegisteredExtras)
-	params.RegisterExtras(extras)
+	return params.RegisterExtras(extras)
 }
 
 // A Stub is a test double for [params.ChainConfigHooks] and
@@ -27,17 +27,17 @@ func Register[C params.ChainConfigHooks, R params.RulesHooks](tb testing.TB, ext
 type Stub struct {
 	CheckConfigForkOrderFn  func() error
 	CheckConfigCompatibleFn func(*params.ChainConfig, *big.Int, uint64) *params.ConfigCompatError
-	DescriptionValue        string
+	DescriptionSuffix       string
 	PrecompileOverrides     map[common.Address]libevm.PrecompiledContract
 	CanExecuteTransactionFn func(common.Address, *common.Address, libevm.StateReader) error
 	CanCreateContractFn     func(*libevm.AddressContext, uint64, libevm.StateReader) (uint64, error)
 }
 
 // Register is a convenience wrapper for registering s as both the
 // [params.ChainConfigHooks] and [params.RulesHooks] via [Register].
-func (s *Stub) Register(tb testing.TB) {
+func (s *Stub) Register(tb testing.TB) params.ExtraPayloads[*Stub, *Stub] {
 	tb.Helper()
-	Register(tb, params.Extras[*Stub, *Stub]{
+	return Register(tb, params.Extras[*Stub, *Stub]{
 		NewRules: func(_ *params.ChainConfig, _ *params.Rules, _ *Stub, blockNum *big.Int, isMerge bool, timestamp uint64) *Stub {
 			return s
 		},
@@ -75,7 +75,7 @@ func (s Stub) CheckConfigCompatible(newcfg *params.ChainConfig, headNumber *big.
 
 // Description returns s.DescriptionValue.
 func (s Stub) Description() string {
-	return s.DescriptionValue
+	return s.DescriptionSuffix
 }
 
 // CanExecuteTransaction proxies arguments to the s.CanExecuteTransactionFn

params/hooks.libevm_test.go

@@ -0,0 +1,62 @@
+package params_test
+
+import (
+	"errors"
+	"fmt"
+	"math/big"
+	"testing"
+
+	"github.com/ethereum/go-ethereum/libevm/ethtest"
+	"github.com/ethereum/go-ethereum/libevm/hookstest"
+	"github.com/ethereum/go-ethereum/params"
+	"github.com/stretchr/testify/require"
+)
+
+func TestChainConfigHooks_Description(t *testing.T) {
+	const suffix = "Arran was here"
+	c := new(params.ChainConfig)
+	want := c.Description() + suffix
+
+	hooks := &hookstest.Stub{
+		DescriptionSuffix: "Arran was here",
+	}
+	hooks.Register(t).SetOnChainConfig(c, hooks)
+	require.Equal(t, want, c.Description(), "ChainConfigHooks.Description() is appended to non-extras equivalent")
+}
+
+func TestChainConfigHooks_CheckConfigForkOrder(t *testing.T) {
+	err := errors.New("uh oh")
+
+	c := new(params.ChainConfig)
+	require.NoError(t, c.CheckConfigForkOrder(), "CheckConfigForkOrder() with no hooks")
+
+	hooks := &hookstest.Stub{
+		CheckConfigForkOrderFn: func() error { return err },
+	}
+	hooks.Register(t).SetOnChainConfig(c, hooks)
+	require.Equal(t, err, c.CheckConfigForkOrder(), "CheckConfigForkOrder() with error-producing hook")
+}
+
+func TestChainConfigHooks_CheckConfigCompatible(t *testing.T) {
+	rng := ethtest.NewPseudoRand(1234567890)
+	newcfg := &params.ChainConfig{
+		ChainID: rng.BigUint64(),
+	}
+	headNumber := rng.Uint64()
+	headTimestamp := rng.Uint64()
+
+	c := new(params.ChainConfig)
+	require.Nil(t, c.CheckCompatible(newcfg, headNumber, headTimestamp), "CheckCompatible() with no hooks")
+
+	makeCompatErr := func(newcfg *params.ChainConfig, headNumber *big.Int, headTimestamp uint64) *params.ConfigCompatError {
+		return &params.ConfigCompatError{
+			What: fmt.Sprintf("ChainID: %v Head #: %v Head Time: %d", newcfg.ChainID, headNumber, headTimestamp),
+		}
+	}
+	hooks := &hookstest.Stub{
+		CheckConfigCompatibleFn: makeCompatErr,
+	}
+	hooks.Register(t).SetOnChainConfig(c, hooks)
+	want := makeCompatErr(newcfg, new(big.Int).SetUint64(headNumber), headTimestamp)
+	require.Equal(t, want, c.CheckCompatible(newcfg, headNumber, headTimestamp), "CheckCompatible() with error-producing hook")
+}