/
mock_getter.go
74 lines (62 loc) · 1.54 KB
/
mock_getter.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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
// Code generated by mockery v1.0.0. DO NOT EDIT.
package vulnsrctest
import (
types "github.com/khulnasoft/tunnel-db/pkg/types"
mock "github.com/stretchr/testify/mock"
)
// MockGetter is an autogenerated mock type for the Getter type
type MockGetter struct {
mock.Mock
}
type GetterGetArgs struct {
_a0 string
_a0Anything bool
_a1 string
_a1Anything bool
}
type GetterGetReturns struct {
_a0 []types.Advisory
_a1 error
}
type GetterGetExpectation struct {
Args GetterGetArgs
Returns GetterGetReturns
}
func (_m *MockGetter) ApplyGetExpectation(e GetterGetExpectation) {
var args []interface{}
if e.Args._a0Anything {
args = append(args, mock.Anything)
} else {
args = append(args, e.Args._a0)
}
if e.Args._a1Anything {
args = append(args, mock.Anything)
} else {
args = append(args, e.Args._a1)
}
_m.On("Get", args...).Return(e.Returns._a0, e.Returns._a1)
}
func (_m *MockGetter) ApplyGetExpectations(expectations []GetterGetExpectation) {
for _, e := range expectations {
_m.ApplyGetExpectation(e)
}
}
// Get provides a mock function with given fields: _a0, _a1
func (_m *MockGetter) Get(_a0 string, _a1 string) ([]types.Advisory, error) {
ret := _m.Called(_a0, _a1)
var r0 []types.Advisory
if rf, ok := ret.Get(0).(func(string, string) []types.Advisory); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]types.Advisory)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(string, string) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}