mock_oservice_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510
  1. // Code generated by mockery v2.52.1. 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 no 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, sess, frame
  173. func (_m *mockOServiceService) RateParamsQuery(ctx context.Context, sess *state.Session, frame wire.SNACFrame) wire.SNACMessage {
  174. ret := _m.Called(ctx, sess, 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, *state.Session, wire.SNACFrame) wire.SNACMessage); ok {
  180. r0 = rf(ctx, sess, 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. // - sess *state.Session
  193. // - frame wire.SNACFrame
  194. func (_e *mockOServiceService_Expecter) RateParamsQuery(ctx interface{}, sess interface{}, frame interface{}) *mockOServiceService_RateParamsQuery_Call {
  195. return &mockOServiceService_RateParamsQuery_Call{Call: _e.mock.On("RateParamsQuery", ctx, sess, frame)}
  196. }
  197. func (_c *mockOServiceService_RateParamsQuery_Call) Run(run func(ctx context.Context, sess *state.Session, frame wire.SNACFrame)) *mockOServiceService_RateParamsQuery_Call {
  198. _c.Call.Run(func(args mock.Arguments) {
  199. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame))
  200. })
  201. return _c
  202. }
  203. func (_c *mockOServiceService_RateParamsQuery_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_RateParamsQuery_Call {
  204. _c.Call.Return(_a0)
  205. return _c
  206. }
  207. func (_c *mockOServiceService_RateParamsQuery_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame) wire.SNACMessage) *mockOServiceService_RateParamsQuery_Call {
  208. _c.Call.Return(run)
  209. return _c
  210. }
  211. // RateParamsSubAdd provides a mock function with given fields: _a0, _a1
  212. func (_m *mockOServiceService) RateParamsSubAdd(_a0 context.Context, _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd) {
  213. _m.Called(_a0, _a1)
  214. }
  215. // mockOServiceService_RateParamsSubAdd_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsSubAdd'
  216. type mockOServiceService_RateParamsSubAdd_Call struct {
  217. *mock.Call
  218. }
  219. // RateParamsSubAdd is a helper method to define mock.On call
  220. // - _a0 context.Context
  221. // - _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd
  222. func (_e *mockOServiceService_Expecter) RateParamsSubAdd(_a0 interface{}, _a1 interface{}) *mockOServiceService_RateParamsSubAdd_Call {
  223. return &mockOServiceService_RateParamsSubAdd_Call{Call: _e.mock.On("RateParamsSubAdd", _a0, _a1)}
  224. }
  225. func (_c *mockOServiceService_RateParamsSubAdd_Call) Run(run func(_a0 context.Context, _a1 wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  226. _c.Call.Run(func(args mock.Arguments) {
  227. run(args[0].(context.Context), args[1].(wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd))
  228. })
  229. return _c
  230. }
  231. func (_c *mockOServiceService_RateParamsSubAdd_Call) Return() *mockOServiceService_RateParamsSubAdd_Call {
  232. _c.Call.Return()
  233. return _c
  234. }
  235. func (_c *mockOServiceService_RateParamsSubAdd_Call) RunAndReturn(run func(context.Context, wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *mockOServiceService_RateParamsSubAdd_Call {
  236. _c.Run(run)
  237. return _c
  238. }
  239. // ServiceRequest provides a mock function with given fields: ctx, sess, frame, bodyIn
  240. func (_m *mockOServiceService) ServiceRequest(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x04_OServiceServiceRequest) (wire.SNACMessage, error) {
  241. ret := _m.Called(ctx, sess, frame, bodyIn)
  242. if len(ret) == 0 {
  243. panic("no return value specified for ServiceRequest")
  244. }
  245. var r0 wire.SNACMessage
  246. var r1 error
  247. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) (wire.SNACMessage, error)); ok {
  248. return rf(ctx, sess, frame, bodyIn)
  249. }
  250. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) wire.SNACMessage); ok {
  251. r0 = rf(ctx, sess, frame, bodyIn)
  252. } else {
  253. r0 = ret.Get(0).(wire.SNACMessage)
  254. }
  255. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x04_OServiceServiceRequest) error); ok {
  256. r1 = rf(ctx, sess, frame, bodyIn)
  257. } else {
  258. r1 = ret.Error(1)
  259. }
  260. return r0, r1
  261. }
  262. // mockOServiceService_ServiceRequest_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ServiceRequest'
  263. type mockOServiceService_ServiceRequest_Call struct {
  264. *mock.Call
  265. }
  266. // ServiceRequest is a helper method to define mock.On call
  267. // - ctx context.Context
  268. // - sess *state.Session
  269. // - frame wire.SNACFrame
  270. // - bodyIn wire.SNAC_0x01_0x04_OServiceServiceRequest
  271. func (_e *mockOServiceService_Expecter) ServiceRequest(ctx interface{}, sess interface{}, frame interface{}, bodyIn interface{}) *mockOServiceService_ServiceRequest_Call {
  272. return &mockOServiceService_ServiceRequest_Call{Call: _e.mock.On("ServiceRequest", ctx, sess, frame, bodyIn)}
  273. }
  274. 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 {
  275. _c.Call.Run(func(args mock.Arguments) {
  276. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x01_0x04_OServiceServiceRequest))
  277. })
  278. return _c
  279. }
  280. func (_c *mockOServiceService_ServiceRequest_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockOServiceService_ServiceRequest_Call {
  281. _c.Call.Return(_a0, _a1)
  282. return _c
  283. }
  284. 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 {
  285. _c.Call.Return(run)
  286. return _c
  287. }
  288. // SetPrivacyFlags provides a mock function with given fields: ctx, bodyIn
  289. func (_m *mockOServiceService) SetPrivacyFlags(ctx context.Context, bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags) {
  290. _m.Called(ctx, bodyIn)
  291. }
  292. // mockOServiceService_SetPrivacyFlags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetPrivacyFlags'
  293. type mockOServiceService_SetPrivacyFlags_Call struct {
  294. *mock.Call
  295. }
  296. // SetPrivacyFlags is a helper method to define mock.On call
  297. // - ctx context.Context
  298. // - bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags
  299. func (_e *mockOServiceService_Expecter) SetPrivacyFlags(ctx interface{}, bodyIn interface{}) *mockOServiceService_SetPrivacyFlags_Call {
  300. return &mockOServiceService_SetPrivacyFlags_Call{Call: _e.mock.On("SetPrivacyFlags", ctx, bodyIn)}
  301. }
  302. func (_c *mockOServiceService_SetPrivacyFlags_Call) Run(run func(ctx context.Context, bodyIn wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags)) *mockOServiceService_SetPrivacyFlags_Call {
  303. _c.Call.Run(func(args mock.Arguments) {
  304. run(args[0].(context.Context), args[1].(wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags))
  305. })
  306. return _c
  307. }
  308. func (_c *mockOServiceService_SetPrivacyFlags_Call) Return() *mockOServiceService_SetPrivacyFlags_Call {
  309. _c.Call.Return()
  310. return _c
  311. }
  312. func (_c *mockOServiceService_SetPrivacyFlags_Call) RunAndReturn(run func(context.Context, wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags)) *mockOServiceService_SetPrivacyFlags_Call {
  313. _c.Run(run)
  314. return _c
  315. }
  316. // SetUserInfoFields provides a mock function with given fields: ctx, sess, frame, bodyIn
  317. func (_m *mockOServiceService) SetUserInfoFields(ctx context.Context, sess *state.Session, frame wire.SNACFrame, bodyIn wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (wire.SNACMessage, error) {
  318. ret := _m.Called(ctx, sess, frame, bodyIn)
  319. if len(ret) == 0 {
  320. panic("no return value specified for SetUserInfoFields")
  321. }
  322. var r0 wire.SNACMessage
  323. var r1 error
  324. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (wire.SNACMessage, error)); ok {
  325. return rf(ctx, sess, frame, bodyIn)
  326. }
  327. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) wire.SNACMessage); ok {
  328. r0 = rf(ctx, sess, frame, bodyIn)
  329. } else {
  330. r0 = ret.Get(0).(wire.SNACMessage)
  331. }
  332. if rf, ok := ret.Get(1).(func(context.Context, *state.Session, wire.SNACFrame, wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error); ok {
  333. r1 = rf(ctx, sess, frame, bodyIn)
  334. } else {
  335. r1 = ret.Error(1)
  336. }
  337. return r0, r1
  338. }
  339. // mockOServiceService_SetUserInfoFields_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetUserInfoFields'
  340. type mockOServiceService_SetUserInfoFields_Call struct {
  341. *mock.Call
  342. }
  343. // SetUserInfoFields is a helper method to define mock.On call
  344. // - ctx context.Context
  345. // - sess *state.Session
  346. // - frame wire.SNACFrame
  347. // - bodyIn wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields
  348. func (_e *mockOServiceService_Expecter) SetUserInfoFields(ctx interface{}, sess interface{}, frame interface{}, bodyIn interface{}) *mockOServiceService_SetUserInfoFields_Call {
  349. return &mockOServiceService_SetUserInfoFields_Call{Call: _e.mock.On("SetUserInfoFields", ctx, sess, frame, bodyIn)}
  350. }
  351. 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 {
  352. _c.Call.Run(func(args mock.Arguments) {
  353. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame), args[3].(wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields))
  354. })
  355. return _c
  356. }
  357. func (_c *mockOServiceService_SetUserInfoFields_Call) Return(_a0 wire.SNACMessage, _a1 error) *mockOServiceService_SetUserInfoFields_Call {
  358. _c.Call.Return(_a0, _a1)
  359. return _c
  360. }
  361. 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 {
  362. _c.Call.Return(run)
  363. return _c
  364. }
  365. // UserInfoQuery provides a mock function with given fields: ctx, sess, frame
  366. func (_m *mockOServiceService) UserInfoQuery(ctx context.Context, sess *state.Session, frame wire.SNACFrame) wire.SNACMessage {
  367. ret := _m.Called(ctx, sess, frame)
  368. if len(ret) == 0 {
  369. panic("no return value specified for UserInfoQuery")
  370. }
  371. var r0 wire.SNACMessage
  372. if rf, ok := ret.Get(0).(func(context.Context, *state.Session, wire.SNACFrame) wire.SNACMessage); ok {
  373. r0 = rf(ctx, sess, frame)
  374. } else {
  375. r0 = ret.Get(0).(wire.SNACMessage)
  376. }
  377. return r0
  378. }
  379. // mockOServiceService_UserInfoQuery_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UserInfoQuery'
  380. type mockOServiceService_UserInfoQuery_Call struct {
  381. *mock.Call
  382. }
  383. // UserInfoQuery is a helper method to define mock.On call
  384. // - ctx context.Context
  385. // - sess *state.Session
  386. // - frame wire.SNACFrame
  387. func (_e *mockOServiceService_Expecter) UserInfoQuery(ctx interface{}, sess interface{}, frame interface{}) *mockOServiceService_UserInfoQuery_Call {
  388. return &mockOServiceService_UserInfoQuery_Call{Call: _e.mock.On("UserInfoQuery", ctx, sess, frame)}
  389. }
  390. func (_c *mockOServiceService_UserInfoQuery_Call) Run(run func(ctx context.Context, sess *state.Session, frame wire.SNACFrame)) *mockOServiceService_UserInfoQuery_Call {
  391. _c.Call.Run(func(args mock.Arguments) {
  392. run(args[0].(context.Context), args[1].(*state.Session), args[2].(wire.SNACFrame))
  393. })
  394. return _c
  395. }
  396. func (_c *mockOServiceService_UserInfoQuery_Call) Return(_a0 wire.SNACMessage) *mockOServiceService_UserInfoQuery_Call {
  397. _c.Call.Return(_a0)
  398. return _c
  399. }
  400. func (_c *mockOServiceService_UserInfoQuery_Call) RunAndReturn(run func(context.Context, *state.Session, wire.SNACFrame) wire.SNACMessage) *mockOServiceService_UserInfoQuery_Call {
  401. _c.Call.Return(run)
  402. return _c
  403. }
  404. // 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.
  405. // The first argument is typically a *testing.T value.
  406. func newMockOServiceService(t interface {
  407. mock.TestingT
  408. Cleanup(func())
  409. }) *mockOServiceService {
  410. mock := &mockOServiceService{}
  411. mock.Mock.Test(t)
  412. t.Cleanup(func() { mock.AssertExpectations(t) })
  413. return mock
  414. }