woop/core/core_test.go

package core

import (
	"math/big"
	"testing"

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

func TestIsEpochBlock(t *testing.T) {
	block1 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(10)).Header(), nil, nil, nil, nil)
	block2 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(0)).Header(), nil, nil, nil, nil)
	block3 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(344064)).Header(), nil, nil, nil, nil)
	block4 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(77)).Header(), nil, nil, nil, nil)
	block5 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(78)).Header(), nil, nil, nil, nil)
	block6 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(188)).Header(), nil, nil, nil, nil)
	block7 := types.NewBlock(block.NewHeaderWith().Number(big.NewInt(189)).Header(), nil, nil, nil, nil)
	tests := []struct {
		schedule shardingconfig.Schedule
		block    *types.Block
		expected bool
	}{
		{
			shardingconfig.MainnetSchedule,
			block1,
			false,
		},
		{
			shardingconfig.MainnetSchedule,
			block2,
			true,
		},
		{
			shardingconfig.MainnetSchedule,
			block3,
			true,
		},
		{
			shardingconfig.TestnetSchedule,
			block4,
			false,
		},
		{
			shardingconfig.TestnetSchedule,
			block5,
			true,
		},
		{
			shardingconfig.TestnetSchedule,
			block6,
			false,
		},
		{
			shardingconfig.TestnetSchedule,
			block7,
			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)
		}
	}
}