forked from aquasecurity/fanal
/
mock_registry.go
123 lines (102 loc) · 2.87 KB
/
mock_registry.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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
// Code generated by mockery v1.0.0. DO NOT EDIT.
package image
import context "context"
import mock "github.com/stretchr/testify/mock"
import types "github.com/goodwithtech/deckoder/types"
// MockRegistry is an autogenerated mock type for the Registry type
type MockRegistry struct {
mock.Mock
}
type RegistryCheckOptionsArgs struct {
Domain string
DomainAnything bool
Option types.DockerOption
OptionAnything bool
}
type RegistryCheckOptionsReturns struct {
_a0 error
}
type RegistryCheckOptionsExpectation struct {
Args RegistryCheckOptionsArgs
Returns RegistryCheckOptionsReturns
}
func (_m *MockRegistry) ApplyCheckOptionsExpectation(e RegistryCheckOptionsExpectation) {
var args []interface{}
if e.Args.DomainAnything {
args = append(args, mock.Anything)
} else {
args = append(args, e.Args.Domain)
}
if e.Args.OptionAnything {
args = append(args, mock.Anything)
} else {
args = append(args, e.Args.Option)
}
_m.On("CheckOptions", args...).Return(e.Returns._a0)
}
func (_m *MockRegistry) ApplyCheckOptionsExpectations(expectations []RegistryCheckOptionsExpectation) {
for _, e := range expectations {
_m.ApplyCheckOptionsExpectation(e)
}
}
// CheckOptions provides a mock function with given fields: domain, option
func (_m *MockRegistry) CheckOptions(domain string, option types.DockerOption) error {
ret := _m.Called(domain, option)
var r0 error
if rf, ok := ret.Get(0).(func(string, types.DockerOption) error); ok {
r0 = rf(domain, option)
} else {
r0 = ret.Error(0)
}
return r0
}
type RegistryGetCredentialArgs struct {
Ctx context.Context
CtxAnything bool
}
type RegistryGetCredentialReturns struct {
_a0 string
_a1 string
_a2 error
}
type RegistryGetCredentialExpectation struct {
Args RegistryGetCredentialArgs
Returns RegistryGetCredentialReturns
}
func (_m *MockRegistry) ApplyGetCredentialExpectation(e RegistryGetCredentialExpectation) {
var args []interface{}
if e.Args.CtxAnything {
args = append(args, mock.Anything)
} else {
args = append(args, e.Args.Ctx)
}
_m.On("GetCredential", args...).Return(e.Returns._a0, e.Returns._a1, e.Returns._a2)
}
func (_m *MockRegistry) ApplyGetCredentialExpectations(expectations []RegistryGetCredentialExpectation) {
for _, e := range expectations {
_m.ApplyGetCredentialExpectation(e)
}
}
// GetCredential provides a mock function with given fields: ctx
func (_m *MockRegistry) GetCredential(ctx context.Context) (string, string, error) {
ret := _m.Called(ctx)
var r0 string
if rf, ok := ret.Get(0).(func(context.Context) string); ok {
r0 = rf(ctx)
} else {
r0 = ret.Get(0).(string)
}
var r1 string
if rf, ok := ret.Get(1).(func(context.Context) string); ok {
r1 = rf(ctx)
} else {
r1 = ret.Get(1).(string)
}
var r2 error
if rf, ok := ret.Get(2).(func(context.Context) error); ok {
r2 = rf(ctx)
} else {
r2 = ret.Error(2)
}
return r0, r1, r2
}