mock_frame_source_test.go 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  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. ret := m.ctrl.Call(m, "AppendControlFrames", arg0, arg1)
  33. ret0, _ := ret[0].([]wire.Frame)
  34. ret1, _ := ret[1].(protocol.ByteCount)
  35. return ret0, ret1
  36. }
  37. // AppendControlFrames indicates an expected call of AppendControlFrames
  38. func (mr *MockFrameSourceMockRecorder) AppendControlFrames(arg0, arg1 interface{}) *gomock.Call {
  39. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendControlFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendControlFrames), arg0, arg1)
  40. }
  41. // AppendStreamFrames mocks base method
  42. func (m *MockFrameSource) AppendStreamFrames(arg0 []wire.Frame, arg1 protocol.ByteCount) []wire.Frame {
  43. ret := m.ctrl.Call(m, "AppendStreamFrames", arg0, arg1)
  44. ret0, _ := ret[0].([]wire.Frame)
  45. return ret0
  46. }
  47. // AppendStreamFrames indicates an expected call of AppendStreamFrames
  48. func (mr *MockFrameSourceMockRecorder) AppendStreamFrames(arg0, arg1 interface{}) *gomock.Call {
  49. return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendStreamFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendStreamFrames), arg0, arg1)
  50. }