(ctx context.Context, batchSize int)
| 564 | } |
| 565 | |
| 566 | func (m *MockDataSource) TakeBalanceSnapshots(ctx context.Context, batchSize int) (int, error) { |
| 567 | args := m.Called(ctx, batchSize) |
| 568 | return args.Get(0).(int), args.Error(1) |
| 569 | } |
| 570 | |
| 571 | func (m *MockDataSource) CreateAPIKey(ctx context.Context, name, ownerID string, scopes []string, expiresAt time.Time) (*model.APIKey, error) { |
| 572 | args := m.Called(ctx, name, ownerID, scopes, expiresAt) |