github.com/masterhung0112/hk_server/v5@v5.0.0-20220302090640-ec71aef15e1c/store/storetest/mocks/ThreadStore.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 store "github.com/masterhung0112/hk_server/v5/store" 12 ) 13 14 // ThreadStore is an autogenerated mock type for the ThreadStore type 15 type ThreadStore struct { 16 mock.Mock 17 } 18 19 // CollectThreadsWithNewerReplies provides a mock function with given fields: userId, channelIds, timestamp 20 func (_m *ThreadStore) CollectThreadsWithNewerReplies(userId string, channelIds []string, timestamp int64) ([]string, error) { 21 ret := _m.Called(userId, channelIds, timestamp) 22 23 var r0 []string 24 if rf, ok := ret.Get(0).(func(string, []string, int64) []string); ok { 25 r0 = rf(userId, channelIds, timestamp) 26 } else { 27 if ret.Get(0) != nil { 28 r0 = ret.Get(0).([]string) 29 } 30 } 31 32 var r1 error 33 if rf, ok := ret.Get(1).(func(string, []string, int64) error); ok { 34 r1 = rf(userId, channelIds, timestamp) 35 } else { 36 r1 = ret.Error(1) 37 } 38 39 return r0, r1 40 } 41 42 // Delete provides a mock function with given fields: postID 43 func (_m *ThreadStore) Delete(postID string) error { 44 ret := _m.Called(postID) 45 46 var r0 error 47 if rf, ok := ret.Get(0).(func(string) error); ok { 48 r0 = rf(postID) 49 } else { 50 r0 = ret.Error(0) 51 } 52 53 return r0 54 } 55 56 // DeleteMembershipForUser provides a mock function with given fields: userId, postID 57 func (_m *ThreadStore) DeleteMembershipForUser(userId string, postID string) error { 58 ret := _m.Called(userId, postID) 59 60 var r0 error 61 if rf, ok := ret.Get(0).(func(string, string) error); ok { 62 r0 = rf(userId, postID) 63 } else { 64 r0 = ret.Error(0) 65 } 66 67 return r0 68 } 69 70 // DeleteOrphanedRows provides a mock function with given fields: limit 71 func (_m *ThreadStore) DeleteOrphanedRows(limit int) (int64, error) { 72 ret := _m.Called(limit) 73 74 var r0 int64 75 if rf, ok := ret.Get(0).(func(int) int64); ok { 76 r0 = rf(limit) 77 } else { 78 r0 = ret.Get(0).(int64) 79 } 80 81 var r1 error 82 if rf, ok := ret.Get(1).(func(int) error); ok { 83 r1 = rf(limit) 84 } else { 85 r1 = ret.Error(1) 86 } 87 88 return r0, r1 89 } 90 91 // Get provides a mock function with given fields: id 92 func (_m *ThreadStore) Get(id string) (*model.Thread, error) { 93 ret := _m.Called(id) 94 95 var r0 *model.Thread 96 if rf, ok := ret.Get(0).(func(string) *model.Thread); ok { 97 r0 = rf(id) 98 } else { 99 if ret.Get(0) != nil { 100 r0 = ret.Get(0).(*model.Thread) 101 } 102 } 103 104 var r1 error 105 if rf, ok := ret.Get(1).(func(string) error); ok { 106 r1 = rf(id) 107 } else { 108 r1 = ret.Error(1) 109 } 110 111 return r0, r1 112 } 113 114 // GetMembershipForUser provides a mock function with given fields: userId, postID 115 func (_m *ThreadStore) GetMembershipForUser(userId string, postID string) (*model.ThreadMembership, error) { 116 ret := _m.Called(userId, postID) 117 118 var r0 *model.ThreadMembership 119 if rf, ok := ret.Get(0).(func(string, string) *model.ThreadMembership); ok { 120 r0 = rf(userId, postID) 121 } else { 122 if ret.Get(0) != nil { 123 r0 = ret.Get(0).(*model.ThreadMembership) 124 } 125 } 126 127 var r1 error 128 if rf, ok := ret.Get(1).(func(string, string) error); ok { 129 r1 = rf(userId, postID) 130 } else { 131 r1 = ret.Error(1) 132 } 133 134 return r0, r1 135 } 136 137 // GetMembershipsForUser provides a mock function with given fields: userId, teamID 138 func (_m *ThreadStore) GetMembershipsForUser(userId string, teamID string) ([]*model.ThreadMembership, error) { 139 ret := _m.Called(userId, teamID) 140 141 var r0 []*model.ThreadMembership 142 if rf, ok := ret.Get(0).(func(string, string) []*model.ThreadMembership); ok { 143 r0 = rf(userId, teamID) 144 } else { 145 if ret.Get(0) != nil { 146 r0 = ret.Get(0).([]*model.ThreadMembership) 147 } 148 } 149 150 var r1 error 151 if rf, ok := ret.Get(1).(func(string, string) error); ok { 152 r1 = rf(userId, teamID) 153 } else { 154 r1 = ret.Error(1) 155 } 156 157 return r0, r1 158 } 159 160 // GetPosts provides a mock function with given fields: threadID, since 161 func (_m *ThreadStore) GetPosts(threadID string, since int64) ([]*model.Post, error) { 162 ret := _m.Called(threadID, since) 163 164 var r0 []*model.Post 165 if rf, ok := ret.Get(0).(func(string, int64) []*model.Post); ok { 166 r0 = rf(threadID, since) 167 } else { 168 if ret.Get(0) != nil { 169 r0 = ret.Get(0).([]*model.Post) 170 } 171 } 172 173 var r1 error 174 if rf, ok := ret.Get(1).(func(string, int64) error); ok { 175 r1 = rf(threadID, since) 176 } else { 177 r1 = ret.Error(1) 178 } 179 180 return r0, r1 181 } 182 183 // GetThreadFollowers provides a mock function with given fields: threadID 184 func (_m *ThreadStore) GetThreadFollowers(threadID string) ([]string, error) { 185 ret := _m.Called(threadID) 186 187 var r0 []string 188 if rf, ok := ret.Get(0).(func(string) []string); ok { 189 r0 = rf(threadID) 190 } else { 191 if ret.Get(0) != nil { 192 r0 = ret.Get(0).([]string) 193 } 194 } 195 196 var r1 error 197 if rf, ok := ret.Get(1).(func(string) error); ok { 198 r1 = rf(threadID) 199 } else { 200 r1 = ret.Error(1) 201 } 202 203 return r0, r1 204 } 205 206 // GetThreadForUser provides a mock function with given fields: teamID, threadMembership, extended 207 func (_m *ThreadStore) GetThreadForUser(teamID string, threadMembership *model.ThreadMembership, extended bool) (*model.ThreadResponse, error) { 208 ret := _m.Called(teamID, threadMembership, extended) 209 210 var r0 *model.ThreadResponse 211 if rf, ok := ret.Get(0).(func(string, *model.ThreadMembership, bool) *model.ThreadResponse); ok { 212 r0 = rf(teamID, threadMembership, extended) 213 } else { 214 if ret.Get(0) != nil { 215 r0 = ret.Get(0).(*model.ThreadResponse) 216 } 217 } 218 219 var r1 error 220 if rf, ok := ret.Get(1).(func(string, *model.ThreadMembership, bool) error); ok { 221 r1 = rf(teamID, threadMembership, extended) 222 } else { 223 r1 = ret.Error(1) 224 } 225 226 return r0, r1 227 } 228 229 // GetThreadsForUser provides a mock function with given fields: userId, teamID, opts 230 func (_m *ThreadStore) GetThreadsForUser(userId string, teamID string, opts model.GetUserThreadsOpts) (*model.Threads, error) { 231 ret := _m.Called(userId, teamID, opts) 232 233 var r0 *model.Threads 234 if rf, ok := ret.Get(0).(func(string, string, model.GetUserThreadsOpts) *model.Threads); ok { 235 r0 = rf(userId, teamID, opts) 236 } else { 237 if ret.Get(0) != nil { 238 r0 = ret.Get(0).(*model.Threads) 239 } 240 } 241 242 var r1 error 243 if rf, ok := ret.Get(1).(func(string, string, model.GetUserThreadsOpts) error); ok { 244 r1 = rf(userId, teamID, opts) 245 } else { 246 r1 = ret.Error(1) 247 } 248 249 return r0, r1 250 } 251 252 // MaintainMembership provides a mock function with given fields: userID, postID, opts 253 func (_m *ThreadStore) MaintainMembership(userID string, postID string, opts store.ThreadMembershipOpts) (*model.ThreadMembership, error) { 254 ret := _m.Called(userID, postID, opts) 255 256 var r0 *model.ThreadMembership 257 if rf, ok := ret.Get(0).(func(string, string, store.ThreadMembershipOpts) *model.ThreadMembership); ok { 258 r0 = rf(userID, postID, opts) 259 } else { 260 if ret.Get(0) != nil { 261 r0 = ret.Get(0).(*model.ThreadMembership) 262 } 263 } 264 265 var r1 error 266 if rf, ok := ret.Get(1).(func(string, string, store.ThreadMembershipOpts) error); ok { 267 r1 = rf(userID, postID, opts) 268 } else { 269 r1 = ret.Error(1) 270 } 271 272 return r0, r1 273 } 274 275 // MarkAllAsRead provides a mock function with given fields: userID, teamID 276 func (_m *ThreadStore) MarkAllAsRead(userID string, teamID string) error { 277 ret := _m.Called(userID, teamID) 278 279 var r0 error 280 if rf, ok := ret.Get(0).(func(string, string) error); ok { 281 r0 = rf(userID, teamID) 282 } else { 283 r0 = ret.Error(0) 284 } 285 286 return r0 287 } 288 289 // MarkAllAsReadInChannels provides a mock function with given fields: userID, channelIDs 290 func (_m *ThreadStore) MarkAllAsReadInChannels(userID string, channelIDs []string) error { 291 ret := _m.Called(userID, channelIDs) 292 293 var r0 error 294 if rf, ok := ret.Get(0).(func(string, []string) error); ok { 295 r0 = rf(userID, channelIDs) 296 } else { 297 r0 = ret.Error(0) 298 } 299 300 return r0 301 } 302 303 // MarkAsRead provides a mock function with given fields: userID, threadID, timestamp 304 func (_m *ThreadStore) MarkAsRead(userID string, threadID string, timestamp int64) error { 305 ret := _m.Called(userID, threadID, timestamp) 306 307 var r0 error 308 if rf, ok := ret.Get(0).(func(string, string, int64) error); ok { 309 r0 = rf(userID, threadID, timestamp) 310 } else { 311 r0 = ret.Error(0) 312 } 313 314 return r0 315 } 316 317 // PermanentDeleteBatchForRetentionPolicies provides a mock function with given fields: now, globalPolicyEndTime, limit, cursor 318 func (_m *ThreadStore) PermanentDeleteBatchForRetentionPolicies(now int64, globalPolicyEndTime int64, limit int64, cursor model.RetentionPolicyCursor) (int64, model.RetentionPolicyCursor, error) { 319 ret := _m.Called(now, globalPolicyEndTime, limit, cursor) 320 321 var r0 int64 322 if rf, ok := ret.Get(0).(func(int64, int64, int64, model.RetentionPolicyCursor) int64); ok { 323 r0 = rf(now, globalPolicyEndTime, limit, cursor) 324 } else { 325 r0 = ret.Get(0).(int64) 326 } 327 328 var r1 model.RetentionPolicyCursor 329 if rf, ok := ret.Get(1).(func(int64, int64, int64, model.RetentionPolicyCursor) model.RetentionPolicyCursor); ok { 330 r1 = rf(now, globalPolicyEndTime, limit, cursor) 331 } else { 332 r1 = ret.Get(1).(model.RetentionPolicyCursor) 333 } 334 335 var r2 error 336 if rf, ok := ret.Get(2).(func(int64, int64, int64, model.RetentionPolicyCursor) error); ok { 337 r2 = rf(now, globalPolicyEndTime, limit, cursor) 338 } else { 339 r2 = ret.Error(2) 340 } 341 342 return r0, r1, r2 343 } 344 345 // PermanentDeleteBatchThreadMembershipsForRetentionPolicies provides a mock function with given fields: now, globalPolicyEndTime, limit, cursor 346 func (_m *ThreadStore) PermanentDeleteBatchThreadMembershipsForRetentionPolicies(now int64, globalPolicyEndTime int64, limit int64, cursor model.RetentionPolicyCursor) (int64, model.RetentionPolicyCursor, error) { 347 ret := _m.Called(now, globalPolicyEndTime, limit, cursor) 348 349 var r0 int64 350 if rf, ok := ret.Get(0).(func(int64, int64, int64, model.RetentionPolicyCursor) int64); ok { 351 r0 = rf(now, globalPolicyEndTime, limit, cursor) 352 } else { 353 r0 = ret.Get(0).(int64) 354 } 355 356 var r1 model.RetentionPolicyCursor 357 if rf, ok := ret.Get(1).(func(int64, int64, int64, model.RetentionPolicyCursor) model.RetentionPolicyCursor); ok { 358 r1 = rf(now, globalPolicyEndTime, limit, cursor) 359 } else { 360 r1 = ret.Get(1).(model.RetentionPolicyCursor) 361 } 362 363 var r2 error 364 if rf, ok := ret.Get(2).(func(int64, int64, int64, model.RetentionPolicyCursor) error); ok { 365 r2 = rf(now, globalPolicyEndTime, limit, cursor) 366 } else { 367 r2 = ret.Error(2) 368 } 369 370 return r0, r1, r2 371 } 372 373 // Save provides a mock function with given fields: thread 374 func (_m *ThreadStore) Save(thread *model.Thread) (*model.Thread, error) { 375 ret := _m.Called(thread) 376 377 var r0 *model.Thread 378 if rf, ok := ret.Get(0).(func(*model.Thread) *model.Thread); ok { 379 r0 = rf(thread) 380 } else { 381 if ret.Get(0) != nil { 382 r0 = ret.Get(0).(*model.Thread) 383 } 384 } 385 386 var r1 error 387 if rf, ok := ret.Get(1).(func(*model.Thread) error); ok { 388 r1 = rf(thread) 389 } else { 390 r1 = ret.Error(1) 391 } 392 393 return r0, r1 394 } 395 396 // SaveMembership provides a mock function with given fields: membership 397 func (_m *ThreadStore) SaveMembership(membership *model.ThreadMembership) (*model.ThreadMembership, error) { 398 ret := _m.Called(membership) 399 400 var r0 *model.ThreadMembership 401 if rf, ok := ret.Get(0).(func(*model.ThreadMembership) *model.ThreadMembership); ok { 402 r0 = rf(membership) 403 } else { 404 if ret.Get(0) != nil { 405 r0 = ret.Get(0).(*model.ThreadMembership) 406 } 407 } 408 409 var r1 error 410 if rf, ok := ret.Get(1).(func(*model.ThreadMembership) error); ok { 411 r1 = rf(membership) 412 } else { 413 r1 = ret.Error(1) 414 } 415 416 return r0, r1 417 } 418 419 // SaveMultiple provides a mock function with given fields: thread 420 func (_m *ThreadStore) SaveMultiple(thread []*model.Thread) ([]*model.Thread, int, error) { 421 ret := _m.Called(thread) 422 423 var r0 []*model.Thread 424 if rf, ok := ret.Get(0).(func([]*model.Thread) []*model.Thread); ok { 425 r0 = rf(thread) 426 } else { 427 if ret.Get(0) != nil { 428 r0 = ret.Get(0).([]*model.Thread) 429 } 430 } 431 432 var r1 int 433 if rf, ok := ret.Get(1).(func([]*model.Thread) int); ok { 434 r1 = rf(thread) 435 } else { 436 r1 = ret.Get(1).(int) 437 } 438 439 var r2 error 440 if rf, ok := ret.Get(2).(func([]*model.Thread) error); ok { 441 r2 = rf(thread) 442 } else { 443 r2 = ret.Error(2) 444 } 445 446 return r0, r1, r2 447 } 448 449 // Update provides a mock function with given fields: thread 450 func (_m *ThreadStore) Update(thread *model.Thread) (*model.Thread, error) { 451 ret := _m.Called(thread) 452 453 var r0 *model.Thread 454 if rf, ok := ret.Get(0).(func(*model.Thread) *model.Thread); ok { 455 r0 = rf(thread) 456 } else { 457 if ret.Get(0) != nil { 458 r0 = ret.Get(0).(*model.Thread) 459 } 460 } 461 462 var r1 error 463 if rf, ok := ret.Get(1).(func(*model.Thread) error); ok { 464 r1 = rf(thread) 465 } else { 466 r1 = ret.Error(1) 467 } 468 469 return r0, r1 470 } 471 472 // UpdateMembership provides a mock function with given fields: membership 473 func (_m *ThreadStore) UpdateMembership(membership *model.ThreadMembership) (*model.ThreadMembership, error) { 474 ret := _m.Called(membership) 475 476 var r0 *model.ThreadMembership 477 if rf, ok := ret.Get(0).(func(*model.ThreadMembership) *model.ThreadMembership); ok { 478 r0 = rf(membership) 479 } else { 480 if ret.Get(0) != nil { 481 r0 = ret.Get(0).(*model.ThreadMembership) 482 } 483 } 484 485 var r1 error 486 if rf, ok := ret.Get(1).(func(*model.ThreadMembership) error); ok { 487 r1 = rf(membership) 488 } else { 489 r1 = ret.Error(1) 490 } 491 492 return r0, r1 493 } 494 495 // UpdateUnreadsByChannel provides a mock function with given fields: userId, changedThreads, timestamp, updateViewedTimestamp 496 func (_m *ThreadStore) UpdateUnreadsByChannel(userId string, changedThreads []string, timestamp int64, updateViewedTimestamp bool) error { 497 ret := _m.Called(userId, changedThreads, timestamp, updateViewedTimestamp) 498 499 var r0 error 500 if rf, ok := ret.Get(0).(func(string, []string, int64, bool) error); ok { 501 r0 = rf(userId, changedThreads, timestamp, updateViewedTimestamp) 502 } else { 503 r0 = ret.Error(0) 504 } 505 506 return r0 507 }