// 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 params_test

import (
	"errors"
	"fmt"
	"math/big"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/ava-labs/libevm/libevm/ethtest"
	"github.com/ava-labs/libevm/libevm/hookstest"
	"github.com/ava-labs/libevm/params"
)

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).ChainConfig.Set(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).ChainConfig.Set(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).ChainConfig.Set(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")
}