mock_oservice_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509
  1. // Code generated by mockery v2.43.2. DO NOT EDIT.
  2. package handler
  3. import (
  4. context "context"
  5. state "github.com/mk6i/retro-aim-server/state"
  6. mock "github.com/stretchr/testify/mock"
  7. wire "github.com/mk6i/retro-aim-server/wire"
  8. )
  9. // mockOServiceService is an autogenerated mock type for the OServiceService type
  10. type mockOServiceService struct {
  11. mock.Mock
  12. }
  13. type mockOServiceService_Expecter struct {
  14. mock *mock.Mock
  15. }
  16. func (_m *mockOServiceService) EXPECT() *mockOServiceService_Expecter {
  17. return &mockOServiceService_Expecter{mock: &_m.Mock}
  18. }
  19. // ClientOnline provides a mock function with given fields: ctx, bodyIn, sess
  20. func (_m *mockOServiceService) ClientOnline(ctx context.Context, bodyIn wire.SNAC_0x01_0x02_OServiceClientOnline, sess *state.Session) error {
  21. ret := _m.Called(ctx, bodyIn, sess)
  22. if len(ret) == 0 {
  23. panic("no return value specified for ClientOnline")
  24. }
  25. var r0 error
  26. if rf, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x01_0x02_OServiceClientOnline, *state.Session) error); ok {
  27. r0 = rf(ctx, bodyIn, sess)
  28. } else {
  29. r0 = ret.Error(0)
  30. }
  31. return r0
  32. }
  33. // mockOServiceService_ClientOnline_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ClientOnline'
  34. type mockOServiceService_ClientOnline_Call struct {
  35. *mock.Call
  36. }
  37. // ClientOnline is a helper method to define mock.On call
  38. // - ctx context.Context
  39. // - bodyIn wire.SNAC_0x01_0x02_OServiceClientOnline
  40. // - sess *state.Session
  41. func (_e *mockOServiceService_Expecter) ClientOnline(ctx interface{}, bodyIn interface{}, sess interface{}) *mockOServiceService_ClientOnline_Call {
  42. return &mockOServiceService_ClientOnline_Call{Call: _e.mock.On("ClientOnline", ctx, bodyIn, sess)}
  43. }
  44. func (_c *mockOServiceService_ClientOnline_Call) Run(run func(ctx context.Context, bodyIn wire.SNAC_0x01_0x02_OServiceClientOnline, sess *state.Session)) *mockOServiceService_ClientOnline_Call {
  45. _c.Call.Run(func(args mock.Arguments) {
  46. run(args[0].(context.Context), args[1].(wire.SNAC_0x01_0x02_OServiceClientOnline), args[2].(*state.Session))
  47. })
  48. return _c
  49. }
  50. func (_c *mockOServiceService_ClientOnline_Call) Return(_a0 error) *mockOServiceService_ClientOnline_Call {
  51. _c.Call.Return(_a0)
  52. return _c
  53. }
  54. func (_c *mockOServiceService_ClientOnline_Call) RunAndReturn(run func(context.Context, wire.SNAC_0x01_0x02_OServiceClientOnline, *state.Session) error) *mockOServiceService_ClientOnline_Call {
  55. _c.Call.Return(run)
  56. return _c
  57. }
  58. // ClientVersions provides a mock function with given fields: ctx, frame, bodyIn
  59. func (_m *mockOServiceService) ClientVersions(ctx context.Context, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x17_OServiceClientVersions) wire.SNACMessage {
  60. ret := _m.Called(ctx, frame, bodyIn)
  61. if len(ret) == 0 {
  62. panic("no return value specified for ClientVersions")
  63. }
  64. var r0 wire.SNACMessage
  65. if rf, ok := ret.Get(0).(func(context.Context, wire.SNACFrame, wire.SNAC_0x01_0x17_OServiceClientVersions) wire.SNACMessage); ok {
  66. r0 = rf(ctx, frame, bodyIn)
  67. } else {
  68. r0 = ret.Get(0).(wire.SNACMessage)
  69. }
  70. return r0
  71. }
  72. // mockOServiceService_ClientVersions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ClientVersions'
  73. type mockOServiceService_ClientVersions_Call struct {
  74. *mock.Call
  75. }
  76. // ClientVersions is a helper method to define mock.On call
  77. // - ctx context.Context
  78. // - frame wire.SNACFrame
  79. // - bodyIn wire.SNAC_0x01_0x17_OServiceClientVersions
  80. func (_e *mockOServiceService_Expecter) ClientVersions(ctx interface{}, frame interface{}, bodyIn interface{}) *mockOServiceService_ClientVersions_Call {
  81. return &mockOServiceService_ClientVersions_Call{Call: _e.mock.On("ClientVersions", ctx, frame, bodyIn)}
  82. }
  83. func (_c *mockOServiceService_ClientVersions_Call) Run(run func(ctx context.Context, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x17_OServiceClientVersions)) *mockOServiceService_ClientVersions_Call {
  84. _c.Call.Run(func(args mock.Arguments) {
  85. run(args[0].(context.Context), args[1].(wire.SNACFrame), args[2].(wire.SNAC_0x01_0x17_OServiceClientVersions))
  86. })
  87. return _c
  88. }
  89. func (_c *mockOServiceService_ClientVersions_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_ClientVersions_Call {
  90. _c.Call.Return(_a0)
  91. return _c
  92. }
  93. func (_c *mockOServiceService_ClientVersions_Call) RunAndReturn(run func(context.Context, wire.SNACFrame, wire.SNAC_0x01_0x17_OServiceClientVersions) wire.SNACMessage) *mockOServiceService_ClientVersions_Call {
  94. _c.Call.Return(run)
  95. return _c
  96. }
  97. // HostOnline provides a mock function with given fields:
  98. func (_m *mockOServiceService) HostOnline() wire.SNACMessage {
  99. ret := _m.Called()
  100. if len(ret) == 0 {
  101. panic("no return value specified for HostOnline")
  102. }
  103. var r0 wire.SNACMessage
  104. if rf, ok := ret.Get(0).(func() wire.SNACMessage); ok {
  105. r0 = rf()
  106. } else {
  107. r0 = ret.Get(0).(wire.SNACMessage)
  108. }
  109. return r0
  110. }
  111. // mockOServiceService_HostOnline_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'HostOnline'
  112. type mockOServiceService_HostOnline_Call struct {
  113. *mock.Call
  114. }
  115. // HostOnline is a helper method to define mock.On call
  116. func (_e *mockOServiceService_Expecter) HostOnline() *mockOServiceService_HostOnline_Call {
  117. return &mockOServiceService_HostOnline_Call{Call: _e.mock.On("HostOnline")}
  118. }
  119. func (_c *mockOServiceService_HostOnline_Call) Run(run func()) *mockOServiceService_HostOnline_Call {
  120. _c.Call.Run(func(args mock.Arguments) {
  121. run()
  122. })
  123. return _c
  124. }
  125. func (_c *mockOServiceService_HostOnline_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_HostOnline_Call {
  126. _c.Call.Return(_a0)
  127. return _c
  128. }
  129. func (_c *mockOServiceService_HostOnline_Call) RunAndReturn(run func() wire.SNACMessage) *mockOServiceService_HostOnline_Call {
  130. _c.Call.Return(run)
  131. return _c
  132. }
  133. // IdleNotification provides a mock function with given fields: ctx, sess, bodyIn
  134. func (_m *mockOServiceService) IdleNotification(ctx context.Context, sess *state.Session, bodyIn wire.SNAC_0x01_0x11_OServiceIdleNotification) error {
  135. ret := _m.Called(ctx, sess, bodyIn)
  136. if len(ret) == 0 {
  137. panic("no return value specified for IdleNotification")
  138. }
  139. var r0 error
  140. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNAC_0x01_0x11_OServiceIdleNotification) error); ok {
  141. r0 = rf(ctx, sess, bodyIn)
  142. } else {
  143. r0 = ret.Error(0)
  144. }
  145. return r0
  146. }
  147. // mockOServiceService_IdleNotification_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IdleNotification'
  148. type mockOServiceService_IdleNotification_Call struct {
  149. *mock.Call
  150. }
  151. // IdleNotification is a helper method to define mock.On call
  152. // - ctx context.Context
  153. // - sess *state.Session
  154. // - bodyIn wire.SNAC_0x01_0x11_OServiceIdleNotification
  155. func (_e *mockOServiceService_Expecter) IdleNotification(ctx interface{}, sess interface{}, bodyIn interface{}) *mockOServiceService_IdleNotification_Call {
  156. return &mockOServiceService_IdleNotification_Call{Call: _e.mock.On("IdleNotification", ctx, sess, bodyIn)}
  157. }
  158. func (_c *mockOServiceService_IdleNotification_Call) Run(run func(ctx context.Context, sess *state.Session, bodyIn wire.SNAC_0x01_0x11_OServiceIdleNotification)) *mockOServiceService_IdleNotification_Call {
  159. _c.Call.Run(func(args mock.Arguments) {
  160. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNAC_0x01_0x11_OServiceIdleNotification))
  161. })
  162. return _c
  163. }
  164. func (_c *mockOServiceService_IdleNotification_Call) Return(_a0 error) *mockOServiceService_IdleNotification_Call {
  165. _c.Call.Return(_a0)
  166. return _c
  167. }
  168. func (_c *mockOServiceService_IdleNotification_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNAC_0x01_0x11_OServiceIdleNotification) error) *mockOServiceService_IdleNotification_Call {
  169. _c.Call.Return(run)
  170. return _c
  171. }
  172. // RateParamsQuery provides a mock function with given fields: ctx, frame
  173. func (_m *mockOServiceService) RateParamsQuery(ctx context.Context, frame wire.SNACFrame) wire.SNACMessage {
  174. ret := _m.Called(ctx, frame)
  175. if len(ret) == 0 {
  176. panic("no return value specified for RateParamsQuery")
  177. }
  178. var r0 wire.SNACMessage
  179. if rf, ok := ret.Get(0).(func(context.Context, wire.SNACFrame) wire.SNACMessage); ok {
  180. r0 = rf(ctx, frame)
  181. } else {
  182. r0 = ret.Get(0).(wire.SNACMessage)
  183. }
  184. return r0
  185. }
  186. // mockOServiceService_RateParamsQuery_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsQuery'
  187. type mockOServiceService_RateParamsQuery_Call struct {
  188. *mock.Call
  189. }
  190. // RateParamsQuery is a helper method to define mock.On call
  191. // - ctx context.Context
  192. // - frame wire.SNACFrame
  193. func (_e *mockOServiceService_Expecter) RateParamsQuery(ctx interface{}, frame interface{}) *mockOServiceService_RateParamsQuery_Call {
  194. return &mockOServiceService_RateParamsQuery_Call{Call: _e.mock.On("RateParamsQuery", ctx, frame)}
  195. }
  196. func (_c *mockOServiceService_RateParamsQuery_Call) Run(run func(ctx context.Context, frame wire.SNACFrame)) *mockOServiceService_RateParamsQuery_Call {
  197. _c.Call.Run(func(args mock.Arguments) {
  198. run(args[0].(context.Context), args[1].(wire.SNACFrame))
  199. })
  200. return _c
  201. }
  202. func (_c *mockOServiceService_RateParamsQuery_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_RateParamsQuery_Call {
  203. _c.Call.Return(_a0)
  204. return _c
  205. }
  206. func (_c *mockOServiceService_RateParamsQuery_Call) RunAndReturn(run func(context.Context, wire.SNACFrame) wire.SNACMessage) *mockOServiceService_RateParamsQuery_Call {
  207. _c.Call.Return(run)
  208. return _c
  209. }
  210. // RateParamsSubAdd provides a mock function with given fields: _a0, _a1
  211. func (_m *mockOServiceService) RateParamsSubAdd(_a0 context.Context, _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd) {
  212. _m.Called(_a0, _a1)
  213. }
  214. // mockOServiceService_RateParamsSubAdd_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsSubAdd'
  215. type mockOServiceService_RateParamsSubAdd_Call struct {
  216. *mock.Call
  217. }
  218. // RateParamsSubAdd is a helper method to define mock.On call
  219. // - _a0 context.Context
  220. // - _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd
  221. func (_e *mockOServiceService_Expecter) RateParamsSubAdd(_a0 interface{}, _a1 interface{}) *mockOServiceService_RateParamsSubAdd_Call {
  222. return &mockOServiceService_RateParamsSubAdd_Call{Call: _e.mock.On("RateParamsSubAdd", _a0, _a1)}
  223. }
  224. func (_c *mockOServiceService_RateParamsSubAdd_Call) Run(run func(_a0 context.Context, _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  225. _c.Call.Run(func(args mock.Arguments) {
  226. run(args[0].(context.Context), args[1].(wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd))
  227. })
  228. return _c
  229. }
  230. func (_c *mockOServiceService_RateParamsSubAdd_Call) Return() *mockOServiceService_RateParamsSubAdd_Call {
  231. _c.Call.Return()
  232. return _c
  233. }
  234. func (_c *mockOServiceService_RateParamsSubAdd_Call) RunAndReturn(run func(context.Context, wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  235. _c.Call.Return(run)
  236. return _c
  237. }
  238. // ServiceRequest provides a mock function with given fields: ctx, sess, frame, bodyIn
  239. func (_m *mockOServiceService) ServiceRequest(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x04_OServiceServiceRequest) (wire.SNACMessage, error) {
  240. ret := _m.Called(ctx, sess, frame, bodyIn)
  241. if len(ret) == 0 {
  242. panic("no return value specified for ServiceRequest")
  243. }
  244. var r0 wire.SNACMessage
  245. var r1 error
  246. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) (wire.SNACMessage, error)); ok {
  247. return rf(ctx, sess, frame, bodyIn)
  248. }
  249. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) wire.SNACMessage); ok {
  250. r0 = rf(ctx, sess, frame, bodyIn)
  251. } else {
  252. r0 = ret.Get(0).(wire.SNACMessage)
  253. }
  254. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) error); ok {
  255. r1 = rf(ctx, sess, frame, bodyIn)
  256. } else {
  257. r1 = ret.Error(1)
  258. }
  259. return r0, r1
  260. }
  261. // mockOServiceService_ServiceRequest_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ServiceRequest'
  262. type mockOServiceService_ServiceRequest_Call struct {
  263. *mock.Call
  264. }
  265. // ServiceRequest is a helper method to define mock.On call
  266. // - ctx context.Context
  267. // - sess *state.Session
  268. // - frame wire.SNACFrame
  269. // - bodyIn wire.SNAC_0x01_0x04_OServiceServiceRequest
  270. func (_e *mockOServiceService_Expecter) ServiceRequest(ctx interface{}, sess interface{}, frame interface{}, bodyIn interface{}) *mockOServiceService_ServiceRequest_Call {
  271. return &mockOServiceService_ServiceRequest_Call{Call: _e.mock.On("ServiceRequest", ctx, sess, frame, bodyIn)}
  272. }
  273. func (_c *mockOServiceService_ServiceRequest_Call) Run(run func(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x04_OServiceServiceRequest)) *mockOServiceService_ServiceRequest_Call {
  274. _c.Call.Run(func(args mock.Arguments) {
  275. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x01_0x04_OServiceServiceRequest))
  276. })
  277. return _c
  278. }
  279. func (_c *mockOServiceService_ServiceRequest_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockOServiceService_ServiceRequest_Call {
  280. _c.Call.Return(_a0, _a1)
  281. return _c
  282. }
  283. func (_c *mockOServiceService_ServiceRequest_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) (wire.SNACMessage, error)) *mockOServiceService_ServiceRequest_Call {
  284. _c.Call.Return(run)
  285. return _c
  286. }
  287. // SetPrivacyFlags provides a mock function with given fields: ctx, bodyIn
  288. func (_m *mockOServiceService) SetPrivacyFlags(ctx context.Context, bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags) {
  289. _m.Called(ctx, bodyIn)
  290. }
  291. // mockOServiceService_SetPrivacyFlags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetPrivacyFlags'
  292. type mockOServiceService_SetPrivacyFlags_Call struct {
  293. *mock.Call
  294. }
  295. // SetPrivacyFlags is a helper method to define mock.On call
  296. // - ctx context.Context
  297. // - bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags
  298. func (_e *mockOServiceService_Expecter) SetPrivacyFlags(ctx interface{}, bodyIn interface{}) *mockOServiceService_SetPrivacyFlags_Call {
  299. return &mockOServiceService_SetPrivacyFlags_Call{Call: _e.mock.On("SetPrivacyFlags", ctx, bodyIn)}
  300. }
  301. func (_c *mockOServiceService_SetPrivacyFlags_Call) Run(run func(ctx context.Context, bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags)) *mockOServiceService_SetPrivacyFlags_Call {
  302. _c.Call.Run(func(args mock.Arguments) {
  303. run(args[0].(context.Context), args[1].(wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags))
  304. })
  305. return _c
  306. }
  307. func (_c *mockOServiceService_SetPrivacyFlags_Call) Return() *mockOServiceService_SetPrivacyFlags_Call {
  308. _c.Call.Return()
  309. return _c
  310. }
  311. func (_c *mockOServiceService_SetPrivacyFlags_Call) RunAndReturn(run func(context.Context, wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags)) *mockOServiceService_SetPrivacyFlags_Call {
  312. _c.Call.Return(run)
  313. return _c
  314. }
  315. // SetUserInfoFields provides a mock function with given fields: ctx, sess, frame, bodyIn
  316. func (_m *mockOServiceService) SetUserInfoFields(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (wire.SNACMessage, error) {
  317. ret := _m.Called(ctx, sess, frame, bodyIn)
  318. if len(ret) == 0 {
  319. panic("no return value specified for SetUserInfoFields")
  320. }
  321. var r0 wire.SNACMessage
  322. var r1 error
  323. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (wire.SNACMessage, error)); ok {
  324. return rf(ctx, sess, frame, bodyIn)
  325. }
  326. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) wire.SNACMessage); ok {
  327. r0 = rf(ctx, sess, frame, bodyIn)
  328. } else {
  329. r0 = ret.Get(0).(wire.SNACMessage)
  330. }
  331. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error); ok {
  332. r1 = rf(ctx, sess, frame, bodyIn)
  333. } else {
  334. r1 = ret.Error(1)
  335. }
  336. return r0, r1
  337. }
  338. // mockOServiceService_SetUserInfoFields_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetUserInfoFields'
  339. type mockOServiceService_SetUserInfoFields_Call struct {
  340. *mock.Call
  341. }
  342. // SetUserInfoFields is a helper method to define mock.On call
  343. // - ctx context.Context
  344. // - sess *state.Session
  345. // - frame wire.SNACFrame
  346. // - bodyIn wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields
  347. func (_e *mockOServiceService_Expecter) SetUserInfoFields(ctx interface{}, sess interface{}, frame interface{}, bodyIn interface{}) *mockOServiceService_SetUserInfoFields_Call {
  348. return &mockOServiceService_SetUserInfoFields_Call{Call: _e.mock.On("SetUserInfoFields", ctx, sess, frame, bodyIn)}
  349. }
  350. func (_c *mockOServiceService_SetUserInfoFields_Call) Run(run func(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields)) *mockOServiceService_SetUserInfoFields_Call {
  351. _c.Call.Run(func(args mock.Arguments) {
  352. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields))
  353. })
  354. return _c
  355. }
  356. func (_c *mockOServiceService_SetUserInfoFields_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockOServiceService_SetUserInfoFields_Call {
  357. _c.Call.Return(_a0, _a1)
  358. return _c
  359. }
  360. func (_c *mockOServiceService_SetUserInfoFields_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (wire.SNACMessage, error)) *mockOServiceService_SetUserInfoFields_Call {
  361. _c.Call.Return(run)
  362. return _c
  363. }
  364. // UserInfoQuery provides a mock function with given fields: ctx, sess, frame
  365. func (_m *mockOServiceService) UserInfoQuery(ctx context.Context, sess *state.Session, frame wire.SNACFrame) wire.SNACMessage {
  366. ret := _m.Called(ctx, sess, frame)
  367. if len(ret) == 0 {
  368. panic("no return value specified for UserInfoQuery")
  369. }
  370. var r0 wire.SNACMessage
  371. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame) wire.SNACMessage); ok {
  372. r0 = rf(ctx, sess, frame)
  373. } else {
  374. r0 = ret.Get(0).(wire.SNACMessage)
  375. }
  376. return r0
  377. }
  378. // mockOServiceService_UserInfoQuery_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UserInfoQuery'
  379. type mockOServiceService_UserInfoQuery_Call struct {
  380. *mock.Call
  381. }
  382. // UserInfoQuery is a helper method to define mock.On call
  383. // - ctx context.Context
  384. // - sess *state.Session
  385. // - frame wire.SNACFrame
  386. func (_e *mockOServiceService_Expecter) UserInfoQuery(ctx interface{}, sess interface{}, frame interface{}) *mockOServiceService_UserInfoQuery_Call {
  387. return &mockOServiceService_UserInfoQuery_Call{Call: _e.mock.On("UserInfoQuery", ctx, sess, frame)}
  388. }
  389. func (_c *mockOServiceService_UserInfoQuery_Call) Run(run func(ctx context.Context, sess *state.Session, frame wire.SNACFrame)) *mockOServiceService_UserInfoQuery_Call {
  390. _c.Call.Run(func(args mock.Arguments) {
  391. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame))
  392. })
  393. return _c
  394. }
  395. func (_c *mockOServiceService_UserInfoQuery_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_UserInfoQuery_Call {
  396. _c.Call.Return(_a0)
  397. return _c
  398. }
  399. func (_c *mockOServiceService_UserInfoQuery_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame) wire.SNACMessage) *mockOServiceService_UserInfoQuery_Call {
  400. _c.Call.Return(run)
  401. return _c
  402. }
  403. // 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.
  404. // The first argument is typically a *testing.T value.
  405. func newMockOServiceService(t interface {
  406. mock.TestingT
  407. Cleanup(func())
  408. }) *mockOServiceService {
  409. mock := &mockOServiceService{}
  410. mock.Mock.Test(t)
  411. t.Cleanup(func() { mock.AssertExpectations(t) })
  412. return mock
  413. }