session_message_relayer.go 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. // Code generated by mockery v2.35.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. // mockMessageRelayer is an autogenerated mock type for the MessageRelayer type
  10. type mockMessageRelayer struct {
  11. mock.Mock
  12. }
  13. type mockMessageRelayer_Expecter struct {
  14. mock *mock.Mock
  15. }
  16. func (_m *mockMessageRelayer) EXPECT() *mockMessageRelayer_Expecter {
  17. return &mockMessageRelayer_Expecter{mock: &_m.Mock}
  18. }
  19. // BroadcastToScreenNames provides a mock function with given fields: ctx, screenNames, msg
  20. func (_m *mockMessageRelayer) BroadcastToScreenNames(ctx context.Context, screenNames []string, msg oscar.SNACMessage) {
  21. _m.Called(ctx, screenNames, msg)
  22. }
  23. // mockMessageRelayer_BroadcastToScreenNames_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BroadcastToScreenNames'
  24. type mockMessageRelayer_BroadcastToScreenNames_Call struct {
  25. *mock.Call
  26. }
  27. // BroadcastToScreenNames is a helper method to define mock.On call
  28. // - ctx context.Context
  29. // - screenNames []string
  30. // - msg oscar.SNACMessage
  31. func (_e *mockMessageRelayer_Expecter) BroadcastToScreenNames(ctx interface{}, screenNames interface{}, msg interface{}) *mockMessageRelayer_BroadcastToScreenNames_Call {
  32. return &mockMessageRelayer_BroadcastToScreenNames_Call{Call: _e.mock.On("BroadcastToScreenNames", ctx, screenNames, msg)}
  33. }
  34. func (_c *mockMessageRelayer_BroadcastToScreenNames_Call) Run(run func(ctx context.Context, screenNames []string, msg oscar.SNACMessage)) *mockMessageRelayer_BroadcastToScreenNames_Call {
  35. _c.Call.Run(func(args mock.Arguments) {
  36. run(args[0].(context.Context), args[1].([]string), args[2].(oscar.SNACMessage))
  37. })
  38. return _c
  39. }
  40. func (_c *mockMessageRelayer_BroadcastToScreenNames_Call) Return() *mockMessageRelayer_BroadcastToScreenNames_Call {
  41. _c.Call.Return()
  42. return _c
  43. }
  44. func (_c *mockMessageRelayer_BroadcastToScreenNames_Call) RunAndReturn(run func(context.Context, []string, oscar.SNACMessage)) *mockMessageRelayer_BroadcastToScreenNames_Call {
  45. _c.Call.Return(run)
  46. return _c
  47. }
  48. // RetrieveByScreenName provides a mock function with given fields: screenName
  49. func (_m *mockMessageRelayer) RetrieveByScreenName(screenName string) *state.Session {
  50. ret := _m.Called(screenName)
  51. var r0 *state.Session
  52. if rf, ok := ret.Get(0).(func(string) *state.Session); ok {
  53. r0 = rf(screenName)
  54. } else {
  55. if ret.Get(0) != nil {
  56. r0 = ret.Get(0).(*state.Session)
  57. }
  58. }
  59. return r0
  60. }
  61. // mockMessageRelayer_RetrieveByScreenName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RetrieveByScreenName'
  62. type mockMessageRelayer_RetrieveByScreenName_Call struct {
  63. *mock.Call
  64. }
  65. // RetrieveByScreenName is a helper method to define mock.On call
  66. // - screenName string
  67. func (_e *mockMessageRelayer_Expecter) RetrieveByScreenName(screenName interface{}) *mockMessageRelayer_RetrieveByScreenName_Call {
  68. return &mockMessageRelayer_RetrieveByScreenName_Call{Call: _e.mock.On("RetrieveByScreenName", screenName)}
  69. }
  70. func (_c *mockMessageRelayer_RetrieveByScreenName_Call) Run(run func(screenName string)) *mockMessageRelayer_RetrieveByScreenName_Call {
  71. _c.Call.Run(func(args mock.Arguments) {
  72. run(args[0].(string))
  73. })
  74. return _c
  75. }
  76. func (_c *mockMessageRelayer_RetrieveByScreenName_Call) Return(_a0 *state.Session) *mockMessageRelayer_RetrieveByScreenName_Call {
  77. _c.Call.Return(_a0)
  78. return _c
  79. }
  80. func (_c *mockMessageRelayer_RetrieveByScreenName_Call) RunAndReturn(run func(string) *state.Session) *mockMessageRelayer_RetrieveByScreenName_Call {
  81. _c.Call.Return(run)
  82. return _c
  83. }
  84. // SendToScreenName provides a mock function with given fields: ctx, screenName, msg
  85. func (_m *mockMessageRelayer) SendToScreenName(ctx context.Context, screenName string, msg oscar.SNACMessage) {
  86. _m.Called(ctx, screenName, msg)
  87. }
  88. // mockMessageRelayer_SendToScreenName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendToScreenName'
  89. type mockMessageRelayer_SendToScreenName_Call struct {
  90. *mock.Call
  91. }
  92. // SendToScreenName is a helper method to define mock.On call
  93. // - ctx context.Context
  94. // - screenName string
  95. // - msg oscar.SNACMessage
  96. func (_e *mockMessageRelayer_Expecter) SendToScreenName(ctx interface{}, screenName interface{}, msg interface{}) *mockMessageRelayer_SendToScreenName_Call {
  97. return &mockMessageRelayer_SendToScreenName_Call{Call: _e.mock.On("SendToScreenName", ctx, screenName, msg)}
  98. }
  99. func (_c *mockMessageRelayer_SendToScreenName_Call) Run(run func(ctx context.Context, screenName string, msg oscar.SNACMessage)) *mockMessageRelayer_SendToScreenName_Call {
  100. _c.Call.Run(func(args mock.Arguments) {
  101. run(args[0].(context.Context), args[1].(string), args[2].(oscar.SNACMessage))
  102. })
  103. return _c
  104. }
  105. func (_c *mockMessageRelayer_SendToScreenName_Call) Return() *mockMessageRelayer_SendToScreenName_Call {
  106. _c.Call.Return()
  107. return _c
  108. }
  109. func (_c *mockMessageRelayer_SendToScreenName_Call) RunAndReturn(run func(context.Context, string, oscar.SNACMessage)) *mockMessageRelayer_SendToScreenName_Call {
  110. _c.Call.Return(run)
  111. return _c
  112. }
  113. // newMockMessageRelayer creates a new instance of mockMessageRelayer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  114. // The first argument is typically a *testing.T value.
  115. func newMockMessageRelayer(t interface {
  116. mock.TestingT
  117. Cleanup(func())
  118. }) *mockMessageRelayer {
  119. mock := &mockMessageRelayer{}
  120. mock.Mock.Test(t)
  121. t.Cleanup(func() { mock.AssertExpectations(t) })
  122. return mock
  123. }