@@ -16,6 +16,7 @@ import (
1616 "github.com/ElrondNetwork/elrond-go/common"
1717 "github.com/ElrondNetwork/elrond-go/sharding/mock"
1818 "github.com/ElrondNetwork/elrond-go/state"
19+ "github.com/ElrondNetwork/elrond-go/testscommon/epochstartmock"
1920 "github.com/ElrondNetwork/elrond-go/testscommon/genericMocks"
2021 "github.com/ElrondNetwork/elrond-go/testscommon/hashingMocks"
2122 "github.com/ElrondNetwork/elrond-go/testscommon/nodeTypeProviderMock"
@@ -77,7 +78,7 @@ func TestIndexHashedGroupSelectorWithRater_OkValShouldWork(t *testing.T) {
7778 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
7879 require .Nil (t , err )
7980
80- epochStartSubscriber := & mock .EpochStartNotifierStub {}
81+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
8182 bootStorer := genericMocks .NewStorerMock ()
8283
8384 arguments := ArgNodesCoordinator {
@@ -176,7 +177,7 @@ func BenchmarkIndexHashedGroupSelectorWithRater_ComputeValidatorsGroup63of400(b
176177 }
177178 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
178179 require .Nil (b , err )
179- epochStartSubscriber := & mock .EpochStartNotifierStub {}
180+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
180181 bootStorer := genericMocks .NewStorerMock ()
181182
182183 arguments := ArgNodesCoordinator {
@@ -254,7 +255,7 @@ func Test_ComputeValidatorsGroup63of400(t *testing.T) {
254255 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
255256 require .Nil (t , err )
256257
257- epochStartSubscriber := & mock .EpochStartNotifierStub {}
258+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
258259 bootStorer := genericMocks .NewStorerMock ()
259260
260261 arguments := ArgNodesCoordinator {
@@ -329,7 +330,7 @@ func TestIndexHashedGroupSelectorWithRater_GetValidatorWithPublicKeyShouldReturn
329330 nodeShuffler , err := NewHashValidatorsShuffler (sufflerArgs )
330331 require .Nil (t , err )
331332
332- epochStartSubscriber := & mock .EpochStartNotifierStub {}
333+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
333334 bootStorer := genericMocks .NewStorerMock ()
334335
335336 arguments := ArgNodesCoordinator {
@@ -385,7 +386,7 @@ func TestIndexHashedGroupSelectorWithRater_GetValidatorWithPublicKeyShouldReturn
385386 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
386387 require .Nil (t , err )
387388
388- epochStartSubscriber := & mock .EpochStartNotifierStub {}
389+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
389390 bootStorer := genericMocks .NewStorerMock ()
390391
391392 arguments := ArgNodesCoordinator {
@@ -451,7 +452,7 @@ func TestIndexHashedGroupSelectorWithRater_GetValidatorWithPublicKeyShouldWork(t
451452 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
452453 require .Nil (t , err )
453454
454- epochStartSubscriber := & mock .EpochStartNotifierStub {}
455+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
455456 bootStorer := genericMocks .NewStorerMock ()
456457
457458 eligibleMap [core .MetachainShardId ] = listMeta
@@ -537,7 +538,7 @@ func TestIndexHashedGroupSelectorWithRater_GetAllEligibleValidatorsPublicKeys(t
537538 }
538539 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
539540 require .Nil (t , err )
540- epochStartSubscriber := & mock .EpochStartNotifierStub {}
541+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
541542 bootStorer := genericMocks .NewStorerMock ()
542543
543544 eligibleMap [core .MetachainShardId ] = listMeta
@@ -850,7 +851,7 @@ func BenchmarkIndexHashedWithRaterGroupSelector_ComputeValidatorsGroup21of400(b
850851 nodeShuffler , err := NewHashValidatorsShuffler (shufflerArgs )
851852 require .Nil (b , err )
852853
853- epochStartSubscriber := & mock .EpochStartNotifierStub {}
854+ epochStartSubscriber := & epochstartmock .EpochStartNotifierStub {}
854855 bootStorer := genericMocks .NewStorerMock ()
855856
856857 arguments := ArgNodesCoordinator {
0 commit comments