MockOServiceHandler.go 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398
  1. // Code generated by mockery v2.35.2. DO NOT EDIT.
  2. package server
  3. import (
  4. io "io"
  5. oscar "github.com/mkaminski/goaim/oscar"
  6. mock "github.com/stretchr/testify/mock"
  7. )
  8. // MockOServiceHandler is an autogenerated mock type for the OServiceHandler type
  9. type MockOServiceHandler struct {
  10. mock.Mock
  11. }
  12. type MockOServiceHandler_Expecter struct {
  13. mock *mock.Mock
  14. }
  15. func (_m *MockOServiceHandler) EXPECT() *MockOServiceHandler_Expecter {
  16. return &MockOServiceHandler_Expecter{mock: &_m.Mock}
  17. }
  18. // ClientOnlineHandler provides a mock function with given fields: snacPayloadIn, onReadyCB, sess, sm, r, w, sequence
  19. func (_m *MockOServiceHandler) ClientOnlineHandler(snacPayloadIn oscar.SNAC_0x01_0x02_OServiceClientOnline, onReadyCB OnReadyCB, sess *Session, sm SessionManager, r io.Reader, w io.Writer, sequence *uint32) error {
  20. ret := _m.Called(snacPayloadIn, onReadyCB, sess, sm, r, w, sequence)
  21. var r0 error
  22. if rf, ok := ret.Get(0).(func(oscar.SNAC_0x01_0x02_OServiceClientOnline, OnReadyCB, *Session, SessionManager, io.Reader, io.Writer, *uint32) error); ok {
  23. r0 = rf(snacPayloadIn, onReadyCB, sess, sm, r, w, sequence)
  24. } else {
  25. r0 = ret.Error(0)
  26. }
  27. return r0
  28. }
  29. // MockOServiceHandler_ClientOnlineHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ClientOnlineHandler'
  30. type MockOServiceHandler_ClientOnlineHandler_Call struct {
  31. *mock.Call
  32. }
  33. // ClientOnlineHandler is a helper method to define mock.On call
  34. // - snacPayloadIn oscar.SNAC_0x01_0x02_OServiceClientOnline
  35. // - onReadyCB OnReadyCB
  36. // - sess *Session
  37. // - sm SessionManager
  38. // - r io.Reader
  39. // - w io.Writer
  40. // - sequence *uint32
  41. func (_e *MockOServiceHandler_Expecter) ClientOnlineHandler(snacPayloadIn interface{}, onReadyCB interface{}, sess interface{}, sm interface{}, r interface{}, w interface{}, sequence interface{}) *MockOServiceHandler_ClientOnlineHandler_Call {
  42. return &MockOServiceHandler_ClientOnlineHandler_Call{Call: _e.mock.On("ClientOnlineHandler", snacPayloadIn, onReadyCB, sess, sm, r, w, sequence)}
  43. }
  44. func (_c *MockOServiceHandler_ClientOnlineHandler_Call) Run(run func(snacPayloadIn oscar.SNAC_0x01_0x02_OServiceClientOnline, onReadyCB OnReadyCB, sess *Session, sm SessionManager, r io.Reader, w io.Writer, sequence *uint32)) *MockOServiceHandler_ClientOnlineHandler_Call {
  45. _c.Call.Run(func(args mock.Arguments) {
  46. run(args[0].(oscar.SNAC_0x01_0x02_OServiceClientOnline), args[1].(OnReadyCB), args[2].(*Session), args[3].(SessionManager), args[4].(io.Reader), args[5].(io.Writer), args[6].(*uint32))
  47. })
  48. return _c
  49. }
  50. func (_c *MockOServiceHandler_ClientOnlineHandler_Call) Return(_a0 error) *MockOServiceHandler_ClientOnlineHandler_Call {
  51. _c.Call.Return(_a0)
  52. return _c
  53. }
  54. func (_c *MockOServiceHandler_ClientOnlineHandler_Call) RunAndReturn(run func(oscar.SNAC_0x01_0x02_OServiceClientOnline, OnReadyCB, *Session, SessionManager, io.Reader, io.Writer, *uint32) error) *MockOServiceHandler_ClientOnlineHandler_Call {
  55. _c.Call.Return(run)
  56. return _c
  57. }
  58. // ClientVersionsHandler provides a mock function with given fields: snacPayloadIn
  59. func (_m *MockOServiceHandler) ClientVersionsHandler(snacPayloadIn oscar.SNAC_0x01_0x17_OServiceClientVersions) XMessage {
  60. ret := _m.Called(snacPayloadIn)
  61. var r0 XMessage
  62. if rf, ok := ret.Get(0).(func(oscar.SNAC_0x01_0x17_OServiceClientVersions) XMessage); ok {
  63. r0 = rf(snacPayloadIn)
  64. } else {
  65. r0 = ret.Get(0).(XMessage)
  66. }
  67. return r0
  68. }
  69. // MockOServiceHandler_ClientVersionsHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ClientVersionsHandler'
  70. type MockOServiceHandler_ClientVersionsHandler_Call struct {
  71. *mock.Call
  72. }
  73. // ClientVersionsHandler is a helper method to define mock.On call
  74. // - snacPayloadIn oscar.SNAC_0x01_0x17_OServiceClientVersions
  75. func (_e *MockOServiceHandler_Expecter) ClientVersionsHandler(snacPayloadIn interface{}) *MockOServiceHandler_ClientVersionsHandler_Call {
  76. return &MockOServiceHandler_ClientVersionsHandler_Call{Call: _e.mock.On("ClientVersionsHandler", snacPayloadIn)}
  77. }
  78. func (_c *MockOServiceHandler_ClientVersionsHandler_Call) Run(run func(snacPayloadIn oscar.SNAC_0x01_0x17_OServiceClientVersions)) *MockOServiceHandler_ClientVersionsHandler_Call {
  79. _c.Call.Run(func(args mock.Arguments) {
  80. run(args[0].(oscar.SNAC_0x01_0x17_OServiceClientVersions))
  81. })
  82. return _c
  83. }
  84. func (_c *MockOServiceHandler_ClientVersionsHandler_Call) Return(_a0 XMessage) *MockOServiceHandler_ClientVersionsHandler_Call {
  85. _c.Call.Return(_a0)
  86. return _c
  87. }
  88. func (_c *MockOServiceHandler_ClientVersionsHandler_Call) RunAndReturn(run func(oscar.SNAC_0x01_0x17_OServiceClientVersions) XMessage) *MockOServiceHandler_ClientVersionsHandler_Call {
  89. _c.Call.Return(run)
  90. return _c
  91. }
  92. // IdleNotificationHandler provides a mock function with given fields: sess, sm, fm, snacPayloadIn
  93. func (_m *MockOServiceHandler) IdleNotificationHandler(sess *Session, sm SessionManager, fm *FeedbagStore, snacPayloadIn oscar.SNAC_0x01_0x11_OServiceIdleNotification) error {
  94. ret := _m.Called(sess, sm, fm, snacPayloadIn)
  95. var r0 error
  96. if rf, ok := ret.Get(0).(func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x11_OServiceIdleNotification) error); ok {
  97. r0 = rf(sess, sm, fm, snacPayloadIn)
  98. } else {
  99. r0 = ret.Error(0)
  100. }
  101. return r0
  102. }
  103. // MockOServiceHandler_IdleNotificationHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IdleNotificationHandler'
  104. type MockOServiceHandler_IdleNotificationHandler_Call struct {
  105. *mock.Call
  106. }
  107. // IdleNotificationHandler is a helper method to define mock.On call
  108. // - sess *Session
  109. // - sm SessionManager
  110. // - fm *FeedbagStore
  111. // - snacPayloadIn oscar.SNAC_0x01_0x11_OServiceIdleNotification
  112. func (_e *MockOServiceHandler_Expecter) IdleNotificationHandler(sess interface{}, sm interface{}, fm interface{}, snacPayloadIn interface{}) *MockOServiceHandler_IdleNotificationHandler_Call {
  113. return &MockOServiceHandler_IdleNotificationHandler_Call{Call: _e.mock.On("IdleNotificationHandler", sess, sm, fm, snacPayloadIn)}
  114. }
  115. func (_c *MockOServiceHandler_IdleNotificationHandler_Call) Run(run func(sess *Session, sm SessionManager, fm *FeedbagStore, snacPayloadIn oscar.SNAC_0x01_0x11_OServiceIdleNotification)) *MockOServiceHandler_IdleNotificationHandler_Call {
  116. _c.Call.Run(func(args mock.Arguments) {
  117. run(args[0].(*Session), args[1].(SessionManager), args[2].(*FeedbagStore), args[3].(oscar.SNAC_0x01_0x11_OServiceIdleNotification))
  118. })
  119. return _c
  120. }
  121. func (_c *MockOServiceHandler_IdleNotificationHandler_Call) Return(_a0 error) *MockOServiceHandler_IdleNotificationHandler_Call {
  122. _c.Call.Return(_a0)
  123. return _c
  124. }
  125. func (_c *MockOServiceHandler_IdleNotificationHandler_Call) RunAndReturn(run func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x11_OServiceIdleNotification) error) *MockOServiceHandler_IdleNotificationHandler_Call {
  126. _c.Call.Return(run)
  127. return _c
  128. }
  129. // RateParamsQueryHandler provides a mock function with given fields:
  130. func (_m *MockOServiceHandler) RateParamsQueryHandler() XMessage {
  131. ret := _m.Called()
  132. var r0 XMessage
  133. if rf, ok := ret.Get(0).(func() XMessage); ok {
  134. r0 = rf()
  135. } else {
  136. r0 = ret.Get(0).(XMessage)
  137. }
  138. return r0
  139. }
  140. // MockOServiceHandler_RateParamsQueryHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsQueryHandler'
  141. type MockOServiceHandler_RateParamsQueryHandler_Call struct {
  142. *mock.Call
  143. }
  144. // RateParamsQueryHandler is a helper method to define mock.On call
  145. func (_e *MockOServiceHandler_Expecter) RateParamsQueryHandler() *MockOServiceHandler_RateParamsQueryHandler_Call {
  146. return &MockOServiceHandler_RateParamsQueryHandler_Call{Call: _e.mock.On("RateParamsQueryHandler")}
  147. }
  148. func (_c *MockOServiceHandler_RateParamsQueryHandler_Call) Run(run func()) *MockOServiceHandler_RateParamsQueryHandler_Call {
  149. _c.Call.Run(func(args mock.Arguments) {
  150. run()
  151. })
  152. return _c
  153. }
  154. func (_c *MockOServiceHandler_RateParamsQueryHandler_Call) Return(_a0 XMessage) *MockOServiceHandler_RateParamsQueryHandler_Call {
  155. _c.Call.Return(_a0)
  156. return _c
  157. }
  158. func (_c *MockOServiceHandler_RateParamsQueryHandler_Call) RunAndReturn(run func() XMessage) *MockOServiceHandler_RateParamsQueryHandler_Call {
  159. _c.Call.Return(run)
  160. return _c
  161. }
  162. // RateParamsSubAddHandler provides a mock function with given fields: _a0
  163. func (_m *MockOServiceHandler) RateParamsSubAddHandler(_a0 oscar.SNAC_0x01_0x08_OServiceRateParamsSubAdd) {
  164. _m.Called(_a0)
  165. }
  166. // MockOServiceHandler_RateParamsSubAddHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RateParamsSubAddHandler'
  167. type MockOServiceHandler_RateParamsSubAddHandler_Call struct {
  168. *mock.Call
  169. }
  170. // RateParamsSubAddHandler is a helper method to define mock.On call
  171. // - _a0 oscar.SNAC_0x01_0x08_OServiceRateParamsSubAdd
  172. func (_e *MockOServiceHandler_Expecter) RateParamsSubAddHandler(_a0 interface{}) *MockOServiceHandler_RateParamsSubAddHandler_Call {
  173. return &MockOServiceHandler_RateParamsSubAddHandler_Call{Call: _e.mock.On("RateParamsSubAddHandler", _a0)}
  174. }
  175. func (_c *MockOServiceHandler_RateParamsSubAddHandler_Call) Run(run func(_a0 oscar.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *MockOServiceHandler_RateParamsSubAddHandler_Call {
  176. _c.Call.Run(func(args mock.Arguments) {
  177. run(args[0].(oscar.SNAC_0x01_0x08_OServiceRateParamsSubAdd))
  178. })
  179. return _c
  180. }
  181. func (_c *MockOServiceHandler_RateParamsSubAddHandler_Call) Return() *MockOServiceHandler_RateParamsSubAddHandler_Call {
  182. _c.Call.Return()
  183. return _c
  184. }
  185. func (_c *MockOServiceHandler_RateParamsSubAddHandler_Call) RunAndReturn(run func(oscar.SNAC_0x01_0x08_OServiceRateParamsSubAdd)) *MockOServiceHandler_RateParamsSubAddHandler_Call {
  186. _c.Call.Return(run)
  187. return _c
  188. }
  189. // ServiceRequestHandler provides a mock function with given fields: cfg, cr, sess, snacPayloadIn
  190. func (_m *MockOServiceHandler) ServiceRequestHandler(cfg Config, cr *ChatRegistry, sess *Session, snacPayloadIn oscar.SNAC_0x01_0x04_OServiceServiceRequest) (XMessage, error) {
  191. ret := _m.Called(cfg, cr, sess, snacPayloadIn)
  192. var r0 XMessage
  193. var r1 error
  194. if rf, ok := ret.Get(0).(func(Config, *ChatRegistry, *Session, oscar.SNAC_0x01_0x04_OServiceServiceRequest) (XMessage, error)); ok {
  195. return rf(cfg, cr, sess, snacPayloadIn)
  196. }
  197. if rf, ok := ret.Get(0).(func(Config, *ChatRegistry, *Session, oscar.SNAC_0x01_0x04_OServiceServiceRequest) XMessage); ok {
  198. r0 = rf(cfg, cr, sess, snacPayloadIn)
  199. } else {
  200. r0 = ret.Get(0).(XMessage)
  201. }
  202. if rf, ok := ret.Get(1).(func(Config, *ChatRegistry, *Session, oscar.SNAC_0x01_0x04_OServiceServiceRequest) error); ok {
  203. r1 = rf(cfg, cr, sess, snacPayloadIn)
  204. } else {
  205. r1 = ret.Error(1)
  206. }
  207. return r0, r1
  208. }
  209. // MockOServiceHandler_ServiceRequestHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ServiceRequestHandler'
  210. type MockOServiceHandler_ServiceRequestHandler_Call struct {
  211. *mock.Call
  212. }
  213. // ServiceRequestHandler is a helper method to define mock.On call
  214. // - cfg Config
  215. // - cr *ChatRegistry
  216. // - sess *Session
  217. // - snacPayloadIn oscar.SNAC_0x01_0x04_OServiceServiceRequest
  218. func (_e *MockOServiceHandler_Expecter) ServiceRequestHandler(cfg interface{}, cr interface{}, sess interface{}, snacPayloadIn interface{}) *MockOServiceHandler_ServiceRequestHandler_Call {
  219. return &MockOServiceHandler_ServiceRequestHandler_Call{Call: _e.mock.On("ServiceRequestHandler", cfg, cr, sess, snacPayloadIn)}
  220. }
  221. func (_c *MockOServiceHandler_ServiceRequestHandler_Call) Run(run func(cfg Config, cr *ChatRegistry, sess *Session, snacPayloadIn oscar.SNAC_0x01_0x04_OServiceServiceRequest)) *MockOServiceHandler_ServiceRequestHandler_Call {
  222. _c.Call.Run(func(args mock.Arguments) {
  223. run(args[0].(Config), args[1].(*ChatRegistry), args[2].(*Session), args[3].(oscar.SNAC_0x01_0x04_OServiceServiceRequest))
  224. })
  225. return _c
  226. }
  227. func (_c *MockOServiceHandler_ServiceRequestHandler_Call) Return(_a0 XMessage, _a1 error) *MockOServiceHandler_ServiceRequestHandler_Call {
  228. _c.Call.Return(_a0, _a1)
  229. return _c
  230. }
  231. func (_c *MockOServiceHandler_ServiceRequestHandler_Call) RunAndReturn(run func(Config, *ChatRegistry, *Session, oscar.SNAC_0x01_0x04_OServiceServiceRequest) (XMessage, error)) *MockOServiceHandler_ServiceRequestHandler_Call {
  232. _c.Call.Return(run)
  233. return _c
  234. }
  235. // SetUserInfoFieldsHandler provides a mock function with given fields: sess, sm, fm, snacPayloadIn
  236. func (_m *MockOServiceHandler) SetUserInfoFieldsHandler(sess *Session, sm SessionManager, fm *FeedbagStore, snacPayloadIn oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (XMessage, error) {
  237. ret := _m.Called(sess, sm, fm, snacPayloadIn)
  238. var r0 XMessage
  239. var r1 error
  240. if rf, ok := ret.Get(0).(func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (XMessage, error)); ok {
  241. return rf(sess, sm, fm, snacPayloadIn)
  242. }
  243. if rf, ok := ret.Get(0).(func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields) XMessage); ok {
  244. r0 = rf(sess, sm, fm, snacPayloadIn)
  245. } else {
  246. r0 = ret.Get(0).(XMessage)
  247. }
  248. if rf, ok := ret.Get(1).(func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields) error); ok {
  249. r1 = rf(sess, sm, fm, snacPayloadIn)
  250. } else {
  251. r1 = ret.Error(1)
  252. }
  253. return r0, r1
  254. }
  255. // MockOServiceHandler_SetUserInfoFieldsHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetUserInfoFieldsHandler'
  256. type MockOServiceHandler_SetUserInfoFieldsHandler_Call struct {
  257. *mock.Call
  258. }
  259. // SetUserInfoFieldsHandler is a helper method to define mock.On call
  260. // - sess *Session
  261. // - sm SessionManager
  262. // - fm *FeedbagStore
  263. // - snacPayloadIn oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields
  264. func (_e *MockOServiceHandler_Expecter) SetUserInfoFieldsHandler(sess interface{}, sm interface{}, fm interface{}, snacPayloadIn interface{}) *MockOServiceHandler_SetUserInfoFieldsHandler_Call {
  265. return &MockOServiceHandler_SetUserInfoFieldsHandler_Call{Call: _e.mock.On("SetUserInfoFieldsHandler", sess, sm, fm, snacPayloadIn)}
  266. }
  267. func (_c *MockOServiceHandler_SetUserInfoFieldsHandler_Call) Run(run func(sess *Session, sm SessionManager, fm *FeedbagStore, snacPayloadIn oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields)) *MockOServiceHandler_SetUserInfoFieldsHandler_Call {
  268. _c.Call.Run(func(args mock.Arguments) {
  269. run(args[0].(*Session), args[1].(SessionManager), args[2].(*FeedbagStore), args[3].(oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields))
  270. })
  271. return _c
  272. }
  273. func (_c *MockOServiceHandler_SetUserInfoFieldsHandler_Call) Return(_a0 XMessage, _a1 error) *MockOServiceHandler_SetUserInfoFieldsHandler_Call {
  274. _c.Call.Return(_a0, _a1)
  275. return _c
  276. }
  277. func (_c *MockOServiceHandler_SetUserInfoFieldsHandler_Call) RunAndReturn(run func(*Session, SessionManager, *FeedbagStore, oscar.SNAC_0x01_0x1E_OServiceSetUserInfoFields) (XMessage, error)) *MockOServiceHandler_SetUserInfoFieldsHandler_Call {
  278. _c.Call.Return(run)
  279. return _c
  280. }
  281. // UserInfoQueryHandler provides a mock function with given fields: sess
  282. func (_m *MockOServiceHandler) UserInfoQueryHandler(sess *Session) XMessage {
  283. ret := _m.Called(sess)
  284. var r0 XMessage
  285. if rf, ok := ret.Get(0).(func(*Session) XMessage); ok {
  286. r0 = rf(sess)
  287. } else {
  288. r0 = ret.Get(0).(XMessage)
  289. }
  290. return r0
  291. }
  292. // MockOServiceHandler_UserInfoQueryHandler_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UserInfoQueryHandler'
  293. type MockOServiceHandler_UserInfoQueryHandler_Call struct {
  294. *mock.Call
  295. }
  296. // UserInfoQueryHandler is a helper method to define mock.On call
  297. // - sess *Session
  298. func (_e *MockOServiceHandler_Expecter) UserInfoQueryHandler(sess interface{}) *MockOServiceHandler_UserInfoQueryHandler_Call {
  299. return &MockOServiceHandler_UserInfoQueryHandler_Call{Call: _e.mock.On("UserInfoQueryHandler", sess)}
  300. }
  301. func (_c *MockOServiceHandler_UserInfoQueryHandler_Call) Run(run func(sess *Session)) *MockOServiceHandler_UserInfoQueryHandler_Call {
  302. _c.Call.Run(func(args mock.Arguments) {
  303. run(args[0].(*Session))
  304. })
  305. return _c
  306. }
  307. func (_c *MockOServiceHandler_UserInfoQueryHandler_Call) Return(_a0 XMessage) *MockOServiceHandler_UserInfoQueryHandler_Call {
  308. _c.Call.Return(_a0)
  309. return _c
  310. }
  311. func (_c *MockOServiceHandler_UserInfoQueryHandler_Call) RunAndReturn(run func(*Session) XMessage) *MockOServiceHandler_UserInfoQueryHandler_Call {
  312. _c.Call.Return(run)
  313. return _c
  314. }
  315. // NewMockOServiceHandler creates a new instance of MockOServiceHandler. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  316. // The first argument is typically a *testing.T value.
  317. func NewMockOServiceHandler(t interface {
  318. mock.TestingT
  319. Cleanup(func())
  320. }) *MockOServiceHandler {
  321. mock := &MockOServiceHandler{}
  322. mock.Mock.Test(t)
  323. t.Cleanup(func() { mock.AssertExpectations(t) })
  324. return mock
  325. }