session_manager_mock.go 12 KB

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