mock_response_writer_test.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. // Code generated by mockery; DO NOT EDIT.
  2. // github.com/vektra/mockery
  3. // template: testify
  4. package oscar
  5. import (
  6. "github.com/mk6i/open-oscar-server/wire"
  7. mock "github.com/stretchr/testify/mock"
  8. )
  9. // newMockResponseWriter creates a new instance of mockResponseWriter. 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 newMockResponseWriter(t interface {
  12. mock.TestingT
  13. Cleanup(func())
  14. }) *mockResponseWriter {
  15. if helper, ok := t.(interface{ Helper() }); ok {
  16. helper.Helper()
  17. }
  18. mock := &mockResponseWriter{}
  19. mock.Mock.Test(t)
  20. t.Cleanup(func() {
  21. if helper, ok := t.(interface{ Helper() }); ok {
  22. helper.Helper()
  23. }
  24. mock.AssertExpectations(t)
  25. })
  26. return mock
  27. }
  28. // mockResponseWriter is an autogenerated mock type for the ResponseWriter type
  29. type mockResponseWriter struct {
  30. mock.Mock
  31. }
  32. type mockResponseWriter_Expecter struct {
  33. mock *mock.Mock
  34. }
  35. func (_m *mockResponseWriter) EXPECT() *mockResponseWriter_Expecter {
  36. return &mockResponseWriter_Expecter{mock: &_m.Mock}
  37. }
  38. // SendSNAC provides a mock function for the type mockResponseWriter
  39. func (_mock *mockResponseWriter) SendSNAC(frame wire.SNACFrame, body any) error {
  40. ret := _mock.Called(frame, body)
  41. if len(ret) == 0 {
  42. panic("no return value specified for SendSNAC")
  43. }
  44. var r0 error
  45. if returnFunc, ok := ret.Get(0).(func(wire.SNACFrame, any) error); ok {
  46. r0 = returnFunc(frame, body)
  47. } else {
  48. r0 = ret.Error(0)
  49. }
  50. return r0
  51. }
  52. // mockResponseWriter_SendSNAC_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendSNAC'
  53. type mockResponseWriter_SendSNAC_Call struct {
  54. *mock.Call
  55. }
  56. // SendSNAC is a helper method to define mock.On call
  57. // - frame wire.SNACFrame
  58. // - body any
  59. func (_e *mockResponseWriter_Expecter) SendSNAC(frame any, body any) *mockResponseWriter_SendSNAC_Call {
  60. return &mockResponseWriter_SendSNAC_Call{Call: _e.mock.On("SendSNAC", frame, body)}
  61. }
  62. func (_c *mockResponseWriter_SendSNAC_Call) Run(run func(frame wire.SNACFrame, body any)) *mockResponseWriter_SendSNAC_Call {
  63. _c.Call.Run(func(args mock.Arguments) {
  64. var arg0 wire.SNACFrame
  65. if args[0] != nil {
  66. arg0 = args[0].(wire.SNACFrame)
  67. }
  68. var arg1 any
  69. if args[1] != nil {
  70. arg1 = args[1].(any)
  71. }
  72. run(
  73. arg0,
  74. arg1,
  75. )
  76. })
  77. return _c
  78. }
  79. func (_c *mockResponseWriter_SendSNAC_Call) Return(err error) *mockResponseWriter_SendSNAC_Call {
  80. _c.Call.Return(err)
  81. return _c
  82. }
  83. func (_c *mockResponseWriter_SendSNAC_Call) RunAndReturn(run func(frame wire.SNACFrame, body any) error) *mockResponseWriter_SendSNAC_Call {
  84. _c.Call.Return(run)
  85. return _c
  86. }