From 7f2c897eb90f7041a6d88eb6d37c47a8f375b2de Mon Sep 17 00:00:00 2001 From: Ian Shim Date: Mon, 24 Jun 2024 15:00:09 -0700 Subject: [PATCH] record args in mock transactor --- core/mock/tx.go | 2 +- disperser/dataapi/server_test.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/core/mock/tx.go b/core/mock/tx.go index b56c98b93..84ae5cd63 100644 --- a/core/mock/tx.go +++ b/core/mock/tx.go @@ -71,7 +71,7 @@ func (t *MockTransactor) UpdateOperatorSocket(ctx context.Context, socket string } func (t *MockTransactor) BuildEjectOperatorsTxn(ctx context.Context, operatorsByQuorum [][]core.OperatorID) (*types.Transaction, error) { - args := t.Called() + args := t.Called(ctx, operatorsByQuorum) result := args.Get(0) return result.(*types.Transaction), args.Error(1) } diff --git a/disperser/dataapi/server_test.go b/disperser/dataapi/server_test.go index 8711a918e..fc3fd5117 100644 --- a/disperser/dataapi/server_test.go +++ b/disperser/dataapi/server_test.go @@ -342,7 +342,7 @@ func TestEjectOperatorHandler(t *testing.T) { addr2 := gethcommon.HexToAddress("0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2") mockTx.On("BatchOperatorIDToAddress").Return([]gethcommon.Address{addr1, addr2}, nil) mockTx.On("GetQuorumBitmapForOperatorsAtBlockNumber").Return([]*big.Int{big.NewInt(3), big.NewInt(0)}, nil) - mockTx.On("BuildEjectOperatorsTxn").Return(types.NewTransaction(0, gethcommon.HexToAddress("0x1"), big.NewInt(0), 0, big.NewInt(0), []byte{}), nil) + mockTx.On("BuildEjectOperatorsTxn", mock.Anything, mock.Anything).Return(types.NewTransaction(0, gethcommon.HexToAddress("0x1"), big.NewInt(0), 0, big.NewInt(0), []byte{}), nil) mockTx.On("EjectOperators").Return(&types.Receipt{ GasUsed: uint64(10), }, nil)