MockSessionManager.go 14 KB

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