mock_frame_source_test.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. // Code generated by MockGen. DO NOT EDIT.
  2. // Source: github.com/lucas-clemente/quic-go (interfaces: FrameSource)
  3. // Package quic is a generated GoMock package.
  4. package quic
  5. import (
  6. reflect "reflect"
  7. gomock "github.com/golang/mock/gomock"
  8. protocol "github.com/lucas-clemente/quic-go/internal/protocol"
  9. wire "github.com/lucas-clemente/quic-go/internal/wire"
  10. )
  11. // MockFrameSource is a mock of FrameSource interface
  12. type MockFrameSource struct {
  13. ctrl *gomock.Controller
  14. recorder *MockFrameSourceMockRecorder
  15. }
  16. // MockFrameSourceMockRecorder is the mock recorder for MockFrameSource
  17. type MockFrameSourceMockRecorder struct {
  18. mock *MockFrameSource
  19. }
  20. // NewMockFrameSource creates a new mock instance
  21. func NewMockFrameSource(ctrl *gomock.Controller) *MockFrameSource {
  22. mock := &MockFrameSource{ctrl: ctrl}
  23. mock.recorder = &MockFrameSourceMockRecorder{mock}
  24. return mock
  25. }
  26. // EXPECT returns an object that allows the caller to indicate expected use
  27. func (m *MockFrameSource) EXPECT() *MockFrameSourceMockRecorder {
  28. return m.recorder
  29. }
  30. // AppendControlFrames mocks base method
  31. func (m *MockFrameSource) AppendControlFrames(arg0 []wire.Frame, arg1 protocol.ByteCount) ([]wire.Frame, protocol.ByteCount) {
  32. m.ctrl.T.Helper()
  33. ret := m.ctrl.Call(m, "AppendControlFrames", arg0, arg1)
  34. ret0, _ := ret[0].([]wire.Frame)
  35. ret1, _ := ret[1].(protocol.ByteCount)
  36. return ret0, ret1
  37. }
  38. // AppendControlFrames indicates an expected call of AppendControlFrames
  39. func (mr *MockFrameSourceMockRecorder) AppendControlFrames(arg0, arg1 interface{}) *gomock.Call {
  40. mr.mock.ctrl.T.Helper()
  41. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendControlFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendControlFrames), arg0, arg1)
  42. }
  43. // AppendStreamFrames mocks base method
  44. func (m *MockFrameSource) AppendStreamFrames(arg0 []wire.Frame, arg1 protocol.ByteCount) ([]wire.Frame, protocol.ByteCount) {
  45. m.ctrl.T.Helper()
  46. ret := m.ctrl.Call(m, "AppendStreamFrames", arg0, arg1)
  47. ret0, _ := ret[0].([]wire.Frame)
  48. ret1, _ := ret[1].(protocol.ByteCount)
  49. return ret0, ret1
  50. }
  51. // AppendStreamFrames indicates an expected call of AppendStreamFrames
  52. func (mr *MockFrameSourceMockRecorder) AppendStreamFrames(arg0, arg1 interface{}) *gomock.Call {
  53. mr.mock.ctrl.T.Helper()
  54. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendStreamFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendStreamFrames), arg0, arg1)
  55. }