fix test

fix

fix test

Author: serichoi65

pkg/eigenState/operatorAllocations/operatorAllocations_test.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/Layr-Labs/sidecar/internal/config"
 	"github.com/Layr-Labs/sidecar/internal/tests"
@@ -52,6 +53,15 @@ func Test_OperatorAllocations(t *testing.T) {
 
 		blockNumber := uint64(102)
 
+		// Create the block first (required for foreign key constraint)
+		block := &storage.Block{
+			Number:    blockNumber,
+			Hash:      "test_hash",
+			BlockTime: time.Now(),
+		}
+		res := grm.Model(&storage.Block{}).Create(block)
+		assert.Nil(t, res.Error)
+
 		log := &storage.TransactionLog{
 			TransactionHash:  "some hash",
 			TransactionIndex: big.NewInt(100).Uint64(),
@@ -87,7 +97,7 @@ func Test_OperatorAllocations(t *testing.T) {
 
 		results := make([]*OperatorAllocation, 0)
 		query := `select * from operator_allocations where block_number = ?`
-		res := model.DB.Raw(query, blockNumber).Scan(&results)
+		res = model.DB.Raw(query, blockNumber).Scan(&results)
 		assert.Nil(t, res.Error)
 		assert.Equal(t, 1, len(results))
 

pkg/rewards/operatorAllocationSnapshots.go

@@ -11,7 +11,7 @@ const operatorAllocationSnapshotsQuery = `
 		SELECT *,
 			   ROW_NUMBER() OVER (PARTITION BY operator, avs, strategy, operator_set_id, cast(block_time AS DATE) ORDER BY block_time DESC, log_index DESC) AS rn
 		FROM operator_allocations oa
-		INNER JOIN blocks b ON oa.block_number = b.number
+		INNER JOIN blocks b ON oa.effective_block = b.number
 		WHERE b.block_time < TIMESTAMP '{{.cutoffDate}}'
 	),
 	-- Get the latest record for each day
@@ -42,8 +42,8 @@ const operatorAllocationSnapshotsQuery = `
 			-- Deallocation (decrease or no change): Round DOWN to current day
 			CASE
 				WHEN LAG(magnitude) OVER (PARTITION BY operator, avs, strategy, operator_set_id ORDER BY block_time, block_number, log_index) IS NULL THEN
-					-- First allocation: round down to current day (conservative default)
-					date_trunc('day', block_time)
+					-- First allocation: round up to next day
+					date_trunc('day', block_time) + INTERVAL '1' day
 				WHEN magnitude > LAG(magnitude) OVER (PARTITION BY operator, avs, strategy, operator_set_id ORDER BY block_time, block_number, log_index) THEN
 					-- Increase: round up to next day
 					date_trunc('day', block_time) + INTERVAL '1' day