mock_buddy_service_test.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package toc
  5. import (
  6. "context"
  7. "github.com/mk6i/open-oscar-server/state"
  8. "github.com/mk6i/open-oscar-server/wire"
  9. mock "github.com/stretchr/testify/mock"
  10. )
  11. // newMockBuddyService creates a new instance of mockBuddyService. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  12. // The first argument is typically a *testing.T value.
  13. func newMockBuddyService(t interface {
  14. mock.TestingT
  15. Cleanup(func())
  16. }) *mockBuddyService {
  17. if helper, ok := t.(interface{ Helper() }); ok {
  18. helper.Helper()
  19. }
  20. mock := &mockBuddyService{}
  21. mock.Mock.Test(t)
  22. t.Cleanup(func() {
  23. if helper, ok := t.(interface{ Helper() }); ok {
  24. helper.Helper()
  25. }
  26. mock.AssertExpectations(t)
  27. })
  28. return mock
  29. }
  30. // mockBuddyService is an autogenerated mock type for the BuddyService type
  31. type mockBuddyService struct {
  32. mock.Mock
  33. }
  34. type mockBuddyService_Expecter struct {
  35. mock *mock.Mock
  36. }
  37. func (_m *mockBuddyService) EXPECT() *mockBuddyService_Expecter {
  38. return &mockBuddyService_Expecter{mock: &_m.Mock}
  39. }
  40. // AddBuddies provides a mock function for the type mockBuddyService
  41. func (_mock *mockBuddyService) AddBuddies(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x03_0x04_BuddyAddBuddies) (*wire.SNACMessage, error) {
  42. ret := _mock.Called(ctx, instance, inFrame, inBody)
  43. if len(ret) == 0 {
  44. panic("no return value specified for AddBuddies")
  45. }
  46. var r0 *wire.SNACMessage
  47. var r1 error
  48. if returnFunc, ok := ret.Get(0).(func(context.Context, *state.SessionInstance, wire.SNACFrame, wire.SNAC_0x03_0x04_BuddyAddBuddies) (*wire.SNACMessage, error)); ok {
  49. return returnFunc(ctx, instance, inFrame, inBody)
  50. }
  51. if returnFunc, ok := ret.Get(0).(func(context.Context, *state.SessionInstance, wire.SNACFrame, wire.SNAC_0x03_0x04_BuddyAddBuddies) *wire.SNACMessage); ok {
  52. r0 = returnFunc(ctx, instance, inFrame, inBody)
  53. } else {
  54. if ret.Get(0) != nil {
  55. r0 = ret.Get(0).(*wire.SNACMessage)
  56. }
  57. }
  58. if returnFunc, ok := ret.Get(1).(func(context.Context, *state.SessionInstance, wire.SNACFrame, wire.SNAC_0x03_0x04_BuddyAddBuddies) error); ok {
  59. r1 = returnFunc(ctx, instance, inFrame, inBody)
  60. } else {
  61. r1 = ret.Error(1)
  62. }
  63. return r0, r1
  64. }
  65. // mockBuddyService_AddBuddies_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddBuddies'
  66. type mockBuddyService_AddBuddies_Call struct {
  67. *mock.Call
  68. }
  69. // AddBuddies is a helper method to define mock.On call
  70. // - ctx context.Context
  71. // - instance *state.SessionInstance
  72. // - inFrame wire.SNACFrame
  73. // - inBody wire.SNAC_0x03_0x04_BuddyAddBuddies
  74. func (_e *mockBuddyService_Expecter) AddBuddies(ctx any, instance any, inFrame any, inBody any) *mockBuddyService_AddBuddies_Call {
  75. return &mockBuddyService_AddBuddies_Call{Call: _e.mock.On("AddBuddies", ctx, instance, inFrame, inBody)}
  76. }
  77. func (_c *mockBuddyService_AddBuddies_Call) Run(run func(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x03_0x04_BuddyAddBuddies)) *mockBuddyService_AddBuddies_Call {
  78. _c.Call.Run(func(args mock.Arguments) {
  79. var arg0 context.Context
  80. if args[0] != nil {
  81. arg0 = args[0].(context.Context)
  82. }
  83. var arg1 *state.SessionInstance
  84. if args[1] != nil {
  85. arg1 = args[1].(*state.SessionInstance)
  86. }
  87. var arg2 wire.SNACFrame
  88. if args[2] != nil {
  89. arg2 = args[2].(wire.SNACFrame)
  90. }
  91. var arg3 wire.SNAC_0x03_0x04_BuddyAddBuddies
  92. if args[3] != nil {
  93. arg3 = args[3].(wire.SNAC_0x03_0x04_BuddyAddBuddies)
  94. }
  95. run(
  96. arg0,
  97. arg1,
  98. arg2,
  99. arg3,
  100. )
  101. })
  102. return _c
  103. }
  104. func (_c *mockBuddyService_AddBuddies_Call) Return(sNACMessage *wire.SNACMessage, err error) *mockBuddyService_AddBuddies_Call {
  105. _c.Call.Return(sNACMessage, err)
  106. return _c
  107. }
  108. func (_c *mockBuddyService_AddBuddies_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, inBody wire.SNAC_0x03_0x04_BuddyAddBuddies) (*wire.SNACMessage, error)) *mockBuddyService_AddBuddies_Call {
  109. _c.Call.Return(run)
  110. return _c
  111. }
  112. // BroadcastBuddyArrived provides a mock function for the type mockBuddyService
  113. func (_mock *mockBuddyService) BroadcastBuddyArrived(ctx context.Context, screenName state.IdentScreenName, userInfo wire.TLVUserInfo) error {
  114. ret := _mock.Called(ctx, screenName, userInfo)
  115. if len(ret) == 0 {
  116. panic("no return value specified for BroadcastBuddyArrived")
  117. }
  118. var r0 error
  119. if returnFunc, ok := ret.Get(0).(func(context.Context, state.IdentScreenName, wire.TLVUserInfo) error); ok {
  120. r0 = returnFunc(ctx, screenName, userInfo)
  121. } else {
  122. r0 = ret.Error(0)
  123. }
  124. return r0
  125. }
  126. // mockBuddyService_BroadcastBuddyArrived_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BroadcastBuddyArrived'
  127. type mockBuddyService_BroadcastBuddyArrived_Call struct {
  128. *mock.Call
  129. }
  130. // BroadcastBuddyArrived is a helper method to define mock.On call
  131. // - ctx context.Context
  132. // - screenName state.IdentScreenName
  133. // - userInfo wire.TLVUserInfo
  134. func (_e *mockBuddyService_Expecter) BroadcastBuddyArrived(ctx any, screenName any, userInfo any) *mockBuddyService_BroadcastBuddyArrived_Call {
  135. return &mockBuddyService_BroadcastBuddyArrived_Call{Call: _e.mock.On("BroadcastBuddyArrived", ctx, screenName, userInfo)}
  136. }
  137. func (_c *mockBuddyService_BroadcastBuddyArrived_Call) Run(run func(ctx context.Context, screenName state.IdentScreenName, userInfo wire.TLVUserInfo)) *mockBuddyService_BroadcastBuddyArrived_Call {
  138. _c.Call.Run(func(args mock.Arguments) {
  139. var arg0 context.Context
  140. if args[0] != nil {
  141. arg0 = args[0].(context.Context)
  142. }
  143. var arg1 state.IdentScreenName
  144. if args[1] != nil {
  145. arg1 = args[1].(state.IdentScreenName)
  146. }
  147. var arg2 wire.TLVUserInfo
  148. if args[2] != nil {
  149. arg2 = args[2].(wire.TLVUserInfo)
  150. }
  151. run(
  152. arg0,
  153. arg1,
  154. arg2,
  155. )
  156. })
  157. return _c
  158. }
  159. func (_c *mockBuddyService_BroadcastBuddyArrived_Call) Return(err error) *mockBuddyService_BroadcastBuddyArrived_Call {
  160. _c.Call.Return(err)
  161. return _c
  162. }
  163. func (_c *mockBuddyService_BroadcastBuddyArrived_Call) RunAndReturn(run func(ctx context.Context, screenName state.IdentScreenName, userInfo wire.TLVUserInfo) error) *mockBuddyService_BroadcastBuddyArrived_Call {
  164. _c.Call.Return(run)
  165. return _c
  166. }
  167. // BroadcastBuddyDeparted provides a mock function for the type mockBuddyService
  168. func (_mock *mockBuddyService) BroadcastBuddyDeparted(ctx context.Context, screenName state.IdentScreenName) error {
  169. ret := _mock.Called(ctx, screenName)
  170. if len(ret) == 0 {
  171. panic("no return value specified for BroadcastBuddyDeparted")
  172. }
  173. var r0 error
  174. if returnFunc, ok := ret.Get(0).(func(context.Context, state.IdentScreenName) error); ok {
  175. r0 = returnFunc(ctx, screenName)
  176. } else {
  177. r0 = ret.Error(0)
  178. }
  179. return r0
  180. }
  181. // mockBuddyService_BroadcastBuddyDeparted_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BroadcastBuddyDeparted'
  182. type mockBuddyService_BroadcastBuddyDeparted_Call struct {
  183. *mock.Call
  184. }
  185. // BroadcastBuddyDeparted is a helper method to define mock.On call
  186. // - ctx context.Context
  187. // - screenName state.IdentScreenName
  188. func (_e *mockBuddyService_Expecter) BroadcastBuddyDeparted(ctx any, screenName any) *mockBuddyService_BroadcastBuddyDeparted_Call {
  189. return &mockBuddyService_BroadcastBuddyDeparted_Call{Call: _e.mock.On("BroadcastBuddyDeparted", ctx, screenName)}
  190. }
  191. func (_c *mockBuddyService_BroadcastBuddyDeparted_Call) Run(run func(ctx context.Context, screenName state.IdentScreenName)) *mockBuddyService_BroadcastBuddyDeparted_Call {
  192. _c.Call.Run(func(args mock.Arguments) {
  193. var arg0 context.Context
  194. if args[0] != nil {
  195. arg0 = args[0].(context.Context)
  196. }
  197. var arg1 state.IdentScreenName
  198. if args[1] != nil {
  199. arg1 = args[1].(state.IdentScreenName)
  200. }
  201. run(
  202. arg0,
  203. arg1,
  204. )
  205. })
  206. return _c
  207. }
  208. func (_c *mockBuddyService_BroadcastBuddyDeparted_Call) Return(err error) *mockBuddyService_BroadcastBuddyDeparted_Call {
  209. _c.Call.Return(err)
  210. return _c
  211. }
  212. func (_c *mockBuddyService_BroadcastBuddyDeparted_Call) RunAndReturn(run func(ctx context.Context, screenName state.IdentScreenName) error) *mockBuddyService_BroadcastBuddyDeparted_Call {
  213. _c.Call.Return(run)
  214. return _c
  215. }
  216. // DelBuddies provides a mock function for the type mockBuddyService
  217. func (_mock *mockBuddyService) DelBuddies(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x03_0x05_BuddyDelBuddies) error {
  218. ret := _mock.Called(ctx, instance, inBody)
  219. if len(ret) == 0 {
  220. panic("no return value specified for DelBuddies")
  221. }
  222. var r0 error
  223. if returnFunc, ok := ret.Get(0).(func(context.Context, *state.SessionInstance, wire.SNAC_0x03_0x05_BuddyDelBuddies) error); ok {
  224. r0 = returnFunc(ctx, instance, inBody)
  225. } else {
  226. r0 = ret.Error(0)
  227. }
  228. return r0
  229. }
  230. // mockBuddyService_DelBuddies_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DelBuddies'
  231. type mockBuddyService_DelBuddies_Call struct {
  232. *mock.Call
  233. }
  234. // DelBuddies is a helper method to define mock.On call
  235. // - ctx context.Context
  236. // - instance *state.SessionInstance
  237. // - inBody wire.SNAC_0x03_0x05_BuddyDelBuddies
  238. func (_e *mockBuddyService_Expecter) DelBuddies(ctx any, instance any, inBody any) *mockBuddyService_DelBuddies_Call {
  239. return &mockBuddyService_DelBuddies_Call{Call: _e.mock.On("DelBuddies", ctx, instance, inBody)}
  240. }
  241. func (_c *mockBuddyService_DelBuddies_Call) Run(run func(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x03_0x05_BuddyDelBuddies)) *mockBuddyService_DelBuddies_Call {
  242. _c.Call.Run(func(args mock.Arguments) {
  243. var arg0 context.Context
  244. if args[0] != nil {
  245. arg0 = args[0].(context.Context)
  246. }
  247. var arg1 *state.SessionInstance
  248. if args[1] != nil {
  249. arg1 = args[1].(*state.SessionInstance)
  250. }
  251. var arg2 wire.SNAC_0x03_0x05_BuddyDelBuddies
  252. if args[2] != nil {
  253. arg2 = args[2].(wire.SNAC_0x03_0x05_BuddyDelBuddies)
  254. }
  255. run(
  256. arg0,
  257. arg1,
  258. arg2,
  259. )
  260. })
  261. return _c
  262. }
  263. func (_c *mockBuddyService_DelBuddies_Call) Return(err error) *mockBuddyService_DelBuddies_Call {
  264. _c.Call.Return(err)
  265. return _c
  266. }
  267. func (_c *mockBuddyService_DelBuddies_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance, inBody wire.SNAC_0x03_0x05_BuddyDelBuddies) error) *mockBuddyService_DelBuddies_Call {
  268. _c.Call.Return(run)
  269. return _c
  270. }
  271. // RightsQuery provides a mock function for the type mockBuddyService
  272. func (_mock *mockBuddyService) RightsQuery(ctx context.Context, inFrame wire.SNACFrame) wire.SNACMessage {
  273. ret := _mock.Called(ctx, inFrame)
  274. if len(ret) == 0 {
  275. panic("no return value specified for RightsQuery")
  276. }
  277. var r0 wire.SNACMessage
  278. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNACFrame) wire.SNACMessage); ok {
  279. r0 = returnFunc(ctx, inFrame)
  280. } else {
  281. r0 = ret.Get(0).(wire.SNACMessage)
  282. }
  283. return r0
  284. }
  285. // mockBuddyService_RightsQuery_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RightsQuery'
  286. type mockBuddyService_RightsQuery_Call struct {
  287. *mock.Call
  288. }
  289. // RightsQuery is a helper method to define mock.On call
  290. // - ctx context.Context
  291. // - inFrame wire.SNACFrame
  292. func (_e *mockBuddyService_Expecter) RightsQuery(ctx any, inFrame any) *mockBuddyService_RightsQuery_Call {
  293. return &mockBuddyService_RightsQuery_Call{Call: _e.mock.On("RightsQuery", ctx, inFrame)}
  294. }
  295. func (_c *mockBuddyService_RightsQuery_Call) Run(run func(ctx context.Context, inFrame wire.SNACFrame)) *mockBuddyService_RightsQuery_Call {
  296. _c.Call.Run(func(args mock.Arguments) {
  297. var arg0 context.Context
  298. if args[0] != nil {
  299. arg0 = args[0].(context.Context)
  300. }
  301. var arg1 wire.SNACFrame
  302. if args[1] != nil {
  303. arg1 = args[1].(wire.SNACFrame)
  304. }
  305. run(
  306. arg0,
  307. arg1,
  308. )
  309. })
  310. return _c
  311. }
  312. func (_c *mockBuddyService_RightsQuery_Call) Return(sNACMessage wire.SNACMessage) *mockBuddyService_RightsQuery_Call {
  313. _c.Call.Return(sNACMessage)
  314. return _c
  315. }
  316. func (_c *mockBuddyService_RightsQuery_Call) RunAndReturn(run func(ctx context.Context, inFrame wire.SNACFrame) wire.SNACMessage) *mockBuddyService_RightsQuery_Call {
  317. _c.Call.Return(run)
  318. return _c
  319. }