github.com/masterhung0112/hk_server/v5@v5.0.0-20220302090640-ec71aef15e1c/store/storetest/mocks/FileInfoStore.go (about) 1 // Code generated by mockery v1.0.0. DO NOT EDIT. 2 3 // 'Regenerate 4 5 package mocks 6 7 import ( 8 model "github.com/masterhung0112/hk_server/v5/model" 9 mock "github.com/stretchr/testify/mock" 10 ) 11 12 // FileInfoStore is an autogenerated mock type for the FileInfoStore type 13 type FileInfoStore struct { 14 mock.Mock 15 } 16 17 // AttachToPost provides a mock function with given fields: fileID, postID, creatorID 18 func (_m *FileInfoStore) AttachToPost(fileID string, postID string, creatorID string) error { 19 ret := _m.Called(fileID, postID, creatorID) 20 21 var r0 error 22 if rf, ok := ret.Get(0).(func(string, string, string) error); ok { 23 r0 = rf(fileID, postID, creatorID) 24 } else { 25 r0 = ret.Error(0) 26 } 27 28 return r0 29 } 30 31 // ClearCaches provides a mock function with given fields: 32 func (_m *FileInfoStore) ClearCaches() { 33 _m.Called() 34 } 35 36 // CountAll provides a mock function with given fields: 37 func (_m *FileInfoStore) CountAll() (int64, error) { 38 ret := _m.Called() 39 40 var r0 int64 41 if rf, ok := ret.Get(0).(func() int64); ok { 42 r0 = rf() 43 } else { 44 r0 = ret.Get(0).(int64) 45 } 46 47 var r1 error 48 if rf, ok := ret.Get(1).(func() error); ok { 49 r1 = rf() 50 } else { 51 r1 = ret.Error(1) 52 } 53 54 return r0, r1 55 } 56 57 // DeleteForPost provides a mock function with given fields: postID 58 func (_m *FileInfoStore) DeleteForPost(postID string) (string, error) { 59 ret := _m.Called(postID) 60 61 var r0 string 62 if rf, ok := ret.Get(0).(func(string) string); ok { 63 r0 = rf(postID) 64 } else { 65 r0 = ret.Get(0).(string) 66 } 67 68 var r1 error 69 if rf, ok := ret.Get(1).(func(string) error); ok { 70 r1 = rf(postID) 71 } else { 72 r1 = ret.Error(1) 73 } 74 75 return r0, r1 76 } 77 78 // Get provides a mock function with given fields: id 79 func (_m *FileInfoStore) Get(id string) (*model.FileInfo, error) { 80 ret := _m.Called(id) 81 82 var r0 *model.FileInfo 83 if rf, ok := ret.Get(0).(func(string) *model.FileInfo); ok { 84 r0 = rf(id) 85 } else { 86 if ret.Get(0) != nil { 87 r0 = ret.Get(0).(*model.FileInfo) 88 } 89 } 90 91 var r1 error 92 if rf, ok := ret.Get(1).(func(string) error); ok { 93 r1 = rf(id) 94 } else { 95 r1 = ret.Error(1) 96 } 97 98 return r0, r1 99 } 100 101 // GetByIds provides a mock function with given fields: ids 102 func (_m *FileInfoStore) GetByIds(ids []string) ([]*model.FileInfo, error) { 103 ret := _m.Called(ids) 104 105 var r0 []*model.FileInfo 106 if rf, ok := ret.Get(0).(func([]string) []*model.FileInfo); ok { 107 r0 = rf(ids) 108 } else { 109 if ret.Get(0) != nil { 110 r0 = ret.Get(0).([]*model.FileInfo) 111 } 112 } 113 114 var r1 error 115 if rf, ok := ret.Get(1).(func([]string) error); ok { 116 r1 = rf(ids) 117 } else { 118 r1 = ret.Error(1) 119 } 120 121 return r0, r1 122 } 123 124 // GetByPath provides a mock function with given fields: path 125 func (_m *FileInfoStore) GetByPath(path string) (*model.FileInfo, error) { 126 ret := _m.Called(path) 127 128 var r0 *model.FileInfo 129 if rf, ok := ret.Get(0).(func(string) *model.FileInfo); ok { 130 r0 = rf(path) 131 } else { 132 if ret.Get(0) != nil { 133 r0 = ret.Get(0).(*model.FileInfo) 134 } 135 } 136 137 var r1 error 138 if rf, ok := ret.Get(1).(func(string) error); ok { 139 r1 = rf(path) 140 } else { 141 r1 = ret.Error(1) 142 } 143 144 return r0, r1 145 } 146 147 // GetFilesBatchForIndexing provides a mock function with given fields: startTime, endTime, limit 148 func (_m *FileInfoStore) GetFilesBatchForIndexing(startTime int64, endTime int64, limit int) ([]*model.FileForIndexing, error) { 149 ret := _m.Called(startTime, endTime, limit) 150 151 var r0 []*model.FileForIndexing 152 if rf, ok := ret.Get(0).(func(int64, int64, int) []*model.FileForIndexing); ok { 153 r0 = rf(startTime, endTime, limit) 154 } else { 155 if ret.Get(0) != nil { 156 r0 = ret.Get(0).([]*model.FileForIndexing) 157 } 158 } 159 160 var r1 error 161 if rf, ok := ret.Get(1).(func(int64, int64, int) error); ok { 162 r1 = rf(startTime, endTime, limit) 163 } else { 164 r1 = ret.Error(1) 165 } 166 167 return r0, r1 168 } 169 170 // GetForPost provides a mock function with given fields: postID, readFromMaster, includeDeleted, allowFromCache 171 func (_m *FileInfoStore) GetForPost(postID string, readFromMaster bool, includeDeleted bool, allowFromCache bool) ([]*model.FileInfo, error) { 172 ret := _m.Called(postID, readFromMaster, includeDeleted, allowFromCache) 173 174 var r0 []*model.FileInfo 175 if rf, ok := ret.Get(0).(func(string, bool, bool, bool) []*model.FileInfo); ok { 176 r0 = rf(postID, readFromMaster, includeDeleted, allowFromCache) 177 } else { 178 if ret.Get(0) != nil { 179 r0 = ret.Get(0).([]*model.FileInfo) 180 } 181 } 182 183 var r1 error 184 if rf, ok := ret.Get(1).(func(string, bool, bool, bool) error); ok { 185 r1 = rf(postID, readFromMaster, includeDeleted, allowFromCache) 186 } else { 187 r1 = ret.Error(1) 188 } 189 190 return r0, r1 191 } 192 193 // GetForUser provides a mock function with given fields: userID 194 func (_m *FileInfoStore) GetForUser(userID string) ([]*model.FileInfo, error) { 195 ret := _m.Called(userID) 196 197 var r0 []*model.FileInfo 198 if rf, ok := ret.Get(0).(func(string) []*model.FileInfo); ok { 199 r0 = rf(userID) 200 } else { 201 if ret.Get(0) != nil { 202 r0 = ret.Get(0).([]*model.FileInfo) 203 } 204 } 205 206 var r1 error 207 if rf, ok := ret.Get(1).(func(string) error); ok { 208 r1 = rf(userID) 209 } else { 210 r1 = ret.Error(1) 211 } 212 213 return r0, r1 214 } 215 216 // GetFromMaster provides a mock function with given fields: id 217 func (_m *FileInfoStore) GetFromMaster(id string) (*model.FileInfo, error) { 218 ret := _m.Called(id) 219 220 var r0 *model.FileInfo 221 if rf, ok := ret.Get(0).(func(string) *model.FileInfo); ok { 222 r0 = rf(id) 223 } else { 224 if ret.Get(0) != nil { 225 r0 = ret.Get(0).(*model.FileInfo) 226 } 227 } 228 229 var r1 error 230 if rf, ok := ret.Get(1).(func(string) error); ok { 231 r1 = rf(id) 232 } else { 233 r1 = ret.Error(1) 234 } 235 236 return r0, r1 237 } 238 239 // GetWithOptions provides a mock function with given fields: page, perPage, opt 240 func (_m *FileInfoStore) GetWithOptions(page int, perPage int, opt *model.GetFileInfosOptions) ([]*model.FileInfo, error) { 241 ret := _m.Called(page, perPage, opt) 242 243 var r0 []*model.FileInfo 244 if rf, ok := ret.Get(0).(func(int, int, *model.GetFileInfosOptions) []*model.FileInfo); ok { 245 r0 = rf(page, perPage, opt) 246 } else { 247 if ret.Get(0) != nil { 248 r0 = ret.Get(0).([]*model.FileInfo) 249 } 250 } 251 252 var r1 error 253 if rf, ok := ret.Get(1).(func(int, int, *model.GetFileInfosOptions) error); ok { 254 r1 = rf(page, perPage, opt) 255 } else { 256 r1 = ret.Error(1) 257 } 258 259 return r0, r1 260 } 261 262 // InvalidateFileInfosForPostCache provides a mock function with given fields: postID, deleted 263 func (_m *FileInfoStore) InvalidateFileInfosForPostCache(postID string, deleted bool) { 264 _m.Called(postID, deleted) 265 } 266 267 // PermanentDelete provides a mock function with given fields: fileID 268 func (_m *FileInfoStore) PermanentDelete(fileID string) error { 269 ret := _m.Called(fileID) 270 271 var r0 error 272 if rf, ok := ret.Get(0).(func(string) error); ok { 273 r0 = rf(fileID) 274 } else { 275 r0 = ret.Error(0) 276 } 277 278 return r0 279 } 280 281 // PermanentDeleteBatch provides a mock function with given fields: endTime, limit 282 func (_m *FileInfoStore) PermanentDeleteBatch(endTime int64, limit int64) (int64, error) { 283 ret := _m.Called(endTime, limit) 284 285 var r0 int64 286 if rf, ok := ret.Get(0).(func(int64, int64) int64); ok { 287 r0 = rf(endTime, limit) 288 } else { 289 r0 = ret.Get(0).(int64) 290 } 291 292 var r1 error 293 if rf, ok := ret.Get(1).(func(int64, int64) error); ok { 294 r1 = rf(endTime, limit) 295 } else { 296 r1 = ret.Error(1) 297 } 298 299 return r0, r1 300 } 301 302 // PermanentDeleteByUser provides a mock function with given fields: userID 303 func (_m *FileInfoStore) PermanentDeleteByUser(userID string) (int64, error) { 304 ret := _m.Called(userID) 305 306 var r0 int64 307 if rf, ok := ret.Get(0).(func(string) int64); ok { 308 r0 = rf(userID) 309 } else { 310 r0 = ret.Get(0).(int64) 311 } 312 313 var r1 error 314 if rf, ok := ret.Get(1).(func(string) error); ok { 315 r1 = rf(userID) 316 } else { 317 r1 = ret.Error(1) 318 } 319 320 return r0, r1 321 } 322 323 // Save provides a mock function with given fields: info 324 func (_m *FileInfoStore) Save(info *model.FileInfo) (*model.FileInfo, error) { 325 ret := _m.Called(info) 326 327 var r0 *model.FileInfo 328 if rf, ok := ret.Get(0).(func(*model.FileInfo) *model.FileInfo); ok { 329 r0 = rf(info) 330 } else { 331 if ret.Get(0) != nil { 332 r0 = ret.Get(0).(*model.FileInfo) 333 } 334 } 335 336 var r1 error 337 if rf, ok := ret.Get(1).(func(*model.FileInfo) error); ok { 338 r1 = rf(info) 339 } else { 340 r1 = ret.Error(1) 341 } 342 343 return r0, r1 344 } 345 346 // Search provides a mock function with given fields: paramsList, userID, teamID, page, perPage 347 func (_m *FileInfoStore) Search(paramsList []*model.SearchParams, userID string, teamID string, page int, perPage int) (*model.FileInfoList, error) { 348 ret := _m.Called(paramsList, userID, teamID, page, perPage) 349 350 var r0 *model.FileInfoList 351 if rf, ok := ret.Get(0).(func([]*model.SearchParams, string, string, int, int) *model.FileInfoList); ok { 352 r0 = rf(paramsList, userID, teamID, page, perPage) 353 } else { 354 if ret.Get(0) != nil { 355 r0 = ret.Get(0).(*model.FileInfoList) 356 } 357 } 358 359 var r1 error 360 if rf, ok := ret.Get(1).(func([]*model.SearchParams, string, string, int, int) error); ok { 361 r1 = rf(paramsList, userID, teamID, page, perPage) 362 } else { 363 r1 = ret.Error(1) 364 } 365 366 return r0, r1 367 } 368 369 // SetContent provides a mock function with given fields: fileID, content 370 func (_m *FileInfoStore) SetContent(fileID string, content string) error { 371 ret := _m.Called(fileID, content) 372 373 var r0 error 374 if rf, ok := ret.Get(0).(func(string, string) error); ok { 375 r0 = rf(fileID, content) 376 } else { 377 r0 = ret.Error(0) 378 } 379 380 return r0 381 } 382 383 // Upsert provides a mock function with given fields: info 384 func (_m *FileInfoStore) Upsert(info *model.FileInfo) (*model.FileInfo, error) { 385 ret := _m.Called(info) 386 387 var r0 *model.FileInfo 388 if rf, ok := ret.Get(0).(func(*model.FileInfo) *model.FileInfo); ok { 389 r0 = rf(info) 390 } else { 391 if ret.Get(0) != nil { 392 r0 = ret.Get(0).(*model.FileInfo) 393 } 394 } 395 396 var r1 error 397 if rf, ok := ret.Get(1).(func(*model.FileInfo) error); ok { 398 r1 = rf(info) 399 } else { 400 r1 = ret.Error(1) 401 } 402 403 return r0, r1 404 }