woop/core/core_test.go

package core

import (
	"math/big"
	"testing"

	blockfactory "github.com/harmony-one/harmony/block/factory"
	"github.com/harmony-one/harmony/core/types"
	shardingconfig "github.com/harmony-one/harmony/internal/configs/sharding"
)

func TestIsEpochBlock(t *testing.T) {
	blockNumbered := func(n int64) *types.Block {
		return types.NewBlock(
			blockfactory.NewTestHeader().With().Number(big.NewInt(n)).Header(),
			nil, nil, nil, nil, nil,
		)
	}
	tests := []struct {
		schedule shardingconfig.Schedule
		block    *types.Block
		expected bool
	}{
		{
			shardingconfig.MainnetSchedule,
			blockNumbered(10),
			false,
		},
		{
			shardingconfig.MainnetSchedule,
			blockNumbered(0),
			true,
		},
		{
			shardingconfig.MainnetSchedule,
			blockNumbered(344064),
			true,
		},
		{
			shardingconfig.TestnetSchedule,
			blockNumbered(74),
			false,
		},
		{
			shardingconfig.TestnetSchedule,
			blockNumbered(75),
			true,
		},
		{
			shardingconfig.TestnetSchedule,
			blockNumbered(149),
			false,
		},
		{
			shardingconfig.TestnetSchedule,
			blockNumbered(150),
			true,
		},
	}
	for i, test := range tests {
		ShardingSchedule = test.schedule
		r := IsEpochBlock(test.block)
		if r != test.expected {
			t.Errorf("index: %v, expected: %v, got: %v\n", i, test.expected, r)
		}
	}
}