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  }