MockOServiceHandler.go 18 KB

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