rename v8Test to RewardsTest

Author: 0xfornax

shared/services/rewards/generator-v8_test.go

@@ -2,28 +2,21 @@ package rewards
 
 import (
 	"fmt"
-	"math/big"
 	"os"
 	"path/filepath"
 	"testing"
 
-	"github.com/ethereum/go-ethereum/common"
-	"github.com/ethereum/go-ethereum/core/types"
-	"github.com/fatih/color"
-	"github.com/rocket-pool/smartnode/shared/services/beacon"
 	"github.com/rocket-pool/smartnode/shared/services/rewards/test"
-	"github.com/rocket-pool/smartnode/shared/services/rewards/test/assets"
 	"github.com/rocket-pool/smartnode/shared/services/state"
-	"github.com/rocket-pool/smartnode/shared/utils/log"
 )
 
-type v8Test struct {
+type RewardsTest struct {
 	*testing.T
 	rp *test.MockRocketPool
 	bc *test.MockBeaconClient
 }
 
-func (t *v8Test) saveArtifacts(prefix string, result *GenerateTreeResult) {
+func (t *RewardsTest) saveArtifacts(prefix string, result *GenerateTreeResult) {
 	tmpDir, err := os.MkdirTemp("", fmt.Sprintf("artifacts-%s", t.Name()))
 	t.failIf(err)
 	rewardsLocalFile := LocalFile[IRewardsFile]{
@@ -42,23 +35,23 @@ func (t *v8Test) saveArtifacts(prefix string, result *GenerateTreeResult) {
 	t.Logf("wrote artifacts to %s\n", tmpDir)
 }
 
-func newV8Test(t *testing.T, index uint64) *v8Test {
+func newRewardsTest(t *testing.T, index uint64) *RewardsTest {
 	rp := test.NewMockRocketPool(t, index)
-	out := &v8Test{
+	out := &RewardsTest{
 		T:  t,
 		rp: rp,
 		bc: test.NewMockBeaconClient(t),
 	}
 	return out
 }
 
-func (t *v8Test) failIf(err error) {
+func (t *RewardsTest) failIf(err error) {
 	if err != nil {
 		t.Fatal(err)
 	}
 }
 
-func (t *v8Test) SetMinipoolPerformance(canonicalMinipoolPerformance IPerformanceFile, networkState *state.NetworkState) {
+func (t *RewardsTest) SetMinipoolPerformance(canonicalMinipoolPerformance IPerformanceFile, networkState *state.NetworkState) {
 	addresses := canonicalMinipoolPerformance.GetMinipoolAddresses()
 	for _, address := range addresses {
 
@@ -79,75 +72,3 @@ func (t *v8Test) SetMinipoolPerformance(canonicalMinipoolPerformance IPerformanc
 		t.bc.SetMinipoolPerformance(validatorStatus.Index, missedSlots)
 	}
 }
-
-// TestV8Mainnet builds a tree using serialized state for a mainnet interval that used v8
-// and checks that the resulting artifacts match their canonical values.
-func TestV8Mainnet(tt *testing.T) {
-	state := assets.GetMainnet20RewardsState()
-
-	t := newV8Test(tt, state.NetworkDetails.RewardIndex)
-
-	canonical, err := DeserializeRewardsFile(assets.GetMainnet20RewardsJSON())
-	t.failIf(err)
-
-	canonicalPerformance, err := DeserializeMinipoolPerformanceFile(assets.GetMainnet20MinipoolPerformanceJSON())
-	t.failIf(err)
-
-	t.Logf("pending rpl rewards: %s", state.NetworkDetails.PendingRPLRewards.String())
-
-	t.bc.SetState(state)
-
-	// Some interval info needed for mocks
-	consensusStartBlock := canonical.GetConsensusStartBlock()
-	executionStartBlock := canonical.GetExecutionStartBlock()
-	consensusEndBlock := canonical.GetConsensusEndBlock()
-
-	// Create a new treeGeneratorImpl_v8
-	logger := log.NewColorLogger(color.Faint)
-	generator := newTreeGeneratorImpl_v8(
-		&logger,
-		t.Name(),
-		state.NetworkDetails.RewardIndex,
-		canonical.GetStartTime(),
-		canonical.GetEndTime(),
-		consensusEndBlock,
-		&types.Header{
-			Number: big.NewInt(int64(canonical.GetExecutionEndBlock())),
-			Time:   assets.Mainnet20ELHeaderTime,
-		},
-		canonical.GetIntervalsPassed(),
-		state,
-	)
-
-	// Load the mock up
-	t.rp.SetRewardSnapshotEvent(assets.GetRewardSnapshotEventInterval19())
-	t.bc.SetBeaconBlock(fmt.Sprint(consensusStartBlock-1), beacon.BeaconBlock{ExecutionBlockNumber: executionStartBlock - 1})
-	t.bc.SetBeaconBlock(fmt.Sprint(consensusStartBlock), beacon.BeaconBlock{ExecutionBlockNumber: executionStartBlock})
-	t.rp.SetHeaderByNumber(big.NewInt(int64(executionStartBlock)), &types.Header{Time: uint64(canonical.GetStartTime().Unix())})
-
-	// Set the critical duties slots
-	t.bc.SetCriticalDutiesSlots(assets.GetMainnet20CriticalDutiesSlots())
-
-	// Set the minipool performance
-	t.SetMinipoolPerformance(canonicalPerformance, state)
-
-	artifacts, err := generator.generateTree(
-		t.rp,
-		"mainnet",
-		make([]common.Address, 0),
-		t.bc,
-	)
-	t.failIf(err)
-
-	// Save the artifacts if verbose mode is enabled
-	if testing.Verbose() {
-		t.saveArtifacts("", artifacts)
-	}
-
-	t.Logf("merkle root: %s\n", artifacts.RewardsFile.GetMerkleRoot())
-	if artifacts.RewardsFile.GetMerkleRoot() != canonical.GetMerkleRoot() {
-		t.Fatalf("Merkle root does not match %s", canonical.GetMerkleRoot())
-	} else {
-		t.Logf("merkle root matches %s", canonical.GetMerkleRoot())
-	}
-}