diff --git a/activation/mocks.go b/activation/mocks.go index 50f241460e..14176cbc12 100644 --- a/activation/mocks.go +++ b/activation/mocks.go @@ -387,13 +387,13 @@ func (c *nipostValidatorPositioningAtxCall) Return(arg0 error) *nipostValidatorP } // Do rewrite *gomock.Call.Do -func (c *nipostValidatorPositioningAtxCall) Do(f func(*types.ATXID, atxProvider, types.ATXID, types.EpochID) error) *nipostValidatorPositioningAtxCall { +func (c *nipostValidatorPositioningAtxCall) Do(f func(types.ATXID, atxProvider, types.ATXID, types.EpochID) error) *nipostValidatorPositioningAtxCall { c.Call = c.Call.Do(f) return c } // DoAndReturn rewrite *gomock.Call.DoAndReturn -func (c *nipostValidatorPositioningAtxCall) DoAndReturn(f func(*types.ATXID, atxProvider, types.ATXID, types.EpochID) error) *nipostValidatorPositioningAtxCall { +func (c *nipostValidatorPositioningAtxCall) DoAndReturn(f func(types.ATXID, atxProvider, types.ATXID, types.EpochID) error) *nipostValidatorPositioningAtxCall { c.Call = c.Call.DoAndReturn(f) return c } diff --git a/activation/nipost_test.go b/activation/nipost_test.go index 3ec687567e..93bb3a51b7 100644 --- a/activation/nipost_test.go +++ b/activation/nipost_test.go @@ -1056,7 +1056,7 @@ func TestNIPostBuilder_Mainnet_PoetRound3_Workaround(t *testing.T) { postProvider.EXPECT().GenerateProof(gomock.Any(), gomock.Any(), gomock.Any()) nipostValidator := NewMocknipostValidator(ctrl) - nipostValidator.EXPECT().Post(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil) + nipostValidator.EXPECT().Post(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil) poets := make([]PoetProvingServiceClient, 0, 2) {