(ctx context.Context, ids []string)
| 278 | } |
| 279 | |
| 280 | func (m *MockDataSource) GetBalancesByIDsLite(ctx context.Context, ids []string) (map[string]*model.Balance, error) { |
| 281 | args := m.Called(ctx, ids) |
| 282 | if args.Get(0) == nil { |
| 283 | return nil, args.Error(1) |
| 284 | } |
| 285 | return args.Get(0).(map[string]*model.Balance), args.Error(1) |
| 286 | } |
| 287 | |
| 288 | func (m *MockDataSource) GetAllBalances(limit, offset int) ([]model.Balance, error) { |
| 289 | args := m.Called(limit, offset) |