github.com/metacubex/quic-go@v0.44.1-0.20240520163451-20b689a59136/internal/mocks/short_header_sealer.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/metacubex/quic-go/internal/handshake (interfaces: ShortHeaderSealer)
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -typed -build_flags=-tags=gomock -package mocks -destination short_header_sealer.go github.com/metacubex/quic-go/internal/handshake ShortHeaderSealer
     7  //
     8  
     9  // Package mocks is a generated GoMock package.
    10  package mocks
    11  
    12  import (
    13  	reflect "reflect"
    14  
    15  	protocol "github.com/metacubex/quic-go/internal/protocol"
    16  	gomock "go.uber.org/mock/gomock"
    17  )
    18  
    19  // MockShortHeaderSealer is a mock of ShortHeaderSealer interface.
    20  type MockShortHeaderSealer struct {
    21  	ctrl     *gomock.Controller
    22  	recorder *MockShortHeaderSealerMockRecorder
    23  }
    24  
    25  // MockShortHeaderSealerMockRecorder is the mock recorder for MockShortHeaderSealer.
    26  type MockShortHeaderSealerMockRecorder struct {
    27  	mock *MockShortHeaderSealer
    28  }
    29  
    30  // NewMockShortHeaderSealer creates a new mock instance.
    31  func NewMockShortHeaderSealer(ctrl *gomock.Controller) *MockShortHeaderSealer {
    32  	mock := &MockShortHeaderSealer{ctrl: ctrl}
    33  	mock.recorder = &MockShortHeaderSealerMockRecorder{mock}
    34  	return mock
    35  }
    36  
    37  // EXPECT returns an object that allows the caller to indicate expected use.
    38  func (m *MockShortHeaderSealer) EXPECT() *MockShortHeaderSealerMockRecorder {
    39  	return m.recorder
    40  }
    41  
    42  // EncryptHeader mocks base method.
    43  func (m *MockShortHeaderSealer) EncryptHeader(arg0 []byte, arg1 *byte, arg2 []byte) {
    44  	m.ctrl.T.Helper()
    45  	m.ctrl.Call(m, "EncryptHeader", arg0, arg1, arg2)
    46  }
    47  
    48  // EncryptHeader indicates an expected call of EncryptHeader.
    49  func (mr *MockShortHeaderSealerMockRecorder) EncryptHeader(arg0, arg1, arg2 any) *MockShortHeaderSealerEncryptHeaderCall {
    50  	mr.mock.ctrl.T.Helper()
    51  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EncryptHeader", reflect.TypeOf((*MockShortHeaderSealer)(nil).EncryptHeader), arg0, arg1, arg2)
    52  	return &MockShortHeaderSealerEncryptHeaderCall{Call: call}
    53  }
    54  
    55  // MockShortHeaderSealerEncryptHeaderCall wrap *gomock.Call
    56  type MockShortHeaderSealerEncryptHeaderCall struct {
    57  	*gomock.Call
    58  }
    59  
    60  // Return rewrite *gomock.Call.Return
    61  func (c *MockShortHeaderSealerEncryptHeaderCall) Return() *MockShortHeaderSealerEncryptHeaderCall {
    62  	c.Call = c.Call.Return()
    63  	return c
    64  }
    65  
    66  // Do rewrite *gomock.Call.Do
    67  func (c *MockShortHeaderSealerEncryptHeaderCall) Do(f func([]byte, *byte, []byte)) *MockShortHeaderSealerEncryptHeaderCall {
    68  	c.Call = c.Call.Do(f)
    69  	return c
    70  }
    71  
    72  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    73  func (c *MockShortHeaderSealerEncryptHeaderCall) DoAndReturn(f func([]byte, *byte, []byte)) *MockShortHeaderSealerEncryptHeaderCall {
    74  	c.Call = c.Call.DoAndReturn(f)
    75  	return c
    76  }
    77  
    78  // KeyPhase mocks base method.
    79  func (m *MockShortHeaderSealer) KeyPhase() protocol.KeyPhaseBit {
    80  	m.ctrl.T.Helper()
    81  	ret := m.ctrl.Call(m, "KeyPhase")
    82  	ret0, _ := ret[0].(protocol.KeyPhaseBit)
    83  	return ret0
    84  }
    85  
    86  // KeyPhase indicates an expected call of KeyPhase.
    87  func (mr *MockShortHeaderSealerMockRecorder) KeyPhase() *MockShortHeaderSealerKeyPhaseCall {
    88  	mr.mock.ctrl.T.Helper()
    89  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KeyPhase", reflect.TypeOf((*MockShortHeaderSealer)(nil).KeyPhase))
    90  	return &MockShortHeaderSealerKeyPhaseCall{Call: call}
    91  }
    92  
    93  // MockShortHeaderSealerKeyPhaseCall wrap *gomock.Call
    94  type MockShortHeaderSealerKeyPhaseCall struct {
    95  	*gomock.Call
    96  }
    97  
    98  // Return rewrite *gomock.Call.Return
    99  func (c *MockShortHeaderSealerKeyPhaseCall) Return(arg0 protocol.KeyPhaseBit) *MockShortHeaderSealerKeyPhaseCall {
   100  	c.Call = c.Call.Return(arg0)
   101  	return c
   102  }
   103  
   104  // Do rewrite *gomock.Call.Do
   105  func (c *MockShortHeaderSealerKeyPhaseCall) Do(f func() protocol.KeyPhaseBit) *MockShortHeaderSealerKeyPhaseCall {
   106  	c.Call = c.Call.Do(f)
   107  	return c
   108  }
   109  
   110  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   111  func (c *MockShortHeaderSealerKeyPhaseCall) DoAndReturn(f func() protocol.KeyPhaseBit) *MockShortHeaderSealerKeyPhaseCall {
   112  	c.Call = c.Call.DoAndReturn(f)
   113  	return c
   114  }
   115  
   116  // Overhead mocks base method.
   117  func (m *MockShortHeaderSealer) Overhead() int {
   118  	m.ctrl.T.Helper()
   119  	ret := m.ctrl.Call(m, "Overhead")
   120  	ret0, _ := ret[0].(int)
   121  	return ret0
   122  }
   123  
   124  // Overhead indicates an expected call of Overhead.
   125  func (mr *MockShortHeaderSealerMockRecorder) Overhead() *MockShortHeaderSealerOverheadCall {
   126  	mr.mock.ctrl.T.Helper()
   127  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Overhead", reflect.TypeOf((*MockShortHeaderSealer)(nil).Overhead))
   128  	return &MockShortHeaderSealerOverheadCall{Call: call}
   129  }
   130  
   131  // MockShortHeaderSealerOverheadCall wrap *gomock.Call
   132  type MockShortHeaderSealerOverheadCall struct {
   133  	*gomock.Call
   134  }
   135  
   136  // Return rewrite *gomock.Call.Return
   137  func (c *MockShortHeaderSealerOverheadCall) Return(arg0 int) *MockShortHeaderSealerOverheadCall {
   138  	c.Call = c.Call.Return(arg0)
   139  	return c
   140  }
   141  
   142  // Do rewrite *gomock.Call.Do
   143  func (c *MockShortHeaderSealerOverheadCall) Do(f func() int) *MockShortHeaderSealerOverheadCall {
   144  	c.Call = c.Call.Do(f)
   145  	return c
   146  }
   147  
   148  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   149  func (c *MockShortHeaderSealerOverheadCall) DoAndReturn(f func() int) *MockShortHeaderSealerOverheadCall {
   150  	c.Call = c.Call.DoAndReturn(f)
   151  	return c
   152  }
   153  
   154  // Seal mocks base method.
   155  func (m *MockShortHeaderSealer) Seal(arg0, arg1 []byte, arg2 protocol.PacketNumber, arg3 []byte) []byte {
   156  	m.ctrl.T.Helper()
   157  	ret := m.ctrl.Call(m, "Seal", arg0, arg1, arg2, arg3)
   158  	ret0, _ := ret[0].([]byte)
   159  	return ret0
   160  }
   161  
   162  // Seal indicates an expected call of Seal.
   163  func (mr *MockShortHeaderSealerMockRecorder) Seal(arg0, arg1, arg2, arg3 any) *MockShortHeaderSealerSealCall {
   164  	mr.mock.ctrl.T.Helper()
   165  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Seal", reflect.TypeOf((*MockShortHeaderSealer)(nil).Seal), arg0, arg1, arg2, arg3)
   166  	return &MockShortHeaderSealerSealCall{Call: call}
   167  }
   168  
   169  // MockShortHeaderSealerSealCall wrap *gomock.Call
   170  type MockShortHeaderSealerSealCall struct {
   171  	*gomock.Call
   172  }
   173  
   174  // Return rewrite *gomock.Call.Return
   175  func (c *MockShortHeaderSealerSealCall) Return(arg0 []byte) *MockShortHeaderSealerSealCall {
   176  	c.Call = c.Call.Return(arg0)
   177  	return c
   178  }
   179  
   180  // Do rewrite *gomock.Call.Do
   181  func (c *MockShortHeaderSealerSealCall) Do(f func([]byte, []byte, protocol.PacketNumber, []byte) []byte) *MockShortHeaderSealerSealCall {
   182  	c.Call = c.Call.Do(f)
   183  	return c
   184  }
   185  
   186  // DoAndReturn rewrite *gomock.Call.DoAndReturn
   187  func (c *MockShortHeaderSealerSealCall) DoAndReturn(f func([]byte, []byte, protocol.PacketNumber, []byte) []byte) *MockShortHeaderSealerSealCall {
   188  	c.Call = c.Call.DoAndReturn(f)
   189  	return c
   190  }