mock_claimable_balance_batch_insert_builder.go

package history

import (
	"context"

	"github.com/stretchr/testify/mock"
)

type MockClaimableBalanceBatchInsertBuilder struct {
	mock.Mock
}

func (m *MockClaimableBalanceBatchInsertBuilder) Add(claimableBalance ClaimableBalance) error {
	a := m.Called(claimableBalance)
	return a.Error(0)
}

func (m *MockClaimableBalanceBatchInsertBuilder) Exec(ctx context.Context) error {
	a := m.Called(ctx)
	return a.Error(0)
}

// Len returns the number of items in the batch.
func (m *MockClaimableBalanceBatchInsertBuilder) Len() int {
	a := m.Called()
	return a.Int(0)
}