session_manager_mock.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445
  1. // Code generated by mockery v2.35.2. DO NOT EDIT.
  2. package server
  3. import (
  4. context "context"
  5. mock "github.com/stretchr/testify/mock"
  6. )
  7. // MockSessionManager is an autogenerated mock type for the SessionManager type
  8. type MockSessionManager struct {
  9. mock.Mock
  10. }
  11. type MockSessionManager_Expecter struct {
  12. mock *mock.Mock
  13. }
  14. func (_m *MockSessionManager) EXPECT() *MockSessionManager_Expecter {
  15. return &MockSessionManager_Expecter{mock: &_m.Mock}
  16. }
  17. // Broadcast provides a mock function with given fields: ctx, msg
  18. func (_m *MockSessionManager) Broadcast(ctx context.Context, msg XMessage) {
  19. _m.Called(ctx, msg)
  20. }
  21. // MockSessionManager_Broadcast_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Broadcast'
  22. type MockSessionManager_Broadcast_Call struct {
  23. *mock.Call
  24. }
  25. // Broadcast is a helper method to define mock.On call
  26. // - ctx context.Context
  27. // - msg XMessage
  28. func (_e *MockSessionManager_Expecter) Broadcast(ctx interface{}, msg interface{}) *MockSessionManager_Broadcast_Call {
  29. return &MockSessionManager_Broadcast_Call{Call: _e.mock.On("Broadcast", ctx, msg)}
  30. }
  31. func (_c *MockSessionManager_Broadcast_Call) Run(run func(ctx context.Context, msg XMessage)) *MockSessionManager_Broadcast_Call {
  32. _c.Call.Run(func(args mock.Arguments) {
  33. run(args[0].(context.Context), args[1].(XMessage))
  34. })
  35. return _c
  36. }
  37. func (_c *MockSessionManager_Broadcast_Call) Return() *MockSessionManager_Broadcast_Call {
  38. _c.Call.Return()
  39. return _c
  40. }
  41. func (_c *MockSessionManager_Broadcast_Call) RunAndReturn(run func(context.Context, XMessage)) *MockSessionManager_Broadcast_Call {
  42. _c.Call.Return(run)
  43. return _c
  44. }
  45. // BroadcastExcept provides a mock function with given fields: ctx, except, msg
  46. func (_m *MockSessionManager) BroadcastExcept(ctx context.Context, except *Session, msg XMessage) {
  47. _m.Called(ctx, except, msg)
  48. }
  49. // MockSessionManager_BroadcastExcept_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BroadcastExcept'
  50. type MockSessionManager_BroadcastExcept_Call struct {
  51. *mock.Call
  52. }
  53. // BroadcastExcept is a helper method to define mock.On call
  54. // - ctx context.Context
  55. // - except *Session
  56. // - msg XMessage
  57. func (_e *MockSessionManager_Expecter) BroadcastExcept(ctx interface{}, except interface{}, msg interface{}) *MockSessionManager_BroadcastExcept_Call {
  58. return &MockSessionManager_BroadcastExcept_Call{Call: _e.mock.On("BroadcastExcept", ctx, except, msg)}
  59. }
  60. func (_c *MockSessionManager_BroadcastExcept_Call) Run(run func(ctx context.Context, except *Session, msg XMessage)) *MockSessionManager_BroadcastExcept_Call {
  61. _c.Call.Run(func(args mock.Arguments) {
  62. run(args[0].(context.Context), args[1].(*Session), args[2].(XMessage))
  63. })
  64. return _c
  65. }
  66. func (_c *MockSessionManager_BroadcastExcept_Call) Return() *MockSessionManager_BroadcastExcept_Call {
  67. _c.Call.Return()
  68. return _c
  69. }
  70. func (_c *MockSessionManager_BroadcastExcept_Call) RunAndReturn(run func(context.Context, *Session, XMessage)) *MockSessionManager_BroadcastExcept_Call {
  71. _c.Call.Return(run)
  72. return _c
  73. }
  74. // BroadcastToScreenNames provides a mock function with given fields: ctx, screenNames, msg
  75. func (_m *MockSessionManager) BroadcastToScreenNames(ctx context.Context, screenNames []string, msg XMessage) {
  76. _m.Called(ctx, screenNames, msg)
  77. }
  78. // MockSessionManager_BroadcastToScreenNames_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BroadcastToScreenNames'
  79. type MockSessionManager_BroadcastToScreenNames_Call struct {
  80. *mock.Call
  81. }
  82. // BroadcastToScreenNames is a helper method to define mock.On call
  83. // - ctx context.Context
  84. // - screenNames []string
  85. // - msg XMessage
  86. func (_e *MockSessionManager_Expecter) BroadcastToScreenNames(ctx interface{}, screenNames interface{}, msg interface{}) *MockSessionManager_BroadcastToScreenNames_Call {
  87. return &MockSessionManager_BroadcastToScreenNames_Call{Call: _e.mock.On("BroadcastToScreenNames", ctx, screenNames, msg)}
  88. }
  89. func (_c *MockSessionManager_BroadcastToScreenNames_Call) Run(run func(ctx context.Context, screenNames []string, msg XMessage)) *MockSessionManager_BroadcastToScreenNames_Call {
  90. _c.Call.Run(func(args mock.Arguments) {
  91. run(args[0].(context.Context), args[1].([]string), args[2].(XMessage))
  92. })
  93. return _c
  94. }
  95. func (_c *MockSessionManager_BroadcastToScreenNames_Call) Return() *MockSessionManager_BroadcastToScreenNames_Call {
  96. _c.Call.Return()
  97. return _c
  98. }
  99. func (_c *MockSessionManager_BroadcastToScreenNames_Call) RunAndReturn(run func(context.Context, []string, XMessage)) *MockSessionManager_BroadcastToScreenNames_Call {
  100. _c.Call.Return(run)
  101. return _c
  102. }
  103. // Empty provides a mock function with given fields:
  104. func (_m *MockSessionManager) Empty() bool {
  105. ret := _m.Called()
  106. var r0 bool
  107. if rf, ok := ret.Get(0).(func() bool); ok {
  108. r0 = rf()
  109. } else {
  110. r0 = ret.Get(0).(bool)
  111. }
  112. return r0
  113. }
  114. // MockSessionManager_Empty_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Empty'
  115. type MockSessionManager_Empty_Call struct {
  116. *mock.Call
  117. }
  118. // Empty is a helper method to define mock.On call
  119. func (_e *MockSessionManager_Expecter) Empty() *MockSessionManager_Empty_Call {
  120. return &MockSessionManager_Empty_Call{Call: _e.mock.On("Empty")}
  121. }
  122. func (_c *MockSessionManager_Empty_Call) Run(run func()) *MockSessionManager_Empty_Call {
  123. _c.Call.Run(func(args mock.Arguments) {
  124. run()
  125. })
  126. return _c
  127. }
  128. func (_c *MockSessionManager_Empty_Call) Return(_a0 bool) *MockSessionManager_Empty_Call {
  129. _c.Call.Return(_a0)
  130. return _c
  131. }
  132. func (_c *MockSessionManager_Empty_Call) RunAndReturn(run func() bool) *MockSessionManager_Empty_Call {
  133. _c.Call.Return(run)
  134. return _c
  135. }
  136. // NewSessionWithSN provides a mock function with given fields: sessID, screenName
  137. func (_m *MockSessionManager) NewSessionWithSN(sessID string, screenName string) *Session {
  138. ret := _m.Called(sessID, screenName)
  139. var r0 *Session
  140. if rf, ok := ret.Get(0).(func(string, string) *Session); ok {
  141. r0 = rf(sessID, screenName)
  142. } else {
  143. if ret.Get(0) != nil {
  144. r0 = ret.Get(0).(*Session)
  145. }
  146. }
  147. return r0
  148. }
  149. // MockSessionManager_NewSessionWithSN_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'NewSessionWithSN'
  150. type MockSessionManager_NewSessionWithSN_Call struct {
  151. *mock.Call
  152. }
  153. // NewSessionWithSN is a helper method to define mock.On call
  154. // - sessID string
  155. // - screenName string
  156. func (_e *MockSessionManager_Expecter) NewSessionWithSN(sessID interface{}, screenName interface{}) *MockSessionManager_NewSessionWithSN_Call {
  157. return &MockSessionManager_NewSessionWithSN_Call{Call: _e.mock.On("NewSessionWithSN", sessID, screenName)}
  158. }
  159. func (_c *MockSessionManager_NewSessionWithSN_Call) Run(run func(sessID string, screenName string)) *MockSessionManager_NewSessionWithSN_Call {
  160. _c.Call.Run(func(args mock.Arguments) {
  161. run(args[0].(string), args[1].(string))
  162. })
  163. return _c
  164. }
  165. func (_c *MockSessionManager_NewSessionWithSN_Call) Return(_a0 *Session) *MockSessionManager_NewSessionWithSN_Call {
  166. _c.Call.Return(_a0)
  167. return _c
  168. }
  169. func (_c *MockSessionManager_NewSessionWithSN_Call) RunAndReturn(run func(string, string) *Session) *MockSessionManager_NewSessionWithSN_Call {
  170. _c.Call.Return(run)
  171. return _c
  172. }
  173. // Participants provides a mock function with given fields:
  174. func (_m *MockSessionManager) Participants() []*Session {
  175. ret := _m.Called()
  176. var r0 []*Session
  177. if rf, ok := ret.Get(0).(func() []*Session); ok {
  178. r0 = rf()
  179. } else {
  180. if ret.Get(0) != nil {
  181. r0 = ret.Get(0).([]*Session)
  182. }
  183. }
  184. return r0
  185. }
  186. // MockSessionManager_Participants_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Participants'
  187. type MockSessionManager_Participants_Call struct {
  188. *mock.Call
  189. }
  190. // Participants is a helper method to define mock.On call
  191. func (_e *MockSessionManager_Expecter) Participants() *MockSessionManager_Participants_Call {
  192. return &MockSessionManager_Participants_Call{Call: _e.mock.On("Participants")}
  193. }
  194. func (_c *MockSessionManager_Participants_Call) Run(run func()) *MockSessionManager_Participants_Call {
  195. _c.Call.Run(func(args mock.Arguments) {
  196. run()
  197. })
  198. return _c
  199. }
  200. func (_c *MockSessionManager_Participants_Call) Return(_a0 []*Session) *MockSessionManager_Participants_Call {
  201. _c.Call.Return(_a0)
  202. return _c
  203. }
  204. func (_c *MockSessionManager_Participants_Call) RunAndReturn(run func() []*Session) *MockSessionManager_Participants_Call {
  205. _c.Call.Return(run)
  206. return _c
  207. }
  208. // Remove provides a mock function with given fields: sess
  209. func (_m *MockSessionManager) Remove(sess *Session) {
  210. _m.Called(sess)
  211. }
  212. // MockSessionManager_Remove_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Remove'
  213. type MockSessionManager_Remove_Call struct {
  214. *mock.Call
  215. }
  216. // Remove is a helper method to define mock.On call
  217. // - sess *Session
  218. func (_e *MockSessionManager_Expecter) Remove(sess interface{}) *MockSessionManager_Remove_Call {
  219. return &MockSessionManager_Remove_Call{Call: _e.mock.On("Remove", sess)}
  220. }
  221. func (_c *MockSessionManager_Remove_Call) Run(run func(sess *Session)) *MockSessionManager_Remove_Call {
  222. _c.Call.Run(func(args mock.Arguments) {
  223. run(args[0].(*Session))
  224. })
  225. return _c
  226. }
  227. func (_c *MockSessionManager_Remove_Call) Return() *MockSessionManager_Remove_Call {
  228. _c.Call.Return()
  229. return _c
  230. }
  231. func (_c *MockSessionManager_Remove_Call) RunAndReturn(run func(*Session)) *MockSessionManager_Remove_Call {
  232. _c.Call.Return(run)
  233. return _c
  234. }
  235. // Retrieve provides a mock function with given fields: ID
  236. func (_m *MockSessionManager) Retrieve(ID string) (*Session, bool) {
  237. ret := _m.Called(ID)
  238. var r0 *Session
  239. var r1 bool
  240. if rf, ok := ret.Get(0).(func(string) (*Session, bool)); ok {
  241. return rf(ID)
  242. }
  243. if rf, ok := ret.Get(0).(func(string) *Session); ok {
  244. r0 = rf(ID)
  245. } else {
  246. if ret.Get(0) != nil {
  247. r0 = ret.Get(0).(*Session)
  248. }
  249. }
  250. if rf, ok := ret.Get(1).(func(string) bool); ok {
  251. r1 = rf(ID)
  252. } else {
  253. r1 = ret.Get(1).(bool)
  254. }
  255. return r0, r1
  256. }
  257. // MockSessionManager_Retrieve_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Retrieve'
  258. type MockSessionManager_Retrieve_Call struct {
  259. *mock.Call
  260. }
  261. // Retrieve is a helper method to define mock.On call
  262. // - ID string
  263. func (_e *MockSessionManager_Expecter) Retrieve(ID interface{}) *MockSessionManager_Retrieve_Call {
  264. return &MockSessionManager_Retrieve_Call{Call: _e.mock.On("Retrieve", ID)}
  265. }
  266. func (_c *MockSessionManager_Retrieve_Call) Run(run func(ID string)) *MockSessionManager_Retrieve_Call {
  267. _c.Call.Run(func(args mock.Arguments) {
  268. run(args[0].(string))
  269. })
  270. return _c
  271. }
  272. func (_c *MockSessionManager_Retrieve_Call) Return(_a0 *Session, _a1 bool) *MockSessionManager_Retrieve_Call {
  273. _c.Call.Return(_a0, _a1)
  274. return _c
  275. }
  276. func (_c *MockSessionManager_Retrieve_Call) RunAndReturn(run func(string) (*Session, bool)) *MockSessionManager_Retrieve_Call {
  277. _c.Call.Return(run)
  278. return _c
  279. }
  280. // RetrieveByScreenName provides a mock function with given fields: screenName
  281. func (_m *MockSessionManager) RetrieveByScreenName(screenName string) (*Session, error) {
  282. ret := _m.Called(screenName)
  283. var r0 *Session
  284. var r1 error
  285. if rf, ok := ret.Get(0).(func(string) (*Session, error)); ok {
  286. return rf(screenName)
  287. }
  288. if rf, ok := ret.Get(0).(func(string) *Session); ok {
  289. r0 = rf(screenName)
  290. } else {
  291. if ret.Get(0) != nil {
  292. r0 = ret.Get(0).(*Session)
  293. }
  294. }
  295. if rf, ok := ret.Get(1).(func(string) error); ok {
  296. r1 = rf(screenName)
  297. } else {
  298. r1 = ret.Error(1)
  299. }
  300. return r0, r1
  301. }
  302. // MockSessionManager_RetrieveByScreenName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RetrieveByScreenName'
  303. type MockSessionManager_RetrieveByScreenName_Call struct {
  304. *mock.Call
  305. }
  306. // RetrieveByScreenName is a helper method to define mock.On call
  307. // - screenName string
  308. func (_e *MockSessionManager_Expecter) RetrieveByScreenName(screenName interface{}) *MockSessionManager_RetrieveByScreenName_Call {
  309. return &MockSessionManager_RetrieveByScreenName_Call{Call: _e.mock.On("RetrieveByScreenName", screenName)}
  310. }
  311. func (_c *MockSessionManager_RetrieveByScreenName_Call) Run(run func(screenName string)) *MockSessionManager_RetrieveByScreenName_Call {
  312. _c.Call.Run(func(args mock.Arguments) {
  313. run(args[0].(string))
  314. })
  315. return _c
  316. }
  317. func (_c *MockSessionManager_RetrieveByScreenName_Call) Return(_a0 *Session, _a1 error) *MockSessionManager_RetrieveByScreenName_Call {
  318. _c.Call.Return(_a0, _a1)
  319. return _c
  320. }
  321. func (_c *MockSessionManager_RetrieveByScreenName_Call) RunAndReturn(run func(string) (*Session, error)) *MockSessionManager_RetrieveByScreenName_Call {
  322. _c.Call.Return(run)
  323. return _c
  324. }
  325. // SendToScreenName provides a mock function with given fields: ctx, screenName, msg
  326. func (_m *MockSessionManager) SendToScreenName(ctx context.Context, screenName string, msg XMessage) {
  327. _m.Called(ctx, screenName, msg)
  328. }
  329. // MockSessionManager_SendToScreenName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendToScreenName'
  330. type MockSessionManager_SendToScreenName_Call struct {
  331. *mock.Call
  332. }
  333. // SendToScreenName is a helper method to define mock.On call
  334. // - ctx context.Context
  335. // - screenName string
  336. // - msg XMessage
  337. func (_e *MockSessionManager_Expecter) SendToScreenName(ctx interface{}, screenName interface{}, msg interface{}) *MockSessionManager_SendToScreenName_Call {
  338. return &MockSessionManager_SendToScreenName_Call{Call: _e.mock.On("SendToScreenName", ctx, screenName, msg)}
  339. }
  340. func (_c *MockSessionManager_SendToScreenName_Call) Run(run func(ctx context.Context, screenName string, msg XMessage)) *MockSessionManager_SendToScreenName_Call {
  341. _c.Call.Run(func(args mock.Arguments) {
  342. run(args[0].(context.Context), args[1].(string), args[2].(XMessage))
  343. })
  344. return _c
  345. }
  346. func (_c *MockSessionManager_SendToScreenName_Call) Return() *MockSessionManager_SendToScreenName_Call {
  347. _c.Call.Return()
  348. return _c
  349. }
  350. func (_c *MockSessionManager_SendToScreenName_Call) RunAndReturn(run func(context.Context, string, XMessage)) *MockSessionManager_SendToScreenName_Call {
  351. _c.Call.Return(run)
  352. return _c
  353. }
  354. // NewMockSessionManager creates a new instance of MockSessionManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  355. // The first argument is typically a *testing.T value.
  356. func NewMockSessionManager(t interface {
  357. mock.TestingT
  358. Cleanup(func())
  359. }) *MockSessionManager {
  360. mock := &MockSessionManager{}
  361. mock.Mock.Test(t)
  362. t.Cleanup(func() { mock.AssertExpectations(t) })
  363. return mock
  364. }