Add mock interval tests

Author: jshufro

shared/services/rewards/generator-v8_test.go

@@ -23,15 +23,15 @@ type v8Test struct {
 	bc *test.MockBeaconClient
 }
 
-func (t *v8Test) saveArtifacts(result *GenerateTreeResult) {
+func (t *v8Test) saveArtifacts(prefix string, result *GenerateTreeResult) {
 	tmpDir, err := os.MkdirTemp("", fmt.Sprintf("artifacts-%s", t.Name()))
 	t.failIf(err)
 	rewardsLocalFile := LocalFile[IRewardsFile]{
-		fullPath: filepath.Join(tmpDir, "rewards.json"),
+		fullPath: filepath.Join(tmpDir, fmt.Sprintf("%s-ewards.json", prefix)),
 		f:        result.RewardsFile,
 	}
 	performanceLocalFile := LocalFile[IMinipoolPerformanceFile]{
-		fullPath: filepath.Join(tmpDir, "minipool-performance.json"),
+		fullPath: filepath.Join(tmpDir, fmt.Sprintf("%s-minipool-performance.json", prefix)),
 		f:        result.MinipoolPerformanceFile,
 	}
 	_, err = rewardsLocalFile.Write()
@@ -106,7 +106,7 @@ func TestV8Mainnet(tt *testing.T) {
 	generator := newTreeGeneratorImpl_v8(
 		&logger,
 		t.Name(),
-		canonical.GetIndex(),
+		state.NetworkDetails.RewardIndex,
 		canonical.GetStartTime(),
 		canonical.GetEndTime(),
 		consensusEndBlock,
@@ -140,7 +140,7 @@ func TestV8Mainnet(tt *testing.T) {
 
 	// Save the artifacts if verbose mode is enabled
 	if testing.Verbose() {
-		t.saveArtifacts(artifacts)
+		t.saveArtifacts("", artifacts)
 	}
 
 	t.Logf("merkle root: %s\n", artifacts.RewardsFile.GetMerkleRoot())

shared/services/rewards/mock_test.go

@@ -0,0 +1,108 @@
+package rewards
+
+// This file contains treegen tests which use mock history.
+// These mocks are faster to process than real history, and are useful for
+// testing new features and refactoring.
+
+import (
+	"fmt"
+	"math/big"
+	"strings"
+	"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/utils/log"
+)
+
+func TestMockIntervalDefaultsTreegenv8v9(tt *testing.T) {
+	t := newV8Test(tt)
+
+	history := test.NewDefaultMockHistory()
+	state := history.GetEndNetworkState()
+
+	t.bc.SetState(state)
+
+	consensusStartBlock := history.GetConsensusStartBlock()
+	executionStartBlock := history.GetExecutionStartBlock()
+	consensusEndBlock := history.GetConsensusEndBlock()
+	executionEndBlock := history.GetExecutionEndBlock()
+
+	logger := log.NewColorLogger(color.Faint)
+	generator := newTreeGeneratorImpl_v8(
+		&logger,
+		t.Name(),
+		state.NetworkDetails.RewardIndex,
+		history.GetStartTime(),
+		history.GetEndTime(),
+		consensusEndBlock,
+		&types.Header{
+			Number: big.NewInt(int64(history.GetExecutionEndBlock())),
+			Time:   assets.Mainnet20ELHeaderTime,
+		},
+		/* intervalsPassed= */ 1,
+		state,
+	)
+
+	t.rp.SetRewardSnapshotEvent(history.GetPreviousRewardSnapshotEvent())
+	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(history.GetStartTime().Unix())})
+
+	for _, validator := range state.ValidatorDetails {
+		t.bc.SetMinipoolPerformance(validator.Index, make([]uint64, 0))
+	}
+
+	v8Artifacts, err := generator.generateTree(
+		t.rp,
+		"mainnet",
+		make([]common.Address, 0),
+		t.bc,
+	)
+	t.failIf(err)
+
+	if testing.Verbose() {
+		t.saveArtifacts("v8", v8Artifacts)
+	}
+	generatorv9 := newTreeGeneratorImpl_v9(
+		&logger,
+		t.Name(),
+		state.NetworkDetails.RewardIndex,
+		&SnapshotEnd{
+			Slot:           consensusEndBlock,
+			ConsensusBlock: consensusEndBlock,
+			ExecutionBlock: executionEndBlock,
+		},
+		&types.Header{
+			Number: big.NewInt(int64(history.GetExecutionEndBlock())),
+			Time:   assets.Mainnet20ELHeaderTime,
+		},
+		/* intervalsPassed= */ 1,
+		state,
+	)
+
+	v9Artifacts, err := generatorv9.generateTree(
+		t.rp,
+		"mainnet",
+		make([]common.Address, 0),
+		t.bc,
+	)
+	t.failIf(err)
+
+	if testing.Verbose() {
+		t.saveArtifacts("v9", v9Artifacts)
+	}
+
+	v8MerkleRoot := v8Artifacts.RewardsFile.GetMerkleRoot()
+	v9MerkleRoot := v9Artifacts.RewardsFile.GetMerkleRoot()
+
+	if !strings.EqualFold(v8MerkleRoot, v9MerkleRoot) {
+		t.Fatalf("Merkle root does not match %s != %s", v8MerkleRoot, v9MerkleRoot)
+	} else {
+		t.Logf("v8/v9 Merkle root matches %s", v8MerkleRoot)
+	}
+}