mock_auth_test.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package oscar
  5. import (
  6. "context"
  7. "github.com/google/uuid"
  8. "github.com/mk6i/open-oscar-server/state"
  9. "github.com/mk6i/open-oscar-server/wire"
  10. mock "github.com/stretchr/testify/mock"
  11. )
  12. // newMockAuthService creates a new instance of mockAuthService. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  13. // The first argument is typically a *testing.T value.
  14. func newMockAuthService(t interface {
  15. mock.TestingT
  16. Cleanup(func())
  17. }) *mockAuthService {
  18. mock := &mockAuthService{}
  19. mock.Mock.Test(t)
  20. t.Cleanup(func() { mock.AssertExpectations(t) })
  21. return mock
  22. }
  23. // mockAuthService is an autogenerated mock type for the AuthService type
  24. type mockAuthService struct {
  25. mock.Mock
  26. }
  27. type mockAuthService_Expecter struct {
  28. mock *mock.Mock
  29. }
  30. func (_m *mockAuthService) EXPECT() *mockAuthService_Expecter {
  31. return &mockAuthService_Expecter{mock: &_m.Mock}
  32. }
  33. // BUCPChallenge provides a mock function for the type mockAuthService
  34. func (_mock *mockAuthService) BUCPChallenge(ctx context.Context, inBody wire.SNAC_0x17_0x06_BUCPChallengeRequest, newUUID func() uuid.UUID) (wire.SNACMessage, error) {
  35. ret := _mock.Called(ctx, inBody, newUUID)
  36. if len(ret) == 0 {
  37. panic("no return value specified for BUCPChallenge")
  38. }
  39. var r0 wire.SNACMessage
  40. var r1 error
  41. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x17_0x06_BUCPChallengeRequest, func() uuid.UUID) (wire.SNACMessage, error)); ok {
  42. return returnFunc(ctx, inBody, newUUID)
  43. }
  44. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x17_0x06_BUCPChallengeRequest, func() uuid.UUID) wire.SNACMessage); ok {
  45. r0 = returnFunc(ctx, inBody, newUUID)
  46. } else {
  47. r0 = ret.Get(0).(wire.SNACMessage)
  48. }
  49. if returnFunc, ok := ret.Get(1).(func(context.Context, wire.SNAC_0x17_0x06_BUCPChallengeRequest, func() uuid.UUID) error); ok {
  50. r1 = returnFunc(ctx, inBody, newUUID)
  51. } else {
  52. r1 = ret.Error(1)
  53. }
  54. return r0, r1
  55. }
  56. // mockAuthService_BUCPChallenge_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BUCPChallenge'
  57. type mockAuthService_BUCPChallenge_Call struct {
  58. *mock.Call
  59. }
  60. // BUCPChallenge is a helper method to define mock.On call
  61. // - ctx context.Context
  62. // - inBody wire.SNAC_0x17_0x06_BUCPChallengeRequest
  63. // - newUUID func() uuid.UUID
  64. func (_e *mockAuthService_Expecter) BUCPChallenge(ctx interface{}, inBody interface{}, newUUID interface{}) *mockAuthService_BUCPChallenge_Call {
  65. return &mockAuthService_BUCPChallenge_Call{Call: _e.mock.On("BUCPChallenge", ctx, inBody, newUUID)}
  66. }
  67. func (_c *mockAuthService_BUCPChallenge_Call) Run(run func(ctx context.Context, inBody wire.SNAC_0x17_0x06_BUCPChallengeRequest, newUUID func() uuid.UUID)) *mockAuthService_BUCPChallenge_Call {
  68. _c.Call.Run(func(args mock.Arguments) {
  69. var arg0 context.Context
  70. if args[0] != nil {
  71. arg0 = args[0].(context.Context)
  72. }
  73. var arg1 wire.SNAC_0x17_0x06_BUCPChallengeRequest
  74. if args[1] != nil {
  75. arg1 = args[1].(wire.SNAC_0x17_0x06_BUCPChallengeRequest)
  76. }
  77. var arg2 func() uuid.UUID
  78. if args[2] != nil {
  79. arg2 = args[2].(func() uuid.UUID)
  80. }
  81. run(
  82. arg0,
  83. arg1,
  84. arg2,
  85. )
  86. })
  87. return _c
  88. }
  89. func (_c *mockAuthService_BUCPChallenge_Call) Return(sNACMessage wire.SNACMessage, err error) *mockAuthService_BUCPChallenge_Call {
  90. _c.Call.Return(sNACMessage, err)
  91. return _c
  92. }
  93. func (_c *mockAuthService_BUCPChallenge_Call) RunAndReturn(run func(ctx context.Context, inBody wire.SNAC_0x17_0x06_BUCPChallengeRequest, newUUID func() uuid.UUID) (wire.SNACMessage, error)) *mockAuthService_BUCPChallenge_Call {
  94. _c.Call.Return(run)
  95. return _c
  96. }
  97. // BUCPLogin provides a mock function for the type mockAuthService
  98. func (_mock *mockAuthService) BUCPLogin(ctx context.Context, inBody wire.SNAC_0x17_0x02_BUCPLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.SNACMessage, error) {
  99. ret := _mock.Called(ctx, inBody, newUserFn, advertisedHost)
  100. if len(ret) == 0 {
  101. panic("no return value specified for BUCPLogin")
  102. }
  103. var r0 wire.SNACMessage
  104. var r1 error
  105. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x17_0x02_BUCPLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) (wire.SNACMessage, error)); ok {
  106. return returnFunc(ctx, inBody, newUserFn, advertisedHost)
  107. }
  108. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x17_0x02_BUCPLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) wire.SNACMessage); ok {
  109. r0 = returnFunc(ctx, inBody, newUserFn, advertisedHost)
  110. } else {
  111. r0 = ret.Get(0).(wire.SNACMessage)
  112. }
  113. if returnFunc, ok := ret.Get(1).(func(context.Context, wire.SNAC_0x17_0x02_BUCPLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) error); ok {
  114. r1 = returnFunc(ctx, inBody, newUserFn, advertisedHost)
  115. } else {
  116. r1 = ret.Error(1)
  117. }
  118. return r0, r1
  119. }
  120. // mockAuthService_BUCPLogin_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BUCPLogin'
  121. type mockAuthService_BUCPLogin_Call struct {
  122. *mock.Call
  123. }
  124. // BUCPLogin is a helper method to define mock.On call
  125. // - ctx context.Context
  126. // - inBody wire.SNAC_0x17_0x02_BUCPLoginRequest
  127. // - newUserFn func(screenName state.DisplayScreenName) (state.User, error)
  128. // - advertisedHost string
  129. func (_e *mockAuthService_Expecter) BUCPLogin(ctx interface{}, inBody interface{}, newUserFn interface{}, advertisedHost interface{}) *mockAuthService_BUCPLogin_Call {
  130. return &mockAuthService_BUCPLogin_Call{Call: _e.mock.On("BUCPLogin", ctx, inBody, newUserFn, advertisedHost)}
  131. }
  132. func (_c *mockAuthService_BUCPLogin_Call) Run(run func(ctx context.Context, inBody wire.SNAC_0x17_0x02_BUCPLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string)) *mockAuthService_BUCPLogin_Call {
  133. _c.Call.Run(func(args mock.Arguments) {
  134. var arg0 context.Context
  135. if args[0] != nil {
  136. arg0 = args[0].(context.Context)
  137. }
  138. var arg1 wire.SNAC_0x17_0x02_BUCPLoginRequest
  139. if args[1] != nil {
  140. arg1 = args[1].(wire.SNAC_0x17_0x02_BUCPLoginRequest)
  141. }
  142. var arg2 func(screenName state.DisplayScreenName) (state.User, error)
  143. if args[2] != nil {
  144. arg2 = args[2].(func(screenName state.DisplayScreenName) (state.User, error))
  145. }
  146. var arg3 string
  147. if args[3] != nil {
  148. arg3 = args[3].(string)
  149. }
  150. run(
  151. arg0,
  152. arg1,
  153. arg2,
  154. arg3,
  155. )
  156. })
  157. return _c
  158. }
  159. func (_c *mockAuthService_BUCPLogin_Call) Return(sNACMessage wire.SNACMessage, err error) *mockAuthService_BUCPLogin_Call {
  160. _c.Call.Return(sNACMessage, err)
  161. return _c
  162. }
  163. func (_c *mockAuthService_BUCPLogin_Call) RunAndReturn(run func(ctx context.Context, inBody wire.SNAC_0x17_0x02_BUCPLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.SNACMessage, error)) *mockAuthService_BUCPLogin_Call {
  164. _c.Call.Return(run)
  165. return _c
  166. }
  167. // CrackCookie provides a mock function for the type mockAuthService
  168. func (_mock *mockAuthService) CrackCookie(authCookie []byte) (state.ServerCookie, error) {
  169. ret := _mock.Called(authCookie)
  170. if len(ret) == 0 {
  171. panic("no return value specified for CrackCookie")
  172. }
  173. var r0 state.ServerCookie
  174. var r1 error
  175. if returnFunc, ok := ret.Get(0).(func([]byte) (state.ServerCookie, error)); ok {
  176. return returnFunc(authCookie)
  177. }
  178. if returnFunc, ok := ret.Get(0).(func([]byte) state.ServerCookie); ok {
  179. r0 = returnFunc(authCookie)
  180. } else {
  181. r0 = ret.Get(0).(state.ServerCookie)
  182. }
  183. if returnFunc, ok := ret.Get(1).(func([]byte) error); ok {
  184. r1 = returnFunc(authCookie)
  185. } else {
  186. r1 = ret.Error(1)
  187. }
  188. return r0, r1
  189. }
  190. // mockAuthService_CrackCookie_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CrackCookie'
  191. type mockAuthService_CrackCookie_Call struct {
  192. *mock.Call
  193. }
  194. // CrackCookie is a helper method to define mock.On call
  195. // - authCookie []byte
  196. func (_e *mockAuthService_Expecter) CrackCookie(authCookie interface{}) *mockAuthService_CrackCookie_Call {
  197. return &mockAuthService_CrackCookie_Call{Call: _e.mock.On("CrackCookie", authCookie)}
  198. }
  199. func (_c *mockAuthService_CrackCookie_Call) Run(run func(authCookie []byte)) *mockAuthService_CrackCookie_Call {
  200. _c.Call.Run(func(args mock.Arguments) {
  201. var arg0 []byte
  202. if args[0] != nil {
  203. arg0 = args[0].([]byte)
  204. }
  205. run(
  206. arg0,
  207. )
  208. })
  209. return _c
  210. }
  211. func (_c *mockAuthService_CrackCookie_Call) Return(serverCookie state.ServerCookie, err error) *mockAuthService_CrackCookie_Call {
  212. _c.Call.Return(serverCookie, err)
  213. return _c
  214. }
  215. func (_c *mockAuthService_CrackCookie_Call) RunAndReturn(run func(authCookie []byte) (state.ServerCookie, error)) *mockAuthService_CrackCookie_Call {
  216. _c.Call.Return(run)
  217. return _c
  218. }
  219. // FLAPLogin provides a mock function for the type mockAuthService
  220. func (_mock *mockAuthService) FLAPLogin(ctx context.Context, inFrame wire.FLAPSignonFrame, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.TLVRestBlock, error) {
  221. ret := _mock.Called(ctx, inFrame, newUserFn, advertisedHost)
  222. if len(ret) == 0 {
  223. panic("no return value specified for FLAPLogin")
  224. }
  225. var r0 wire.TLVRestBlock
  226. var r1 error
  227. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.FLAPSignonFrame, func(screenName state.DisplayScreenName) (state.User, error), string) (wire.TLVRestBlock, error)); ok {
  228. return returnFunc(ctx, inFrame, newUserFn, advertisedHost)
  229. }
  230. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.FLAPSignonFrame, func(screenName state.DisplayScreenName) (state.User, error), string) wire.TLVRestBlock); ok {
  231. r0 = returnFunc(ctx, inFrame, newUserFn, advertisedHost)
  232. } else {
  233. r0 = ret.Get(0).(wire.TLVRestBlock)
  234. }
  235. if returnFunc, ok := ret.Get(1).(func(context.Context, wire.FLAPSignonFrame, func(screenName state.DisplayScreenName) (state.User, error), string) error); ok {
  236. r1 = returnFunc(ctx, inFrame, newUserFn, advertisedHost)
  237. } else {
  238. r1 = ret.Error(1)
  239. }
  240. return r0, r1
  241. }
  242. // mockAuthService_FLAPLogin_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FLAPLogin'
  243. type mockAuthService_FLAPLogin_Call struct {
  244. *mock.Call
  245. }
  246. // FLAPLogin is a helper method to define mock.On call
  247. // - ctx context.Context
  248. // - inFrame wire.FLAPSignonFrame
  249. // - newUserFn func(screenName state.DisplayScreenName) (state.User, error)
  250. // - advertisedHost string
  251. func (_e *mockAuthService_Expecter) FLAPLogin(ctx interface{}, inFrame interface{}, newUserFn interface{}, advertisedHost interface{}) *mockAuthService_FLAPLogin_Call {
  252. return &mockAuthService_FLAPLogin_Call{Call: _e.mock.On("FLAPLogin", ctx, inFrame, newUserFn, advertisedHost)}
  253. }
  254. func (_c *mockAuthService_FLAPLogin_Call) Run(run func(ctx context.Context, inFrame wire.FLAPSignonFrame, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string)) *mockAuthService_FLAPLogin_Call {
  255. _c.Call.Run(func(args mock.Arguments) {
  256. var arg0 context.Context
  257. if args[0] != nil {
  258. arg0 = args[0].(context.Context)
  259. }
  260. var arg1 wire.FLAPSignonFrame
  261. if args[1] != nil {
  262. arg1 = args[1].(wire.FLAPSignonFrame)
  263. }
  264. var arg2 func(screenName state.DisplayScreenName) (state.User, error)
  265. if args[2] != nil {
  266. arg2 = args[2].(func(screenName state.DisplayScreenName) (state.User, error))
  267. }
  268. var arg3 string
  269. if args[3] != nil {
  270. arg3 = args[3].(string)
  271. }
  272. run(
  273. arg0,
  274. arg1,
  275. arg2,
  276. arg3,
  277. )
  278. })
  279. return _c
  280. }
  281. func (_c *mockAuthService_FLAPLogin_Call) Return(tLVRestBlock wire.TLVRestBlock, err error) *mockAuthService_FLAPLogin_Call {
  282. _c.Call.Return(tLVRestBlock, err)
  283. return _c
  284. }
  285. func (_c *mockAuthService_FLAPLogin_Call) RunAndReturn(run func(ctx context.Context, inFrame wire.FLAPSignonFrame, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.TLVRestBlock, error)) *mockAuthService_FLAPLogin_Call {
  286. _c.Call.Return(run)
  287. return _c
  288. }
  289. // KerberosLogin provides a mock function for the type mockAuthService
  290. func (_mock *mockAuthService) KerberosLogin(ctx context.Context, inBody wire.SNAC_0x050C_0x0002_KerberosLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.SNACMessage, error) {
  291. ret := _mock.Called(ctx, inBody, newUserFn, advertisedHost)
  292. if len(ret) == 0 {
  293. panic("no return value specified for KerberosLogin")
  294. }
  295. var r0 wire.SNACMessage
  296. var r1 error
  297. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x050C_0x0002_KerberosLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) (wire.SNACMessage, error)); ok {
  298. return returnFunc(ctx, inBody, newUserFn, advertisedHost)
  299. }
  300. if returnFunc, ok := ret.Get(0).(func(context.Context, wire.SNAC_0x050C_0x0002_KerberosLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) wire.SNACMessage); ok {
  301. r0 = returnFunc(ctx, inBody, newUserFn, advertisedHost)
  302. } else {
  303. r0 = ret.Get(0).(wire.SNACMessage)
  304. }
  305. if returnFunc, ok := ret.Get(1).(func(context.Context, wire.SNAC_0x050C_0x0002_KerberosLoginRequest, func(screenName state.DisplayScreenName) (state.User, error), string) error); ok {
  306. r1 = returnFunc(ctx, inBody, newUserFn, advertisedHost)
  307. } else {
  308. r1 = ret.Error(1)
  309. }
  310. return r0, r1
  311. }
  312. // mockAuthService_KerberosLogin_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'KerberosLogin'
  313. type mockAuthService_KerberosLogin_Call struct {
  314. *mock.Call
  315. }
  316. // KerberosLogin is a helper method to define mock.On call
  317. // - ctx context.Context
  318. // - inBody wire.SNAC_0x050C_0x0002_KerberosLoginRequest
  319. // - newUserFn func(screenName state.DisplayScreenName) (state.User, error)
  320. // - advertisedHost string
  321. func (_e *mockAuthService_Expecter) KerberosLogin(ctx interface{}, inBody interface{}, newUserFn interface{}, advertisedHost interface{}) *mockAuthService_KerberosLogin_Call {
  322. return &mockAuthService_KerberosLogin_Call{Call: _e.mock.On("KerberosLogin", ctx, inBody, newUserFn, advertisedHost)}
  323. }
  324. func (_c *mockAuthService_KerberosLogin_Call) Run(run func(ctx context.Context, inBody wire.SNAC_0x050C_0x0002_KerberosLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string)) *mockAuthService_KerberosLogin_Call {
  325. _c.Call.Run(func(args mock.Arguments) {
  326. var arg0 context.Context
  327. if args[0] != nil {
  328. arg0 = args[0].(context.Context)
  329. }
  330. var arg1 wire.SNAC_0x050C_0x0002_KerberosLoginRequest
  331. if args[1] != nil {
  332. arg1 = args[1].(wire.SNAC_0x050C_0x0002_KerberosLoginRequest)
  333. }
  334. var arg2 func(screenName state.DisplayScreenName) (state.User, error)
  335. if args[2] != nil {
  336. arg2 = args[2].(func(screenName state.DisplayScreenName) (state.User, error))
  337. }
  338. var arg3 string
  339. if args[3] != nil {
  340. arg3 = args[3].(string)
  341. }
  342. run(
  343. arg0,
  344. arg1,
  345. arg2,
  346. arg3,
  347. )
  348. })
  349. return _c
  350. }
  351. func (_c *mockAuthService_KerberosLogin_Call) Return(sNACMessage wire.SNACMessage, err error) *mockAuthService_KerberosLogin_Call {
  352. _c.Call.Return(sNACMessage, err)
  353. return _c
  354. }
  355. func (_c *mockAuthService_KerberosLogin_Call) RunAndReturn(run func(ctx context.Context, inBody wire.SNAC_0x050C_0x0002_KerberosLoginRequest, newUserFn func(screenName state.DisplayScreenName) (state.User, error), advertisedHost string) (wire.SNACMessage, error)) *mockAuthService_KerberosLogin_Call {
  356. _c.Call.Return(run)
  357. return _c
  358. }
  359. // RegisterBOSSession provides a mock function for the type mockAuthService
  360. func (_mock *mockAuthService) RegisterBOSSession(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error) {
  361. ret := _mock.Called(ctx, authCookie)
  362. if len(ret) == 0 {
  363. panic("no return value specified for RegisterBOSSession")
  364. }
  365. var r0 *state.SessionInstance
  366. var r1 error
  367. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) (*state.SessionInstance, error)); ok {
  368. return returnFunc(ctx, authCookie)
  369. }
  370. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) *state.SessionInstance); ok {
  371. r0 = returnFunc(ctx, authCookie)
  372. } else {
  373. if ret.Get(0) != nil {
  374. r0 = ret.Get(0).(*state.SessionInstance)
  375. }
  376. }
  377. if returnFunc, ok := ret.Get(1).(func(context.Context, state.ServerCookie) error); ok {
  378. r1 = returnFunc(ctx, authCookie)
  379. } else {
  380. r1 = ret.Error(1)
  381. }
  382. return r0, r1
  383. }
  384. // mockAuthService_RegisterBOSSession_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RegisterBOSSession'
  385. type mockAuthService_RegisterBOSSession_Call struct {
  386. *mock.Call
  387. }
  388. // RegisterBOSSession is a helper method to define mock.On call
  389. // - ctx context.Context
  390. // - authCookie state.ServerCookie
  391. func (_e *mockAuthService_Expecter) RegisterBOSSession(ctx interface{}, authCookie interface{}) *mockAuthService_RegisterBOSSession_Call {
  392. return &mockAuthService_RegisterBOSSession_Call{Call: _e.mock.On("RegisterBOSSession", ctx, authCookie)}
  393. }
  394. func (_c *mockAuthService_RegisterBOSSession_Call) Run(run func(ctx context.Context, authCookie state.ServerCookie)) *mockAuthService_RegisterBOSSession_Call {
  395. _c.Call.Run(func(args mock.Arguments) {
  396. var arg0 context.Context
  397. if args[0] != nil {
  398. arg0 = args[0].(context.Context)
  399. }
  400. var arg1 state.ServerCookie
  401. if args[1] != nil {
  402. arg1 = args[1].(state.ServerCookie)
  403. }
  404. run(
  405. arg0,
  406. arg1,
  407. )
  408. })
  409. return _c
  410. }
  411. func (_c *mockAuthService_RegisterBOSSession_Call) Return(sessionInstance *state.SessionInstance, err error) *mockAuthService_RegisterBOSSession_Call {
  412. _c.Call.Return(sessionInstance, err)
  413. return _c
  414. }
  415. func (_c *mockAuthService_RegisterBOSSession_Call) RunAndReturn(run func(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error)) *mockAuthService_RegisterBOSSession_Call {
  416. _c.Call.Return(run)
  417. return _c
  418. }
  419. // RegisterChatSession provides a mock function for the type mockAuthService
  420. func (_mock *mockAuthService) RegisterChatSession(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error) {
  421. ret := _mock.Called(ctx, authCookie)
  422. if len(ret) == 0 {
  423. panic("no return value specified for RegisterChatSession")
  424. }
  425. var r0 *state.SessionInstance
  426. var r1 error
  427. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) (*state.SessionInstance, error)); ok {
  428. return returnFunc(ctx, authCookie)
  429. }
  430. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) *state.SessionInstance); ok {
  431. r0 = returnFunc(ctx, authCookie)
  432. } else {
  433. if ret.Get(0) != nil {
  434. r0 = ret.Get(0).(*state.SessionInstance)
  435. }
  436. }
  437. if returnFunc, ok := ret.Get(1).(func(context.Context, state.ServerCookie) error); ok {
  438. r1 = returnFunc(ctx, authCookie)
  439. } else {
  440. r1 = ret.Error(1)
  441. }
  442. return r0, r1
  443. }
  444. // mockAuthService_RegisterChatSession_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RegisterChatSession'
  445. type mockAuthService_RegisterChatSession_Call struct {
  446. *mock.Call
  447. }
  448. // RegisterChatSession is a helper method to define mock.On call
  449. // - ctx context.Context
  450. // - authCookie state.ServerCookie
  451. func (_e *mockAuthService_Expecter) RegisterChatSession(ctx interface{}, authCookie interface{}) *mockAuthService_RegisterChatSession_Call {
  452. return &mockAuthService_RegisterChatSession_Call{Call: _e.mock.On("RegisterChatSession", ctx, authCookie)}
  453. }
  454. func (_c *mockAuthService_RegisterChatSession_Call) Run(run func(ctx context.Context, authCookie state.ServerCookie)) *mockAuthService_RegisterChatSession_Call {
  455. _c.Call.Run(func(args mock.Arguments) {
  456. var arg0 context.Context
  457. if args[0] != nil {
  458. arg0 = args[0].(context.Context)
  459. }
  460. var arg1 state.ServerCookie
  461. if args[1] != nil {
  462. arg1 = args[1].(state.ServerCookie)
  463. }
  464. run(
  465. arg0,
  466. arg1,
  467. )
  468. })
  469. return _c
  470. }
  471. func (_c *mockAuthService_RegisterChatSession_Call) Return(sessionInstance *state.SessionInstance, err error) *mockAuthService_RegisterChatSession_Call {
  472. _c.Call.Return(sessionInstance, err)
  473. return _c
  474. }
  475. func (_c *mockAuthService_RegisterChatSession_Call) RunAndReturn(run func(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error)) *mockAuthService_RegisterChatSession_Call {
  476. _c.Call.Return(run)
  477. return _c
  478. }
  479. // RetrieveBOSSession provides a mock function for the type mockAuthService
  480. func (_mock *mockAuthService) RetrieveBOSSession(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error) {
  481. ret := _mock.Called(ctx, authCookie)
  482. if len(ret) == 0 {
  483. panic("no return value specified for RetrieveBOSSession")
  484. }
  485. var r0 *state.SessionInstance
  486. var r1 error
  487. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) (*state.SessionInstance, error)); ok {
  488. return returnFunc(ctx, authCookie)
  489. }
  490. if returnFunc, ok := ret.Get(0).(func(context.Context, state.ServerCookie) *state.SessionInstance); ok {
  491. r0 = returnFunc(ctx, authCookie)
  492. } else {
  493. if ret.Get(0) != nil {
  494. r0 = ret.Get(0).(*state.SessionInstance)
  495. }
  496. }
  497. if returnFunc, ok := ret.Get(1).(func(context.Context, state.ServerCookie) error); ok {
  498. r1 = returnFunc(ctx, authCookie)
  499. } else {
  500. r1 = ret.Error(1)
  501. }
  502. return r0, r1
  503. }
  504. // mockAuthService_RetrieveBOSSession_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RetrieveBOSSession'
  505. type mockAuthService_RetrieveBOSSession_Call struct {
  506. *mock.Call
  507. }
  508. // RetrieveBOSSession is a helper method to define mock.On call
  509. // - ctx context.Context
  510. // - authCookie state.ServerCookie
  511. func (_e *mockAuthService_Expecter) RetrieveBOSSession(ctx interface{}, authCookie interface{}) *mockAuthService_RetrieveBOSSession_Call {
  512. return &mockAuthService_RetrieveBOSSession_Call{Call: _e.mock.On("RetrieveBOSSession", ctx, authCookie)}
  513. }
  514. func (_c *mockAuthService_RetrieveBOSSession_Call) Run(run func(ctx context.Context, authCookie state.ServerCookie)) *mockAuthService_RetrieveBOSSession_Call {
  515. _c.Call.Run(func(args mock.Arguments) {
  516. var arg0 context.Context
  517. if args[0] != nil {
  518. arg0 = args[0].(context.Context)
  519. }
  520. var arg1 state.ServerCookie
  521. if args[1] != nil {
  522. arg1 = args[1].(state.ServerCookie)
  523. }
  524. run(
  525. arg0,
  526. arg1,
  527. )
  528. })
  529. return _c
  530. }
  531. func (_c *mockAuthService_RetrieveBOSSession_Call) Return(sessionInstance *state.SessionInstance, err error) *mockAuthService_RetrieveBOSSession_Call {
  532. _c.Call.Return(sessionInstance, err)
  533. return _c
  534. }
  535. func (_c *mockAuthService_RetrieveBOSSession_Call) RunAndReturn(run func(ctx context.Context, authCookie state.ServerCookie) (*state.SessionInstance, error)) *mockAuthService_RetrieveBOSSession_Call {
  536. _c.Call.Return(run)
  537. return _c
  538. }
  539. // Signout provides a mock function for the type mockAuthService
  540. func (_mock *mockAuthService) Signout(ctx context.Context, instance *state.SessionInstance) {
  541. _mock.Called(ctx, instance)
  542. return
  543. }
  544. // mockAuthService_Signout_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Signout'
  545. type mockAuthService_Signout_Call struct {
  546. *mock.Call
  547. }
  548. // Signout is a helper method to define mock.On call
  549. // - ctx context.Context
  550. // - instance *state.SessionInstance
  551. func (_e *mockAuthService_Expecter) Signout(ctx interface{}, instance interface{}) *mockAuthService_Signout_Call {
  552. return &mockAuthService_Signout_Call{Call: _e.mock.On("Signout", ctx, instance)}
  553. }
  554. func (_c *mockAuthService_Signout_Call) Run(run func(ctx context.Context, instance *state.SessionInstance)) *mockAuthService_Signout_Call {
  555. _c.Call.Run(func(args mock.Arguments) {
  556. var arg0 context.Context
  557. if args[0] != nil {
  558. arg0 = args[0].(context.Context)
  559. }
  560. var arg1 *state.SessionInstance
  561. if args[1] != nil {
  562. arg1 = args[1].(*state.SessionInstance)
  563. }
  564. run(
  565. arg0,
  566. arg1,
  567. )
  568. })
  569. return _c
  570. }
  571. func (_c *mockAuthService_Signout_Call) Return() *mockAuthService_Signout_Call {
  572. _c.Call.Return()
  573. return _c
  574. }
  575. func (_c *mockAuthService_Signout_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance)) *mockAuthService_Signout_Call {
  576. _c.Run(run)
  577. return _c
  578. }
  579. // SignoutChat provides a mock function for the type mockAuthService
  580. func (_mock *mockAuthService) SignoutChat(ctx context.Context, instance *state.SessionInstance) {
  581. _mock.Called(ctx, instance)
  582. return
  583. }
  584. // mockAuthService_SignoutChat_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SignoutChat'
  585. type mockAuthService_SignoutChat_Call struct {
  586. *mock.Call
  587. }
  588. // SignoutChat is a helper method to define mock.On call
  589. // - ctx context.Context
  590. // - instance *state.SessionInstance
  591. func (_e *mockAuthService_Expecter) SignoutChat(ctx interface{}, instance interface{}) *mockAuthService_SignoutChat_Call {
  592. return &mockAuthService_SignoutChat_Call{Call: _e.mock.On("SignoutChat", ctx, instance)}
  593. }
  594. func (_c *mockAuthService_SignoutChat_Call) Run(run func(ctx context.Context, instance *state.SessionInstance)) *mockAuthService_SignoutChat_Call {
  595. _c.Call.Run(func(args mock.Arguments) {
  596. var arg0 context.Context
  597. if args[0] != nil {
  598. arg0 = args[0].(context.Context)
  599. }
  600. var arg1 *state.SessionInstance
  601. if args[1] != nil {
  602. arg1 = args[1].(*state.SessionInstance)
  603. }
  604. run(
  605. arg0,
  606. arg1,
  607. )
  608. })
  609. return _c
  610. }
  611. func (_c *mockAuthService_SignoutChat_Call) Return() *mockAuthService_SignoutChat_Call {
  612. _c.Call.Return()
  613. return _c
  614. }
  615. func (_c *mockAuthService_SignoutChat_Call) RunAndReturn(run func(ctx context.Context, instance *state.SessionInstance)) *mockAuthService_SignoutChat_Call {
  616. _c.Run(run)
  617. return _c
  618. }