(ctx context.Context, balanceID string)
| 621 | } |
| 622 | |
| 623 | func (m *MockDataSource) GetLineageMappings(ctx context.Context, balanceID string) ([]model.LineageMapping, error) { |
| 624 | args := m.Called(ctx, balanceID) |
| 625 | return args.Get(0).([]model.LineageMapping), args.Error(1) |
| 626 | } |
| 627 | |
| 628 | func (m *MockDataSource) GetLineageMappingByProvider(ctx context.Context, balanceID, provider string) (*model.LineageMapping, error) { |
| 629 | args := m.Called(ctx, balanceID, provider) |