MockOServiceHandler.go 17 KB

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