1// Code generated by mockery; DO NOT EDIT.
2// github.com/vektra/mockery
3// template: testify
4
5package mocks
6
7import (
8 "context"
9
10 "github.com/bnema/dumber/internal/domain/entity"
11 mock "github.com/stretchr/testify/mock"
12)
13
14// NewMockSessionStateRepository creates a new instance of MockSessionStateRepository. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
15// The first argument is typically a *testing.T value.
16func NewMockSessionStateRepository(t interface {
17 mock.TestingT
18 Cleanup(func())
19}) *MockSessionStateRepository {
20 mock := &MockSessionStateRepository{}
21 mock.Mock.Test(t)
22
23 t.Cleanup(func() { mock.AssertExpectations(t) })
24
25 return mock
26}
27
28// MockSessionStateRepository is an autogenerated mock type for the SessionStateRepository type
29type MockSessionStateRepository struct {
30 mock.Mock
31}
32
33type MockSessionStateRepository_Expecter struct {
34 mock *mock.Mock
35}
36
37func (_m *MockSessionStateRepository) EXPECT() *MockSessionStateRepository_Expecter {
38 return &MockSessionStateRepository_Expecter{mock: &_m.Mock}
39}
40
41// DeleteSnapshot provides a mock function for the type MockSessionStateRepository
42func (_mock *MockSessionStateRepository) DeleteSnapshot(ctx context.Context, sessionID entity.SessionID) error {
43 ret := _mock.Called(ctx, sessionID)
44
45 if len(ret) == 0 {
46 panic("no return value specified for DeleteSnapshot")
47 }
48
49 var r0 error
50 if returnFunc, ok := ret.Get(0).(func(context.Context, entity.SessionID) error); ok {
51 r0 = returnFunc(ctx, sessionID)
52 } else {
53 r0 = ret.Error(0)
54 }
55 return r0
56}
57
58// MockSessionStateRepository_DeleteSnapshot_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteSnapshot'
59type MockSessionStateRepository_DeleteSnapshot_Call struct {
60 *mock.Call
61}
62
63// DeleteSnapshot is a helper method to define mock.On call
64// - ctx context.Context
65// - sessionID entity.SessionID
66func (_e *MockSessionStateRepository_Expecter) DeleteSnapshot(ctx any, sessionID any) *MockSessionStateRepository_DeleteSnapshot_Call {
67 return &MockSessionStateRepository_DeleteSnapshot_Call{Call: _e.mock.On("DeleteSnapshot", ctx, sessionID)}
68}
69
70func (_c *MockSessionStateRepository_DeleteSnapshot_Call) Run(run func(ctx context.Context, sessionID entity.SessionID)) *MockSessionStateRepository_DeleteSnapshot_Call {
71 _c.Call.Run(func(args mock.Arguments) {
72 var arg0 context.Context
73 if args[0] != nil {
74 arg0 = args[0].(context.Context)
75 }
76 var arg1 entity.SessionID
77 if args[1] != nil {
78 arg1 = args[1].(entity.SessionID)
79 }
80 run(
81 arg0,
82 arg1,
83 )
84 })
85 return _c
86}
87
88func (_c *MockSessionStateRepository_DeleteSnapshot_Call) Return(err error) *MockSessionStateRepository_DeleteSnapshot_Call {
89 _c.Call.Return(err)
90 return _c
91}
92
93func (_c *MockSessionStateRepository_DeleteSnapshot_Call) RunAndReturn(run func(ctx context.Context, sessionID entity.SessionID) error) *MockSessionStateRepository_DeleteSnapshot_Call {
94 _c.Call.Return(run)
95 return _c
96}
97
98// GetAllSnapshots provides a mock function for the type MockSessionStateRepository
99func (_mock *MockSessionStateRepository) GetAllSnapshots(ctx context.Context) ([]*entity.SessionState, error) {
100 ret := _mock.Called(ctx)
101
102 if len(ret) == 0 {
103 panic("no return value specified for GetAllSnapshots")
104 }
105
106 var r0 []*entity.SessionState
107 var r1 error
108 if returnFunc, ok := ret.Get(0).(func(context.Context) ([]*entity.SessionState, error)); ok {
109 return returnFunc(ctx)
110 }
111 if returnFunc, ok := ret.Get(0).(func(context.Context) []*entity.SessionState); ok {
112 r0 = returnFunc(ctx)
113 } else {
114 if ret.Get(0) != nil {
115 r0 = ret.Get(0).([]*entity.SessionState)
116 }
117 }
118 if returnFunc, ok := ret.Get(1).(func(context.Context) error); ok {
119 r1 = returnFunc(ctx)
120 } else {
121 r1 = ret.Error(1)
122 }
123 return r0, r1
124}
125
126// MockSessionStateRepository_GetAllSnapshots_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetAllSnapshots'
127type MockSessionStateRepository_GetAllSnapshots_Call struct {
128 *mock.Call
129}
130
131// GetAllSnapshots is a helper method to define mock.On call
132// - ctx context.Context
133func (_e *MockSessionStateRepository_Expecter) GetAllSnapshots(ctx any) *MockSessionStateRepository_GetAllSnapshots_Call {
134 return &MockSessionStateRepository_GetAllSnapshots_Call{Call: _e.mock.On("GetAllSnapshots", ctx)}
135}
136
137func (_c *MockSessionStateRepository_GetAllSnapshots_Call) Run(run func(ctx context.Context)) *MockSessionStateRepository_GetAllSnapshots_Call {
138 _c.Call.Run(func(args mock.Arguments) {
139 var arg0 context.Context
140 if args[0] != nil {
141 arg0 = args[0].(context.Context)
142 }
143 run(
144 arg0,
145 )
146 })
147 return _c
148}
149
150func (_c *MockSessionStateRepository_GetAllSnapshots_Call) Return(sessionStates []*entity.SessionState, err error) *MockSessionStateRepository_GetAllSnapshots_Call {
151 _c.Call.Return(sessionStates, err)
152 return _c
153}
154
155func (_c *MockSessionStateRepository_GetAllSnapshots_Call) RunAndReturn(run func(ctx context.Context) ([]*entity.SessionState, error)) *MockSessionStateRepository_GetAllSnapshots_Call {
156 _c.Call.Return(run)
157 return _c
158}
159
160// GetSnapshot provides a mock function for the type MockSessionStateRepository
161func (_mock *MockSessionStateRepository) GetSnapshot(ctx context.Context, sessionID entity.SessionID) (*entity.SessionState, error) {
162 ret := _mock.Called(ctx, sessionID)
163
164 if len(ret) == 0 {
165 panic("no return value specified for GetSnapshot")
166 }
167
168 var r0 *entity.SessionState
169 var r1 error
170 if returnFunc, ok := ret.Get(0).(func(context.Context, entity.SessionID) (*entity.SessionState, error)); ok {
171 return returnFunc(ctx, sessionID)
172 }
173 if returnFunc, ok := ret.Get(0).(func(context.Context, entity.SessionID) *entity.SessionState); ok {
174 r0 = returnFunc(ctx, sessionID)
175 } else {
176 if ret.Get(0) != nil {
177 r0 = ret.Get(0).(*entity.SessionState)
178 }
179 }
180 if returnFunc, ok := ret.Get(1).(func(context.Context, entity.SessionID) error); ok {
181 r1 = returnFunc(ctx, sessionID)
182 } else {
183 r1 = ret.Error(1)
184 }
185 return r0, r1
186}
187
188// MockSessionStateRepository_GetSnapshot_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetSnapshot'
189type MockSessionStateRepository_GetSnapshot_Call struct {
190 *mock.Call
191}
192
193// GetSnapshot is a helper method to define mock.On call
194// - ctx context.Context
195// - sessionID entity.SessionID
196func (_e *MockSessionStateRepository_Expecter) GetSnapshot(ctx any, sessionID any) *MockSessionStateRepository_GetSnapshot_Call {
197 return &MockSessionStateRepository_GetSnapshot_Call{Call: _e.mock.On("GetSnapshot", ctx, sessionID)}
198}
199
200func (_c *MockSessionStateRepository_GetSnapshot_Call) Run(run func(ctx context.Context, sessionID entity.SessionID)) *MockSessionStateRepository_GetSnapshot_Call {
201 _c.Call.Run(func(args mock.Arguments) {
202 var arg0 context.Context
203 if args[0] != nil {
204 arg0 = args[0].(context.Context)
205 }
206 var arg1 entity.SessionID
207 if args[1] != nil {
208 arg1 = args[1].(entity.SessionID)
209 }
210 run(
211 arg0,
212 arg1,
213 )
214 })
215 return _c
216}
217
218func (_c *MockSessionStateRepository_GetSnapshot_Call) Return(sessionState *entity.SessionState, err error) *MockSessionStateRepository_GetSnapshot_Call {
219 _c.Call.Return(sessionState, err)
220 return _c
221}
222
223func (_c *MockSessionStateRepository_GetSnapshot_Call) RunAndReturn(run func(ctx context.Context, sessionID entity.SessionID) (*entity.SessionState, error)) *MockSessionStateRepository_GetSnapshot_Call {
224 _c.Call.Return(run)
225 return _c
226}
227
228// GetTotalSnapshotsSize provides a mock function for the type MockSessionStateRepository
229func (_mock *MockSessionStateRepository) GetTotalSnapshotsSize(ctx context.Context) (int64, error) {
230 ret := _mock.Called(ctx)
231
232 if len(ret) == 0 {
233 panic("no return value specified for GetTotalSnapshotsSize")
234 }
235
236 var r0 int64
237 var r1 error
238 if returnFunc, ok := ret.Get(0).(func(context.Context) (int64, error)); ok {
239 return returnFunc(ctx)
240 }
241 if returnFunc, ok := ret.Get(0).(func(context.Context) int64); ok {
242 r0 = returnFunc(ctx)
243 } else {
244 r0 = ret.Get(0).(int64)
245 }
246 if returnFunc, ok := ret.Get(1).(func(context.Context) error); ok {
247 r1 = returnFunc(ctx)
248 } else {
249 r1 = ret.Error(1)
250 }
251 return r0, r1
252}
253
254// MockSessionStateRepository_GetTotalSnapshotsSize_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetTotalSnapshotsSize'
255type MockSessionStateRepository_GetTotalSnapshotsSize_Call struct {
256 *mock.Call
257}
258
259// GetTotalSnapshotsSize is a helper method to define mock.On call
260// - ctx context.Context
261func (_e *MockSessionStateRepository_Expecter) GetTotalSnapshotsSize(ctx any) *MockSessionStateRepository_GetTotalSnapshotsSize_Call {
262 return &MockSessionStateRepository_GetTotalSnapshotsSize_Call{Call: _e.mock.On("GetTotalSnapshotsSize", ctx)}
263}
264
265func (_c *MockSessionStateRepository_GetTotalSnapshotsSize_Call) Run(run func(ctx context.Context)) *MockSessionStateRepository_GetTotalSnapshotsSize_Call {
266 _c.Call.Run(func(args mock.Arguments) {
267 var arg0 context.Context
268 if args[0] != nil {
269 arg0 = args[0].(context.Context)
270 }
271 run(
272 arg0,
273 )
274 })
275 return _c
276}
277
278func (_c *MockSessionStateRepository_GetTotalSnapshotsSize_Call) Return(n int64, err error) *MockSessionStateRepository_GetTotalSnapshotsSize_Call {
279 _c.Call.Return(n, err)
280 return _c
281}
282
283func (_c *MockSessionStateRepository_GetTotalSnapshotsSize_Call) RunAndReturn(run func(ctx context.Context) (int64, error)) *MockSessionStateRepository_GetTotalSnapshotsSize_Call {
284 _c.Call.Return(run)
285 return _c
286}
287
288// SaveSnapshot provides a mock function for the type MockSessionStateRepository
289func (_mock *MockSessionStateRepository) SaveSnapshot(ctx context.Context, state *entity.SessionState) error {
290 ret := _mock.Called(ctx, state)
291
292 if len(ret) == 0 {
293 panic("no return value specified for SaveSnapshot")
294 }
295
296 var r0 error
297 if returnFunc, ok := ret.Get(0).(func(context.Context, *entity.SessionState) error); ok {
298 r0 = returnFunc(ctx, state)
299 } else {
300 r0 = ret.Error(0)
301 }
302 return r0
303}
304
305// MockSessionStateRepository_SaveSnapshot_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SaveSnapshot'
306type MockSessionStateRepository_SaveSnapshot_Call struct {
307 *mock.Call
308}
309
310// SaveSnapshot is a helper method to define mock.On call
311// - ctx context.Context
312// - state *entity.SessionState
313func (_e *MockSessionStateRepository_Expecter) SaveSnapshot(ctx any, state any) *MockSessionStateRepository_SaveSnapshot_Call {
314 return &MockSessionStateRepository_SaveSnapshot_Call{Call: _e.mock.On("SaveSnapshot", ctx, state)}
315}
316
317func (_c *MockSessionStateRepository_SaveSnapshot_Call) Run(run func(ctx context.Context, state *entity.SessionState)) *MockSessionStateRepository_SaveSnapshot_Call {
318 _c.Call.Run(func(args mock.Arguments) {
319 var arg0 context.Context
320 if args[0] != nil {
321 arg0 = args[0].(context.Context)
322 }
323 var arg1 *entity.SessionState
324 if args[1] != nil {
325 arg1 = args[1].(*entity.SessionState)
326 }
327 run(
328 arg0,
329 arg1,
330 )
331 })
332 return _c
333}
334
335func (_c *MockSessionStateRepository_SaveSnapshot_Call) Return(err error) *MockSessionStateRepository_SaveSnapshot_Call {
336 _c.Call.Return(err)
337 return _c
338}
339
340func (_c *MockSessionStateRepository_SaveSnapshot_Call) RunAndReturn(run func(ctx context.Context, state *entity.SessionState) error) *MockSessionStateRepository_SaveSnapshot_Call {
341 _c.Call.Return(run)
342 return _c
343}