mock_buddy_service_test.go 11 KB

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