Add missing mock for new GenesisHash call

ChainSafe · Sep 24, 2024 · f879df0 · f879df0
1 parent f2541a3
commit f879df0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dot/network/service.go b/dot/network/service.go
@@ -258,8 +258,8 @@ func (s *Service) Start() error {
 	genesisHashProtocolId := protocol.ID(s.cfg.BlockState.GenesisHash().String())
 
 	s.host.registerStreamHandler(s.host.protocolID+SyncID, s.handleSyncStream)
-	s.host.registerStreamHandler(genesisHashProtocolId+WarpSyncID, s.handleWarpSyncStream)
 	s.host.registerStreamHandler(s.host.protocolID+lightID, s.handleLightStream)
+	s.host.registerStreamHandler(genesisHashProtocolId+WarpSyncID, s.handleWarpSyncStream)
 
 	// register block announce protocol
 	err := s.RegisterNotificationsProtocol(

diff --git a/dot/rpc/modules/system_integration_test.go b/dot/rpc/modules/system_integration_test.go
@@ -52,6 +52,9 @@ func newNetworkService(t *testing.T) *network.Service {
 	blockStateMock.EXPECT().
 		GetHighestFinalisedHeader().
 		Return(types.NewEmptyHeader(), nil).AnyTimes()
+	blockStateMock.EXPECT().
+		GenesisHash().
+		Return(common.MustBlake2bHash([]byte("genesis"))).AnyTimes()
 
 	syncerMock := NewMockSyncer(ctrl)