1
2
3
4
5 package mocks
6
7 import (
8 context "context"
9 reflect "reflect"
10
11 containerd "edge-infra.dev/pkg/sds/lib/containerd"
12 containerd0 "github.com/containerd/containerd"
13 gomock "github.com/golang/mock/gomock"
14 )
15
16
17 type MockClient struct {
18 ctrl *gomock.Controller
19 recorder *MockClientMockRecorder
20 }
21
22
23 type MockClientMockRecorder struct {
24 mock *MockClient
25 }
26
27
28 func NewMockClient(ctrl *gomock.Controller) *MockClient {
29 mock := &MockClient{ctrl: ctrl}
30 mock.recorder = &MockClientMockRecorder{mock}
31 return mock
32 }
33
34
35 func (m *MockClient) EXPECT() *MockClientMockRecorder {
36 return m.recorder
37 }
38
39
40 func (m *MockClient) Close() error {
41 m.ctrl.T.Helper()
42 ret := m.ctrl.Call(m, "Close")
43 ret0, _ := ret[0].(error)
44 return ret0
45 }
46
47
48 func (mr *MockClientMockRecorder) Close() *gomock.Call {
49 mr.mock.ctrl.T.Helper()
50 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockClient)(nil).Close))
51 }
52
53
54 func (m *MockClient) DeleteContainer(arg0 context.Context, arg1 containerd0.Container, arg2 ...containerd0.DeleteOpts) error {
55 m.ctrl.T.Helper()
56 varargs := []interface{}{arg0, arg1}
57 for _, a := range arg2 {
58 varargs = append(varargs, a)
59 }
60 ret := m.ctrl.Call(m, "DeleteContainer", varargs...)
61 ret0, _ := ret[0].(error)
62 return ret0
63 }
64
65
66 func (mr *MockClientMockRecorder) DeleteContainer(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
67 mr.mock.ctrl.T.Helper()
68 varargs := append([]interface{}{arg0, arg1}, arg2...)
69 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteContainer", reflect.TypeOf((*MockClient)(nil).DeleteContainer), varargs...)
70 }
71
72
73 func (m *MockClient) GetContainersByLabels(arg0 context.Context, arg1 containerd.LabelFilters) ([]containerd0.Container, error) {
74 m.ctrl.T.Helper()
75 ret := m.ctrl.Call(m, "GetContainersByLabels", arg0, arg1)
76 ret0, _ := ret[0].([]containerd0.Container)
77 ret1, _ := ret[1].(error)
78 return ret0, ret1
79 }
80
81
82 func (mr *MockClientMockRecorder) GetContainersByLabels(arg0, arg1 interface{}) *gomock.Call {
83 mr.mock.ctrl.T.Helper()
84 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetContainersByLabels", reflect.TypeOf((*MockClient)(nil).GetContainersByLabels), arg0, arg1)
85 }
86
87
88 func (m *MockClient) KillTask(arg0 context.Context, arg1 containerd0.Task) error {
89 m.ctrl.T.Helper()
90 ret := m.ctrl.Call(m, "KillTask", arg0, arg1)
91 ret0, _ := ret[0].(error)
92 return ret0
93 }
94
95
96 func (mr *MockClientMockRecorder) KillTask(arg0, arg1 interface{}) *gomock.Call {
97 mr.mock.ctrl.T.Helper()
98 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KillTask", reflect.TypeOf((*MockClient)(nil).KillTask), arg0, arg1)
99 }
100
101
102 func (m *MockClient) TerminateContainer(arg0 context.Context, arg1 containerd0.Container) error {
103 m.ctrl.T.Helper()
104 ret := m.ctrl.Call(m, "TerminateContainer", arg0, arg1)
105 ret0, _ := ret[0].(error)
106 return ret0
107 }
108
109
110 func (mr *MockClientMockRecorder) TerminateContainer(arg0, arg1 interface{}) *gomock.Call {
111 mr.mock.ctrl.T.Helper()
112 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TerminateContainer", reflect.TypeOf((*MockClient)(nil).TerminateContainer), arg0, arg1)
113 }
114
115
116 func (m *MockClient) TerminateContainers(arg0 context.Context, arg1 bool, arg2 ...containerd0.Container) error {
117 m.ctrl.T.Helper()
118 varargs := []interface{}{arg0, arg1}
119 for _, a := range arg2 {
120 varargs = append(varargs, a)
121 }
122 ret := m.ctrl.Call(m, "TerminateContainers", varargs...)
123 ret0, _ := ret[0].(error)
124 return ret0
125 }
126
127
128 func (mr *MockClientMockRecorder) TerminateContainers(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
129 mr.mock.ctrl.T.Helper()
130 varargs := append([]interface{}{arg0, arg1}, arg2...)
131 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TerminateContainers", reflect.TypeOf((*MockClient)(nil).TerminateContainers), varargs...)
132 }
133
134
135 func (m *MockClient) WithTerminate(arg0 context.Context, arg1 containerd.LabelFilters, arg2 func() error) error {
136 m.ctrl.T.Helper()
137 ret := m.ctrl.Call(m, "WithTerminate", arg0, arg1, arg2)
138 ret0, _ := ret[0].(error)
139 return ret0
140 }
141
142
143 func (mr *MockClientMockRecorder) WithTerminate(arg0, arg1, arg2 interface{}) *gomock.Call {
144 mr.mock.ctrl.T.Helper()
145 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTerminate", reflect.TypeOf((*MockClient)(nil).WithTerminate), arg0, arg1, arg2)
146 }
147
View as plain text