mock_user_manager_manager_test.go 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package foodgroup
  5. import (
  6. "context"
  7. "time"
  8. "github.com/mk6i/open-oscar-server/state"
  9. mock "github.com/stretchr/testify/mock"
  10. )
  11. // newMockUserManager creates a new instance of mockUserManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  12. // The first argument is typically a *testing.T value.
  13. func newMockUserManager(t interface {
  14. mock.TestingT
  15. Cleanup(func())
  16. }) *mockUserManager {
  17. mock := &mockUserManager{}
  18. mock.Mock.Test(t)
  19. t.Cleanup(func() { mock.AssertExpectations(t) })
  20. return mock
  21. }
  22. // mockUserManager is an autogenerated mock type for the UserManager type
  23. type mockUserManager struct {
  24. mock.Mock
  25. }
  26. type mockUserManager_Expecter struct {
  27. mock *mock.Mock
  28. }
  29. func (_m *mockUserManager) EXPECT() *mockUserManager_Expecter {
  30. return &mockUserManager_Expecter{mock: &_m.Mock}
  31. }
  32. // InsertUser provides a mock function for the type mockUserManager
  33. func (_mock *mockUserManager) InsertUser(ctx context.Context, u state.User) error {
  34. ret := _mock.Called(ctx, u)
  35. if len(ret) == 0 {
  36. panic("no return value specified for InsertUser")
  37. }
  38. var r0 error
  39. if returnFunc, ok := ret.Get(0).(func(context.Context, state.User) error); ok {
  40. r0 = returnFunc(ctx, u)
  41. } else {
  42. r0 = ret.Error(0)
  43. }
  44. return r0
  45. }
  46. // mockUserManager_InsertUser_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InsertUser'
  47. type mockUserManager_InsertUser_Call struct {
  48. *mock.Call
  49. }
  50. // InsertUser is a helper method to define mock.On call
  51. // - ctx context.Context
  52. // - u state.User
  53. func (_e *mockUserManager_Expecter) InsertUser(ctx interface{}, u interface{}) *mockUserManager_InsertUser_Call {
  54. return &mockUserManager_InsertUser_Call{Call: _e.mock.On("InsertUser", ctx, u)}
  55. }
  56. func (_c *mockUserManager_InsertUser_Call) Run(run func(ctx context.Context, u state.User)) *mockUserManager_InsertUser_Call {
  57. _c.Call.Run(func(args mock.Arguments) {
  58. var arg0 context.Context
  59. if args[0] != nil {
  60. arg0 = args[0].(context.Context)
  61. }
  62. var arg1 state.User
  63. if args[1] != nil {
  64. arg1 = args[1].(state.User)
  65. }
  66. run(
  67. arg0,
  68. arg1,
  69. )
  70. })
  71. return _c
  72. }
  73. func (_c *mockUserManager_InsertUser_Call) Return(err error) *mockUserManager_InsertUser_Call {
  74. _c.Call.Return(err)
  75. return _c
  76. }
  77. func (_c *mockUserManager_InsertUser_Call) RunAndReturn(run func(ctx context.Context, u state.User) error) *mockUserManager_InsertUser_Call {
  78. _c.Call.Return(run)
  79. return _c
  80. }
  81. // SetWarnLevel provides a mock function for the type mockUserManager
  82. func (_mock *mockUserManager) SetWarnLevel(ctx context.Context, user state.IdentScreenName, lastWarnUpdate time.Time, lastWarnLevel uint16) error {
  83. ret := _mock.Called(ctx, user, lastWarnUpdate, lastWarnLevel)
  84. if len(ret) == 0 {
  85. panic("no return value specified for SetWarnLevel")
  86. }
  87. var r0 error
  88. if returnFunc, ok := ret.Get(0).(func(context.Context, state.IdentScreenName, time.Time, uint16) error); ok {
  89. r0 = returnFunc(ctx, user, lastWarnUpdate, lastWarnLevel)
  90. } else {
  91. r0 = ret.Error(0)
  92. }
  93. return r0
  94. }
  95. // mockUserManager_SetWarnLevel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetWarnLevel'
  96. type mockUserManager_SetWarnLevel_Call struct {
  97. *mock.Call
  98. }
  99. // SetWarnLevel is a helper method to define mock.On call
  100. // - ctx context.Context
  101. // - user state.IdentScreenName
  102. // - lastWarnUpdate time.Time
  103. // - lastWarnLevel uint16
  104. func (_e *mockUserManager_Expecter) SetWarnLevel(ctx interface{}, user interface{}, lastWarnUpdate interface{}, lastWarnLevel interface{}) *mockUserManager_SetWarnLevel_Call {
  105. return &mockUserManager_SetWarnLevel_Call{Call: _e.mock.On("SetWarnLevel", ctx, user, lastWarnUpdate, lastWarnLevel)}
  106. }
  107. func (_c *mockUserManager_SetWarnLevel_Call) Run(run func(ctx context.Context, user state.IdentScreenName, lastWarnUpdate time.Time, lastWarnLevel uint16)) *mockUserManager_SetWarnLevel_Call {
  108. _c.Call.Run(func(args mock.Arguments) {
  109. var arg0 context.Context
  110. if args[0] != nil {
  111. arg0 = args[0].(context.Context)
  112. }
  113. var arg1 state.IdentScreenName
  114. if args[1] != nil {
  115. arg1 = args[1].(state.IdentScreenName)
  116. }
  117. var arg2 time.Time
  118. if args[2] != nil {
  119. arg2 = args[2].(time.Time)
  120. }
  121. var arg3 uint16
  122. if args[3] != nil {
  123. arg3 = args[3].(uint16)
  124. }
  125. run(
  126. arg0,
  127. arg1,
  128. arg2,
  129. arg3,
  130. )
  131. })
  132. return _c
  133. }
  134. func (_c *mockUserManager_SetWarnLevel_Call) Return(err error) *mockUserManager_SetWarnLevel_Call {
  135. _c.Call.Return(err)
  136. return _c
  137. }
  138. func (_c *mockUserManager_SetWarnLevel_Call) RunAndReturn(run func(ctx context.Context, user state.IdentScreenName, lastWarnUpdate time.Time, lastWarnLevel uint16) error) *mockUserManager_SetWarnLevel_Call {
  139. _c.Call.Return(run)
  140. return _c
  141. }
  142. // User provides a mock function for the type mockUserManager
  143. func (_mock *mockUserManager) User(ctx context.Context, screenName state.IdentScreenName) (*state.User, error) {
  144. ret := _mock.Called(ctx, screenName)
  145. if len(ret) == 0 {
  146. panic("no return value specified for User")
  147. }
  148. var r0 *state.User
  149. var r1 error
  150. if returnFunc, ok := ret.Get(0).(func(context.Context, state.IdentScreenName) (*state.User, error)); ok {
  151. return returnFunc(ctx, screenName)
  152. }
  153. if returnFunc, ok := ret.Get(0).(func(context.Context, state.IdentScreenName) *state.User); ok {
  154. r0 = returnFunc(ctx, screenName)
  155. } else {
  156. if ret.Get(0) != nil {
  157. r0 = ret.Get(0).(*state.User)
  158. }
  159. }
  160. if returnFunc, ok := ret.Get(1).(func(context.Context, state.IdentScreenName) error); ok {
  161. r1 = returnFunc(ctx, screenName)
  162. } else {
  163. r1 = ret.Error(1)
  164. }
  165. return r0, r1
  166. }
  167. // mockUserManager_User_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'User'
  168. type mockUserManager_User_Call struct {
  169. *mock.Call
  170. }
  171. // User is a helper method to define mock.On call
  172. // - ctx context.Context
  173. // - screenName state.IdentScreenName
  174. func (_e *mockUserManager_Expecter) User(ctx interface{}, screenName interface{}) *mockUserManager_User_Call {
  175. return &mockUserManager_User_Call{Call: _e.mock.On("User", ctx, screenName)}
  176. }
  177. func (_c *mockUserManager_User_Call) Run(run func(ctx context.Context, screenName state.IdentScreenName)) *mockUserManager_User_Call {
  178. _c.Call.Run(func(args mock.Arguments) {
  179. var arg0 context.Context
  180. if args[0] != nil {
  181. arg0 = args[0].(context.Context)
  182. }
  183. var arg1 state.IdentScreenName
  184. if args[1] != nil {
  185. arg1 = args[1].(state.IdentScreenName)
  186. }
  187. run(
  188. arg0,
  189. arg1,
  190. )
  191. })
  192. return _c
  193. }
  194. func (_c *mockUserManager_User_Call) Return(user *state.User, err error) *mockUserManager_User_Call {
  195. _c.Call.Return(user, err)
  196. return _c
  197. }
  198. func (_c *mockUserManager_User_Call) RunAndReturn(run func(ctx context.Context, screenName state.IdentScreenName) (*state.User, error)) *mockUserManager_User_Call {
  199. _c.Call.Return(run)
  200. return _c
  201. }