session_manager_mock.go 12 KB

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