github.com/quic-go/quic-go@v0.44.0/mock_frame_source_test.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/quic-go/quic-go (interfaces: FrameSource)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -typed -build_flags=-tags=gomock -package quic -self_package github.com/quic-go/quic-go -destination mock_frame_source_test.go github.com/quic-go/quic-go FrameSource
     7  //
     8  
     9  // Package quic is a generated GoMock package.
    10  package quic
    11  
    12  import (
    13  	reflect "reflect"
    14  
    15  	ackhandler "github.com/quic-go/quic-go/internal/ackhandler"
    16  	protocol "github.com/quic-go/quic-go/internal/protocol"
    17  	gomock "go.uber.org/mock/gomock"
    18  )
    19  
    20  // MockFrameSource is a mock of FrameSource interface.
    21  type MockFrameSource struct {
    22  	ctrl     *gomock.Controller
    23  	recorder *MockFrameSourceMockRecorder
    24  }
    25  
    26  // MockFrameSourceMockRecorder is the mock recorder for MockFrameSource.
    27  type MockFrameSourceMockRecorder struct {
    28  	mock *MockFrameSource
    29  }
    30  
    31  // NewMockFrameSource creates a new mock instance.
    32  func NewMockFrameSource(ctrl *gomock.Controller) *MockFrameSource {
    33  	mock := &MockFrameSource{ctrl: ctrl}
    34  	mock.recorder = &MockFrameSourceMockRecorder{mock}
    35  	return mock
    36  }
    37  
    38  // EXPECT returns an object that allows the caller to indicate expected use.
    39  func (m *MockFrameSource) EXPECT() *MockFrameSourceMockRecorder {
    40  	return m.recorder
    41  }
    42  
    43  // AppendControlFrames mocks base method.
    44  func (m *MockFrameSource) AppendControlFrames(arg0 []ackhandler.Frame, arg1 protocol.ByteCount, arg2 protocol.Version) ([]ackhandler.Frame, protocol.ByteCount) {
    45  	m.ctrl.T.Helper()
    46  	ret := m.ctrl.Call(m, "AppendControlFrames", arg0, arg1, arg2)
    47  	ret0, _ := ret[0].([]ackhandler.Frame)
    48  	ret1, _ := ret[1].(protocol.ByteCount)
    49  	return ret0, ret1
    50  }
    51  
    52  // AppendControlFrames indicates an expected call of AppendControlFrames.
    53  func (mr *MockFrameSourceMockRecorder) AppendControlFrames(arg0, arg1, arg2 any) *MockFrameSourceAppendControlFramesCall {
    54  	mr.mock.ctrl.T.Helper()
    55  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendControlFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendControlFrames), arg0, arg1, arg2)
    56  	return &MockFrameSourceAppendControlFramesCall{Call: call}
    57  }
    58  
    59  // MockFrameSourceAppendControlFramesCall wrap *gomock.Call
    60  type MockFrameSourceAppendControlFramesCall struct {
    61  	*gomock.Call
    62  }
    63  
    64  // Return rewrite *gomock.Call.Return
    65  func (c *MockFrameSourceAppendControlFramesCall) Return(arg0 []ackhandler.Frame, arg1 protocol.ByteCount) *MockFrameSourceAppendControlFramesCall {
    66  	c.Call = c.Call.Return(arg0, arg1)
    67  	return c
    68  }
    69  
    70  // Do rewrite *gomock.Call.Do
    71  func (c *MockFrameSourceAppendControlFramesCall) Do(f func([]ackhandler.Frame, protocol.ByteCount, protocol.Version) ([]ackhandler.Frame, protocol.ByteCount)) *MockFrameSourceAppendControlFramesCall {
    72  	c.Call = c.Call.Do(f)
    73  	return c
    74  }
    75  
    76  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    77  func (c *MockFrameSourceAppendControlFramesCall) DoAndReturn(f func([]ackhandler.Frame, protocol.ByteCount, protocol.Version) ([]ackhandler.Frame, protocol.ByteCount)) *MockFrameSourceAppendControlFramesCall {
    78  	c.Call = c.Call.DoAndReturn(f)
    79  	return c
    80  }
    81  
    82  // AppendStreamFrames mocks base method.
    83  func (m *MockFrameSource) AppendStreamFrames(arg0 []ackhandler.StreamFrame, arg1 protocol.ByteCount, arg2 protocol.Version) ([]ackhandler.StreamFrame, protocol.ByteCount) {
    84  	m.ctrl.T.Helper()
    85  	ret := m.ctrl.Call(m, "AppendStreamFrames", arg0, arg1, arg2)
    86  	ret0, _ := ret[0].([]ackhandler.StreamFrame)
    87  	ret1, _ := ret[1].(protocol.ByteCount)
    88  	return ret0, ret1
    89  }
    90  
    91  // AppendStreamFrames indicates an expected call of AppendStreamFrames.
    92  func (mr *MockFrameSourceMockRecorder) AppendStreamFrames(arg0, arg1, arg2 any) *MockFrameSourceAppendStreamFramesCall {
    93  	mr.mock.ctrl.T.Helper()
    94  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendStreamFrames", reflect.TypeOf((*MockFrameSource)(nil).AppendStreamFrames), arg0, arg1, arg2)
    95  	return &MockFrameSourceAppendStreamFramesCall{Call: call}
    96  }
    97  
    98  // MockFrameSourceAppendStreamFramesCall wrap *gomock.Call
    99  type MockFrameSourceAppendStreamFramesCall struct {
   100  	*gomock.Call
   101  }
   102  
   103  // Return rewrite *gomock.Call.Return
   104  func (c *MockFrameSourceAppendStreamFramesCall) Return(arg0 []ackhandler.StreamFrame, arg1 protocol.ByteCount) *MockFrameSourceAppendStreamFramesCall {
   105  	c.Call = c.Call.Return(arg0, arg1)
   106  	return c
   107  }
   108  
   109  // Do rewrite *gomock.Call.Do
   110  func (c *MockFrameSourceAppendStreamFramesCall) Do(f func([]ackhandler.StreamFrame, protocol.ByteCount, protocol.Version) ([]ackhandler.StreamFrame, protocol.ByteCount)) *MockFrameSourceAppendStreamFramesCall {
   111  	c.Call = c.Call.Do(f)
   112  	return c
   113  }
   114  
   115  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   116  func (c *MockFrameSourceAppendStreamFramesCall) DoAndReturn(f func([]ackhandler.StreamFrame, protocol.ByteCount, protocol.Version) ([]ackhandler.StreamFrame, protocol.ByteCount)) *MockFrameSourceAppendStreamFramesCall {
   117  	c.Call = c.Call.DoAndReturn(f)
   118  	return c
   119  }
   120  
   121  // HasData mocks base method.
   122  func (m *MockFrameSource) HasData() bool {
   123  	m.ctrl.T.Helper()
   124  	ret := m.ctrl.Call(m, "HasData")
   125  	ret0, _ := ret[0].(bool)
   126  	return ret0
   127  }
   128  
   129  // HasData indicates an expected call of HasData.
   130  func (mr *MockFrameSourceMockRecorder) HasData() *MockFrameSourceHasDataCall {
   131  	mr.mock.ctrl.T.Helper()
   132  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HasData", reflect.TypeOf((*MockFrameSource)(nil).HasData))
   133  	return &MockFrameSourceHasDataCall{Call: call}
   134  }
   135  
   136  // MockFrameSourceHasDataCall wrap *gomock.Call
   137  type MockFrameSourceHasDataCall struct {
   138  	*gomock.Call
   139  }
   140  
   141  // Return rewrite *gomock.Call.Return
   142  func (c *MockFrameSourceHasDataCall) Return(arg0 bool) *MockFrameSourceHasDataCall {
   143  	c.Call = c.Call.Return(arg0)
   144  	return c
   145  }
   146  
   147  // Do rewrite *gomock.Call.Do
   148  func (c *MockFrameSourceHasDataCall) Do(f func() bool) *MockFrameSourceHasDataCall {
   149  	c.Call = c.Call.Do(f)
   150  	return c
   151  }
   152  
   153  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   154  func (c *MockFrameSourceHasDataCall) DoAndReturn(f func() bool) *MockFrameSourceHasDataCall {
   155  	c.Call = c.Call.DoAndReturn(f)
   156  	return c
   157  }