mock_oservice_service_test.go 9.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package webapi
  5. import (
  6. "context"
  7. "github.com/mk6i/open-oscar-server/state"
  8. "github.com/mk6i/open-oscar-server/wire"
  9. mock "github.com/stretchr/testify/mock"
  10. )
  11. // newMockOServiceService creates a new instance of mockOServiceService. 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 newMockOServiceService(t interface {
  14. mock.TestingT
  15. Cleanup(func())
  16. }) *mockOServiceService {
  17. if helper, ok := t.(interface{ Helper() }); ok {
  18. helper.Helper()
  19. }
  20. mock := &mockOServiceService{}
  21. mock.Mock.Test(t)
  22. t.Cleanup(func() {
  23. if helper, ok := t.(interface{ Helper() }); ok {
  24. helper.Helper()
  25. }
  26. mock.AssertExpectations(t)
  27. })
  28. return mock
  29. }
  30. // mockOServiceService is an autogenerated mock type for the OServiceService type
  31. type mockOServiceService struct {
  32. mock.Mock
  33. }
  34. type mockOServiceService_Expecter struct {
  35. mock *mock.Mock
  36. }
  37. func (_m *mockOServiceService) EXPECT() *mockOServiceService_Expecter {
  38. return &mockOServiceService_Expecter{mock: &_m.Mock}
  39. }
  40. // ClientOnline provides a mock function for the type mockOServiceService
  41. func (_mock *mockOServiceService) ClientOnline(ctx context.Context, service uint16, inBody wire.SNAC_0x01_0x02_OServiceClientOnline, instance *state.SessionInstance) error {
  42. ret := _mock.Called(ctx, service, inBody, instance)
  43. if len(ret) == 0 {
  44. panic("no return value specified for ClientOnline")
  45. }
  46. var r0 error
  47. if returnFunc, ok := ret.Get(0).(func(context.Context, uint16, wire.SNAC_0x01_0x02_OServiceClientOnline, *state.SessionInstance) error); ok {
  48. r0 = returnFunc(ctx, service, inBody, instance)
  49. } else {
  50. r0 = ret.Error(0)
  51. }
  52. return r0
  53. }
  54. // mockOServiceService_ClientOnline_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ClientOnline'
  55. type mockOServiceService_ClientOnline_Call struct {
  56. *mock.Call
  57. }
  58. // ClientOnline is a helper method to define mock.On call
  59. // - ctx context.Context
  60. // - service uint16
  61. // - inBody wire.SNAC_0x01_0x02_OServiceClientOnline
  62. // - instance *state.SessionInstance
  63. func (_e *mockOServiceService_Expecter) ClientOnline(ctx any, service any, inBody any, instance any) *mockOServiceService_ClientOnline_Call {
  64. return &mockOServiceService_ClientOnline_Call{Call: _e.mock.On("ClientOnline", ctx, service, inBody, instance)}
  65. }
  66. func (_c *mockOServiceService_ClientOnline_Call) Run(run func(ctx context.Context, service uint16, inBody wire.SNAC_0x01_0x02_OServiceClientOnline, instance *state.SessionInstance)) *mockOServiceService_ClientOnline_Call {
  67. _c.Call.Run(func(args mock.Arguments) {
  68. var arg0 context.Context
  69. if args[0] != nil {
  70. arg0 = args[0].(context.Context)
  71. }
  72. var arg1 uint16
  73. if args[1] != nil {
  74. arg1 = args[1].(uint16)
  75. }
  76. var arg2 wire.SNAC_0x01_0x02_OServiceClientOnline
  77. if args[2] != nil {
  78. arg2 = args[2].(wire.SNAC_0x01_0x02_OServiceClientOnline)
  79. }
  80. var arg3 *state.SessionInstance
  81. if args[3] != nil {
  82. arg3 = args[3].(*state.SessionInstance)
  83. }
  84. run(
  85. arg0,
  86. arg1,
  87. arg2,
  88. arg3,
  89. )
  90. })
  91. return _c
  92. }
  93. func (_c *mockOServiceService_ClientOnline_Call) Return(err error) *mockOServiceService_ClientOnline_Call {
  94. _c.Call.Return(err)
  95. return _c
  96. }
  97. func (_c *mockOServiceService_ClientOnline_Call) RunAndReturn(run func(ctx context.Context, service uint16, inBody wire.SNAC_0x01_0x02_OServiceClientOnline, instance *state.SessionInstance) error) *mockOServiceService_ClientOnline_Call {
  98. _c.Call.Return(run)
  99. return _c
  100. }
  101. // MonitorRateLimits provides a mock function for the type mockOServiceService
  102. func (_mock *mockOServiceService) MonitorRateLimits(ctx context.Context, session *state.Session) {
  103. _mock.Called(ctx, session)
  104. return
  105. }
  106. // mockOServiceService_MonitorRateLimits_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MonitorRateLimits'
  107. type mockOServiceService_MonitorRateLimits_Call struct {
  108. *mock.Call
  109. }
  110. // MonitorRateLimits is a helper method to define mock.On call
  111. // - ctx context.Context
  112. // - session *state.Session
  113. func (_e *mockOServiceService_Expecter) MonitorRateLimits(ctx any, session any) *mockOServiceService_MonitorRateLimits_Call {
  114. return &mockOServiceService_MonitorRateLimits_Call{Call: _e.mock.On("MonitorRateLimits", ctx, session)}
  115. }
  116. func (_c *mockOServiceService_MonitorRateLimits_Call) Run(run func(ctx context.Context, session *state.Session)) *mockOServiceService_MonitorRateLimits_Call {
  117. _c.Call.Run(func(args mock.Arguments) {
  118. var arg0 context.Context
  119. if args[0] != nil {
  120. arg0 = args[0].(context.Context)
  121. }
  122. var arg1 *state.Session
  123. if args[1] != nil {
  124. arg1 = args[1].(*state.Session)
  125. }
  126. run(
  127. arg0,
  128. arg1,
  129. )
  130. })
  131. return _c
  132. }
  133. func (_c *mockOServiceService_MonitorRateLimits_Call) Return() *mockOServiceService_MonitorRateLimits_Call {
  134. _c.Call.Return()
  135. return _c
  136. }
  137. func (_c *mockOServiceService_MonitorRateLimits_Call) RunAndReturn(run func(ctx context.Context, session *state.Session)) *mockOServiceService_MonitorRateLimits_Call {
  138. _c.Run(run)
  139. return _c
  140. }
  141. // RateParamsSubAdd provides a mock function for the type mockOServiceService
  142. func (_mock *mockOServiceService) RateParamsSubAdd(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd) {
  143. _mock.Called(ctx, instance, inBody)
  144. return
  145. }
  146. // mockOServiceService_RateParamsSubAdd_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsSubAdd'
  147. type mockOServiceService_RateParamsSubAdd_Call struct {
  148. *mock.Call
  149. }
  150. // RateParamsSubAdd is a helper method to define mock.On call
  151. // - ctx context.Context
  152. // - instance *state.SessionInstance
  153. // - inBody wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd
  154. func (_e *mockOServiceService_Expecter) RateParamsSubAdd(ctx any, instance any, inBody any) *mockOServiceService_RateParamsSubAdd_Call {
  155. return &mockOServiceService_RateParamsSubAdd_Call{Call: _e.mock.On("RateParamsSubAdd", ctx, instance, inBody)}
  156. }
  157. func (_c *mockOServiceService_RateParamsSubAdd_Call) Run(run func(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  158. _c.Call.Run(func(args mock.Arguments) {
  159. var arg0 context.Context
  160. if args[0] != nil {
  161. arg0 = args[0].(context.Context)
  162. }
  163. var arg1 *state.SessionInstance
  164. if args[1] != nil {
  165. arg1 = args[1].(*state.SessionInstance)
  166. }
  167. var arg2 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd
  168. if args[2] != nil {
  169. arg2 = args[2].(wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)
  170. }
  171. run(
  172. arg0,
  173. arg1,
  174. arg2,
  175. )
  176. })
  177. return _c
  178. }
  179. func (_c *mockOServiceService_RateParamsSubAdd_Call) Return() *mockOServiceService_RateParamsSubAdd_Call {
  180. _c.Call.Return()
  181. return _c
  182. }
  183. func (_c *mockOServiceService_RateParamsSubAdd_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  184. _c.Run(run)
  185. return _c
  186. }
  187. // SetUserInfoFields provides a mock function for the type mockOServiceService
  188. func (_mock *mockOServiceService) SetUserInfoFields(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error {
  189. ret := _mock.Called(ctx, instance, inFrame, inBody)
  190. if len(ret) == 0 {
  191. panic("no return value specified for SetUserInfoFields")
  192. }
  193. var r0 error
  194. if returnFunc, ok := ret.Get(0).(func(context.Context, *state.SessionInstance, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error); ok {
  195. r0 = returnFunc(ctx, instance, inFrame, inBody)
  196. } else {
  197. r0 = ret.Error(0)
  198. }
  199. return r0
  200. }
  201. // mockOServiceService_SetUserInfoFields_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetUserInfoFields'
  202. type mockOServiceService_SetUserInfoFields_Call struct {
  203. *mock.Call
  204. }
  205. // SetUserInfoFields is a helper method to define mock.On call
  206. // - ctx context.Context
  207. // - instance *state.SessionInstance
  208. // - inFrame wire.SNACFrame
  209. // - inBody wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields
  210. func (_e *mockOServiceService_Expecter) SetUserInfoFields(ctx any, instance any, inFrame any, inBody any) *mockOServiceService_SetUserInfoFields_Call {
  211. return &mockOServiceService_SetUserInfoFields_Call{Call: _e.mock.On("SetUserInfoFields", ctx, instance, inFrame, inBody)}
  212. }
  213. func (_c *mockOServiceService_SetUserInfoFields_Call) Run(run func(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields)) *mockOServiceService_SetUserInfoFields_Call {
  214. _c.Call.Run(func(args mock.Arguments) {
  215. var arg0 context.Context
  216. if args[0] != nil {
  217. arg0 = args[0].(context.Context)
  218. }
  219. var arg1 *state.SessionInstance
  220. if args[1] != nil {
  221. arg1 = args[1].(*state.SessionInstance)
  222. }
  223. var arg2 wire.SNACFrame
  224. if args[2] != nil {
  225. arg2 = args[2].(wire.SNACFrame)
  226. }
  227. var arg3 wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields
  228. if args[3] != nil {
  229. arg3 = args[3].(wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields)
  230. }
  231. run(
  232. arg0,
  233. arg1,
  234. arg2,
  235. arg3,
  236. )
  237. })
  238. return _c
  239. }
  240. func (_c *mockOServiceService_SetUserInfoFields_Call) Return(err error) *mockOServiceService_SetUserInfoFields_Call {
  241. _c.Call.Return(err)
  242. return _c
  243. }
  244. func (_c *mockOServiceService_SetUserInfoFields_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error) *mockOServiceService_SetUserInfoFields_Call {
  245. _c.Call.Return(run)
  246. return _c
  247. }