github.com/xzl8028/xenia-server@v0.0.0-20190809101854-18450a97da63/store/storetest/mocks/JobStore.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  // Regenerate this file using `make store-mocks`.
     4  
     5  package mocks
     6  
     7  import mock "github.com/stretchr/testify/mock"
     8  import model "github.com/xzl8028/xenia-server/model"
     9  
    10  // JobStore is an autogenerated mock type for the JobStore type
    11  type JobStore struct {
    12  	mock.Mock
    13  }
    14  
    15  // Delete provides a mock function with given fields: id
    16  func (_m *JobStore) Delete(id string) (string, *model.AppError) {
    17  	ret := _m.Called(id)
    18  
    19  	var r0 string
    20  	if rf, ok := ret.Get(0).(func(string) string); ok {
    21  		r0 = rf(id)
    22  	} else {
    23  		r0 = ret.Get(0).(string)
    24  	}
    25  
    26  	var r1 *model.AppError
    27  	if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
    28  		r1 = rf(id)
    29  	} else {
    30  		if ret.Get(1) != nil {
    31  			r1 = ret.Get(1).(*model.AppError)
    32  		}
    33  	}
    34  
    35  	return r0, r1
    36  }
    37  
    38  // Get provides a mock function with given fields: id
    39  func (_m *JobStore) Get(id string) (*model.Job, *model.AppError) {
    40  	ret := _m.Called(id)
    41  
    42  	var r0 *model.Job
    43  	if rf, ok := ret.Get(0).(func(string) *model.Job); ok {
    44  		r0 = rf(id)
    45  	} else {
    46  		if ret.Get(0) != nil {
    47  			r0 = ret.Get(0).(*model.Job)
    48  		}
    49  	}
    50  
    51  	var r1 *model.AppError
    52  	if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
    53  		r1 = rf(id)
    54  	} else {
    55  		if ret.Get(1) != nil {
    56  			r1 = ret.Get(1).(*model.AppError)
    57  		}
    58  	}
    59  
    60  	return r0, r1
    61  }
    62  
    63  // GetAllByStatus provides a mock function with given fields: status
    64  func (_m *JobStore) GetAllByStatus(status string) ([]*model.Job, *model.AppError) {
    65  	ret := _m.Called(status)
    66  
    67  	var r0 []*model.Job
    68  	if rf, ok := ret.Get(0).(func(string) []*model.Job); ok {
    69  		r0 = rf(status)
    70  	} else {
    71  		if ret.Get(0) != nil {
    72  			r0 = ret.Get(0).([]*model.Job)
    73  		}
    74  	}
    75  
    76  	var r1 *model.AppError
    77  	if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
    78  		r1 = rf(status)
    79  	} else {
    80  		if ret.Get(1) != nil {
    81  			r1 = ret.Get(1).(*model.AppError)
    82  		}
    83  	}
    84  
    85  	return r0, r1
    86  }
    87  
    88  // GetAllByType provides a mock function with given fields: jobType
    89  func (_m *JobStore) GetAllByType(jobType string) ([]*model.Job, *model.AppError) {
    90  	ret := _m.Called(jobType)
    91  
    92  	var r0 []*model.Job
    93  	if rf, ok := ret.Get(0).(func(string) []*model.Job); ok {
    94  		r0 = rf(jobType)
    95  	} else {
    96  		if ret.Get(0) != nil {
    97  			r0 = ret.Get(0).([]*model.Job)
    98  		}
    99  	}
   100  
   101  	var r1 *model.AppError
   102  	if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
   103  		r1 = rf(jobType)
   104  	} else {
   105  		if ret.Get(1) != nil {
   106  			r1 = ret.Get(1).(*model.AppError)
   107  		}
   108  	}
   109  
   110  	return r0, r1
   111  }
   112  
   113  // GetAllByTypePage provides a mock function with given fields: jobType, offset, limit
   114  func (_m *JobStore) GetAllByTypePage(jobType string, offset int, limit int) ([]*model.Job, *model.AppError) {
   115  	ret := _m.Called(jobType, offset, limit)
   116  
   117  	var r0 []*model.Job
   118  	if rf, ok := ret.Get(0).(func(string, int, int) []*model.Job); ok {
   119  		r0 = rf(jobType, offset, limit)
   120  	} else {
   121  		if ret.Get(0) != nil {
   122  			r0 = ret.Get(0).([]*model.Job)
   123  		}
   124  	}
   125  
   126  	var r1 *model.AppError
   127  	if rf, ok := ret.Get(1).(func(string, int, int) *model.AppError); ok {
   128  		r1 = rf(jobType, offset, limit)
   129  	} else {
   130  		if ret.Get(1) != nil {
   131  			r1 = ret.Get(1).(*model.AppError)
   132  		}
   133  	}
   134  
   135  	return r0, r1
   136  }
   137  
   138  // GetAllPage provides a mock function with given fields: offset, limit
   139  func (_m *JobStore) GetAllPage(offset int, limit int) ([]*model.Job, *model.AppError) {
   140  	ret := _m.Called(offset, limit)
   141  
   142  	var r0 []*model.Job
   143  	if rf, ok := ret.Get(0).(func(int, int) []*model.Job); ok {
   144  		r0 = rf(offset, limit)
   145  	} else {
   146  		if ret.Get(0) != nil {
   147  			r0 = ret.Get(0).([]*model.Job)
   148  		}
   149  	}
   150  
   151  	var r1 *model.AppError
   152  	if rf, ok := ret.Get(1).(func(int, int) *model.AppError); ok {
   153  		r1 = rf(offset, limit)
   154  	} else {
   155  		if ret.Get(1) != nil {
   156  			r1 = ret.Get(1).(*model.AppError)
   157  		}
   158  	}
   159  
   160  	return r0, r1
   161  }
   162  
   163  // GetCountByStatusAndType provides a mock function with given fields: status, jobType
   164  func (_m *JobStore) GetCountByStatusAndType(status string, jobType string) (int64, *model.AppError) {
   165  	ret := _m.Called(status, jobType)
   166  
   167  	var r0 int64
   168  	if rf, ok := ret.Get(0).(func(string, string) int64); ok {
   169  		r0 = rf(status, jobType)
   170  	} else {
   171  		r0 = ret.Get(0).(int64)
   172  	}
   173  
   174  	var r1 *model.AppError
   175  	if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
   176  		r1 = rf(status, jobType)
   177  	} else {
   178  		if ret.Get(1) != nil {
   179  			r1 = ret.Get(1).(*model.AppError)
   180  		}
   181  	}
   182  
   183  	return r0, r1
   184  }
   185  
   186  // GetNewestJobByStatusAndType provides a mock function with given fields: status, jobType
   187  func (_m *JobStore) GetNewestJobByStatusAndType(status string, jobType string) (*model.Job, *model.AppError) {
   188  	ret := _m.Called(status, jobType)
   189  
   190  	var r0 *model.Job
   191  	if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
   192  		r0 = rf(status, jobType)
   193  	} else {
   194  		if ret.Get(0) != nil {
   195  			r0 = ret.Get(0).(*model.Job)
   196  		}
   197  	}
   198  
   199  	var r1 *model.AppError
   200  	if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
   201  		r1 = rf(status, jobType)
   202  	} else {
   203  		if ret.Get(1) != nil {
   204  			r1 = ret.Get(1).(*model.AppError)
   205  		}
   206  	}
   207  
   208  	return r0, r1
   209  }
   210  
   211  // Save provides a mock function with given fields: job
   212  func (_m *JobStore) Save(job *model.Job) (*model.Job, *model.AppError) {
   213  	ret := _m.Called(job)
   214  
   215  	var r0 *model.Job
   216  	if rf, ok := ret.Get(0).(func(*model.Job) *model.Job); ok {
   217  		r0 = rf(job)
   218  	} else {
   219  		if ret.Get(0) != nil {
   220  			r0 = ret.Get(0).(*model.Job)
   221  		}
   222  	}
   223  
   224  	var r1 *model.AppError
   225  	if rf, ok := ret.Get(1).(func(*model.Job) *model.AppError); ok {
   226  		r1 = rf(job)
   227  	} else {
   228  		if ret.Get(1) != nil {
   229  			r1 = ret.Get(1).(*model.AppError)
   230  		}
   231  	}
   232  
   233  	return r0, r1
   234  }
   235  
   236  // UpdateOptimistically provides a mock function with given fields: job, currentStatus
   237  func (_m *JobStore) UpdateOptimistically(job *model.Job, currentStatus string) (bool, *model.AppError) {
   238  	ret := _m.Called(job, currentStatus)
   239  
   240  	var r0 bool
   241  	if rf, ok := ret.Get(0).(func(*model.Job, string) bool); ok {
   242  		r0 = rf(job, currentStatus)
   243  	} else {
   244  		r0 = ret.Get(0).(bool)
   245  	}
   246  
   247  	var r1 *model.AppError
   248  	if rf, ok := ret.Get(1).(func(*model.Job, string) *model.AppError); ok {
   249  		r1 = rf(job, currentStatus)
   250  	} else {
   251  		if ret.Get(1) != nil {
   252  			r1 = ret.Get(1).(*model.AppError)
   253  		}
   254  	}
   255  
   256  	return r0, r1
   257  }
   258  
   259  // UpdateStatus provides a mock function with given fields: id, status
   260  func (_m *JobStore) UpdateStatus(id string, status string) (*model.Job, *model.AppError) {
   261  	ret := _m.Called(id, status)
   262  
   263  	var r0 *model.Job
   264  	if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
   265  		r0 = rf(id, status)
   266  	} else {
   267  		if ret.Get(0) != nil {
   268  			r0 = ret.Get(0).(*model.Job)
   269  		}
   270  	}
   271  
   272  	var r1 *model.AppError
   273  	if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
   274  		r1 = rf(id, status)
   275  	} else {
   276  		if ret.Get(1) != nil {
   277  			r1 = ret.Get(1).(*model.AppError)
   278  		}
   279  	}
   280  
   281  	return r0, r1
   282  }
   283  
   284  // UpdateStatusOptimistically provides a mock function with given fields: id, currentStatus, newStatus
   285  func (_m *JobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) (bool, *model.AppError) {
   286  	ret := _m.Called(id, currentStatus, newStatus)
   287  
   288  	var r0 bool
   289  	if rf, ok := ret.Get(0).(func(string, string, string) bool); ok {
   290  		r0 = rf(id, currentStatus, newStatus)
   291  	} else {
   292  		r0 = ret.Get(0).(bool)
   293  	}
   294  
   295  	var r1 *model.AppError
   296  	if rf, ok := ret.Get(1).(func(string, string, string) *model.AppError); ok {
   297  		r1 = rf(id, currentStatus, newStatus)
   298  	} else {
   299  		if ret.Get(1) != nil {
   300  			r1 = ret.Get(1).(*model.AppError)
   301  		}
   302  	}
   303  
   304  	return r0, r1
   305  }