mock_packet_sender_test.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package icq_legacy
  5. import (
  6. "net"
  7. mock "github.com/stretchr/testify/mock"
  8. )
  9. // newMockPacketSender creates a new instance of mockPacketSender. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  10. // The first argument is typically a *testing.T value.
  11. func newMockPacketSender(t interface {
  12. mock.TestingT
  13. Cleanup(func())
  14. }) *mockPacketSender {
  15. mock := &mockPacketSender{}
  16. mock.Mock.Test(t)
  17. t.Cleanup(func() { mock.AssertExpectations(t) })
  18. return mock
  19. }
  20. // mockPacketSender is an autogenerated mock type for the PacketSender type
  21. type mockPacketSender struct {
  22. mock.Mock
  23. }
  24. type mockPacketSender_Expecter struct {
  25. mock *mock.Mock
  26. }
  27. func (_m *mockPacketSender) EXPECT() *mockPacketSender_Expecter {
  28. return &mockPacketSender_Expecter{mock: &_m.Mock}
  29. }
  30. // SendPacket provides a mock function for the type mockPacketSender
  31. func (_mock *mockPacketSender) SendPacket(addr *net.UDPAddr, packet []byte) error {
  32. ret := _mock.Called(addr, packet)
  33. if len(ret) == 0 {
  34. panic("no return value specified for SendPacket")
  35. }
  36. var r0 error
  37. if returnFunc, ok := ret.Get(0).(func(*net.UDPAddr, []byte) error); ok {
  38. r0 = returnFunc(addr, packet)
  39. } else {
  40. r0 = ret.Error(0)
  41. }
  42. return r0
  43. }
  44. // mockPacketSender_SendPacket_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendPacket'
  45. type mockPacketSender_SendPacket_Call struct {
  46. *mock.Call
  47. }
  48. // SendPacket is a helper method to define mock.On call
  49. // - addr *net.UDPAddr
  50. // - packet []byte
  51. func (_e *mockPacketSender_Expecter) SendPacket(addr interface{}, packet interface{}) *mockPacketSender_SendPacket_Call {
  52. return &mockPacketSender_SendPacket_Call{Call: _e.mock.On("SendPacket", addr, packet)}
  53. }
  54. func (_c *mockPacketSender_SendPacket_Call) Run(run func(addr *net.UDPAddr, packet []byte)) *mockPacketSender_SendPacket_Call {
  55. _c.Call.Run(func(args mock.Arguments) {
  56. var arg0 *net.UDPAddr
  57. if args[0] != nil {
  58. arg0 = args[0].(*net.UDPAddr)
  59. }
  60. var arg1 []byte
  61. if args[1] != nil {
  62. arg1 = args[1].([]byte)
  63. }
  64. run(
  65. arg0,
  66. arg1,
  67. )
  68. })
  69. return _c
  70. }
  71. func (_c *mockPacketSender_SendPacket_Call) Return(err error) *mockPacketSender_SendPacket_Call {
  72. _c.Call.Return(err)
  73. return _c
  74. }
  75. func (_c *mockPacketSender_SendPacket_Call) RunAndReturn(run func(addr *net.UDPAddr, packet []byte) error) *mockPacketSender_SendPacket_Call {
  76. _c.Call.Return(run)
  77. return _c
  78. }
  79. // SendToSession provides a mock function for the type mockPacketSender
  80. func (_mock *mockPacketSender) SendToSession(session *LegacySession, packet []byte) error {
  81. ret := _mock.Called(session, packet)
  82. if len(ret) == 0 {
  83. panic("no return value specified for SendToSession")
  84. }
  85. var r0 error
  86. if returnFunc, ok := ret.Get(0).(func(*LegacySession, []byte) error); ok {
  87. r0 = returnFunc(session, packet)
  88. } else {
  89. r0 = ret.Error(0)
  90. }
  91. return r0
  92. }
  93. // mockPacketSender_SendToSession_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendToSession'
  94. type mockPacketSender_SendToSession_Call struct {
  95. *mock.Call
  96. }
  97. // SendToSession is a helper method to define mock.On call
  98. // - session *LegacySession
  99. // - packet []byte
  100. func (_e *mockPacketSender_Expecter) SendToSession(session interface{}, packet interface{}) *mockPacketSender_SendToSession_Call {
  101. return &mockPacketSender_SendToSession_Call{Call: _e.mock.On("SendToSession", session, packet)}
  102. }
  103. func (_c *mockPacketSender_SendToSession_Call) Run(run func(session *LegacySession, packet []byte)) *mockPacketSender_SendToSession_Call {
  104. _c.Call.Run(func(args mock.Arguments) {
  105. var arg0 *LegacySession
  106. if args[0] != nil {
  107. arg0 = args[0].(*LegacySession)
  108. }
  109. var arg1 []byte
  110. if args[1] != nil {
  111. arg1 = args[1].([]byte)
  112. }
  113. run(
  114. arg0,
  115. arg1,
  116. )
  117. })
  118. return _c
  119. }
  120. func (_c *mockPacketSender_SendToSession_Call) Return(err error) *mockPacketSender_SendToSession_Call {
  121. _c.Call.Return(err)
  122. return _c
  123. }
  124. func (_c *mockPacketSender_SendToSession_Call) RunAndReturn(run func(session *LegacySession, packet []byte) error) *mockPacketSender_SendToSession_Call {
  125. _c.Call.Return(run)
  126. return _c
  127. }