@@ -51,7 +51,7 @@ func TestApiController_RPCHandler(t *testing.T) {
5151 recordRepoMock .On ("Save" , mock .Anything ).Return (nil )
5252
5353 actionsMockObject := new (actionMocks.Actions )
54- actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything ).Return ()
54+ actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything , mock . Anything ).Return ()
5555
5656 apiController := NewApiController (false , repositories.Repos {
5757 NodeRepo : & nodeRepoMock ,
@@ -160,7 +160,7 @@ func TestApiController_BatchRPCHandler(t *testing.T) {
160160 recordRepoMock .On ("Save" , mock .Anything ).Return (nil )
161161
162162 actionsMockObject := new (actionMocks.Actions )
163- actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything ).Return ()
163+ actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything , mock . Anything ).Return ()
164164
165165 apiController := NewApiController (false , repositories.Repos {
166166 NodeRepo : & nodeRepoMock ,
@@ -242,7 +242,7 @@ func TestApiController_RPCHandler_InvalidBody(t *testing.T) {
242242 }
243243
244244 actionsMockObject := new (actionMocks.Actions )
245- actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything ).Return ()
245+ actionsMockObject .On ("PenalizeNode" , mock .Anything , mock .Anything , mock . Anything ).Return ()
246246
247247 apiController := NewApiController (false , repositories.Repos {}, actionsMockObject )
248248
0 commit comments