FEAT: Unit tests and check subcomps directed staked factories

multiversx · Dec 3, 2024 · 481ffdc · 481ffdc
1 parent 0c04227
commit 481ffdc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/factory/runType/runTypeComponentsHandler.go b/factory/runType/runTypeComponentsHandler.go
@@ -252,6 +252,15 @@ func (mrc *managedRunTypeComponents) CheckSubcomponents() error {
 	if check.IfNil(mrc.apiRewardTxHandler) {
 		return errors.ErrNilAPIRewardsHandler
 	}
+	if check.IfNil(mrc.delegatedListFactoryHandler) {
+		return factory.ErrNilDelegatedListFactory
+	}
+	if check.IfNil(mrc.directStakedListFactoryHandler) {
+		return factory.ErrNilDirectStakedListFactory
+	}
+	if check.IfNil(mrc.totalStakedValueFactoryHandler) {
+		return factory.ErrNilTotalStakedValueFactory
+	}
 
 	return nil
 }

diff --git a/factory/runType/runTypeComponentsHandler_test.go b/factory/runType/runTypeComponentsHandler_test.go
@@ -103,6 +103,9 @@ func TestManagedRunTypeComponents_Create(t *testing.T) {
 		require.Nil(t, managedRunTypeComponents.ValidatorAccountsSyncerFactoryHandler())
 		require.Nil(t, managedRunTypeComponents.ShardRequestersContainerCreatorHandler())
 		require.Nil(t, managedRunTypeComponents.APIRewardsTxHandler())
+		require.Nil(t, managedRunTypeComponents.DelegatedListFactoryHandler())
+		require.Nil(t, managedRunTypeComponents.DirectStakedListFactoryHandler())
+		require.Nil(t, managedRunTypeComponents.TotalStakedValueFactoryHandler())
 
 		err = managedRunTypeComponents.Create()
 		require.NoError(t, err)
@@ -155,6 +158,9 @@ func TestManagedRunTypeComponents_Create(t *testing.T) {
 		require.NotNil(t, managedRunTypeComponents.ValidatorAccountsSyncerFactoryHandler())
 		require.NotNil(t, managedRunTypeComponents.ShardRequestersContainerCreatorHandler())
 		require.NotNil(t, managedRunTypeComponents.APIRewardsTxHandler())
+		require.NotNil(t, managedRunTypeComponents.DelegatedListFactoryHandler())
+		require.NotNil(t, managedRunTypeComponents.DirectStakedListFactoryHandler())
+		require.NotNil(t, managedRunTypeComponents.TotalStakedValueFactoryHandler())
 
 		require.Equal(t, factory.RunTypeComponentsName, managedRunTypeComponents.String())
 		require.NoError(t, managedRunTypeComponents.Close())