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  }