set mock statenotifier in testServiceOptsWithDB

kasey · kasey · commit d430e296564d · 2025-03-07T14:23:24.000-06:00
diff --git a/beacon-chain/blockchain/mock_test.go b/beacon-chain/blockchain/mock_test.go
@@ -3,6 +3,7 @@ package blockchain
 import (
 	"testing"
 
+	mock "github.com/prysmaticlabs/prysm/v5/beacon-chain/blockchain/testing"
 	testDB "github.com/prysmaticlabs/prysm/v5/beacon-chain/db/testing"
 	doublylinkedtree "github.com/prysmaticlabs/prysm/v5/beacon-chain/forkchoice/doubly-linked-tree"
 	"github.com/prysmaticlabs/prysm/v5/beacon-chain/startup"
@@ -18,6 +19,7 @@ func testServiceOptsWithDB(t *testing.T) []Option {
 		WithStateGen(stategen.New(beaconDB, fcs)),
 		WithForkChoiceStore(fcs),
 		WithClockSynchronizer(cs),
+		WithStateNotifier(&mock.MockStateNotifier{RecordEvents: true}),
 	}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,7 @@ package blockchain`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ mock "github.com/prysmaticlabs/prysm/v5/beacon-chain/blockchain/testing"`
`6`	`7`	`testDB "github.com/prysmaticlabs/prysm/v5/beacon-chain/db/testing"`
`7`	`8`	`doublylinkedtree "github.com/prysmaticlabs/prysm/v5/beacon-chain/forkchoice/doubly-linked-tree"`
`8`	`9`	`"github.com/prysmaticlabs/prysm/v5/beacon-chain/startup"`
`@@ -18,6 +19,7 @@ func testServiceOptsWithDB(t *testing.T) []Option {`
`18`	`19`	`WithStateGen(stategen.New(beaconDB, fcs)),`
`19`	`20`	`WithForkChoiceStore(fcs),`
`20`	`21`	`WithClockSynchronizer(cs),`
	`22`	`+ WithStateNotifier(&mock.MockStateNotifier{RecordEvents: true}),`
`21`	`23`	`}`
`22`	`24`	`}`
`23`	`25`