-
Notifications
You must be signed in to change notification settings - Fork 496
/
mock_q_filter.go
32 lines (25 loc) · 961 Bytes
/
mock_q_filter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package history
import (
"context"
"github.com/stretchr/testify/mock"
)
// MockQAccountFilterWhitelist is a mock implementation of the QAccountFilterWhitelist interface
type MockQFilter struct {
mock.Mock
}
func (m *MockQFilter) GetAccountFilterConfig(ctx context.Context) (AccountFilterConfig, error) {
a := m.Called(ctx)
return a.Get(0).(AccountFilterConfig), a.Error(1)
}
func (m *MockQFilter) GetAssetFilterConfig(ctx context.Context) (AssetFilterConfig, error) {
a := m.Called(ctx)
return a.Get(0).(AssetFilterConfig), a.Error(1)
}
func (m *MockQFilter) UpdateAccountFilterConfig(ctx context.Context, config AccountFilterConfig) (AccountFilterConfig, error) {
a := m.Called(ctx, config)
return a.Get(0).(AccountFilterConfig), a.Error(0)
}
func (m *MockQFilter) UpdateAssetFilterConfig(ctx context.Context, config AssetFilterConfig) (AssetFilterConfig, error) {
a := m.Called(ctx, config)
return a.Get(0).(AssetFilterConfig), a.Error(0)
}