get ACK and STOP_WAITING frames in the packet packer

mock_packer_test.go

@@ -0,0 +1,129 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: github.com/lucas-clemente/quic-go (interfaces: Packer)
+
+// Package quic is a generated GoMock package.
+package quic
+
+import (
+	reflect "reflect"
+
+	gomock "github.com/golang/mock/gomock"
+	ackhandler "github.com/lucas-clemente/quic-go/internal/ackhandler"
+	protocol "github.com/lucas-clemente/quic-go/internal/protocol"
+	wire "github.com/lucas-clemente/quic-go/internal/wire"
+)
+
+// MockPacker is a mock of Packer interface
+type MockPacker struct {
+	ctrl     *gomock.Controller
+	recorder *MockPackerMockRecorder
+}
+
+// MockPackerMockRecorder is the mock recorder for MockPacker
+type MockPackerMockRecorder struct {
+	mock *MockPacker
+}
+
+// NewMockPacker creates a new mock instance
+func NewMockPacker(ctrl *gomock.Controller) *MockPacker {
+	mock := &MockPacker{ctrl: ctrl}
+	mock.recorder = &MockPackerMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use
+func (m *MockPacker) EXPECT() *MockPackerMockRecorder {
+	return m.recorder
+}
+
+// ChangeDestConnectionID mocks base method
+func (m *MockPacker) ChangeDestConnectionID(arg0 protocol.ConnectionID) {
+	m.ctrl.Call(m, "ChangeDestConnectionID", arg0)
+}
+
+// ChangeDestConnectionID indicates an expected call of ChangeDestConnectionID
+func (mr *MockPackerMockRecorder) ChangeDestConnectionID(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeDestConnectionID", reflect.TypeOf((*MockPacker)(nil).ChangeDestConnectionID), arg0)
+}
+
+// MaybePackAckPacket mocks base method
+func (m *MockPacker) MaybePackAckPacket() (*packedPacket, error) {
+	ret := m.ctrl.Call(m, "MaybePackAckPacket")
+	ret0, _ := ret[0].(*packedPacket)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// MaybePackAckPacket indicates an expected call of MaybePackAckPacket
+func (mr *MockPackerMockRecorder) MaybePackAckPacket() *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MaybePackAckPacket", reflect.TypeOf((*MockPacker)(nil).MaybePackAckPacket))
+}
+
+// PackConnectionClose mocks base method
+func (m *MockPacker) PackConnectionClose(arg0 *wire.ConnectionCloseFrame) (*packedPacket, error) {
+	ret := m.ctrl.Call(m, "PackConnectionClose", arg0)
+	ret0, _ := ret[0].(*packedPacket)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// PackConnectionClose indicates an expected call of PackConnectionClose
+func (mr *MockPackerMockRecorder) PackConnectionClose(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PackConnectionClose", reflect.TypeOf((*MockPacker)(nil).PackConnectionClose), arg0)
+}
+
+// PackPacket mocks base method
+func (m *MockPacker) PackPacket() (*packedPacket, error) {
+	ret := m.ctrl.Call(m, "PackPacket")
+	ret0, _ := ret[0].(*packedPacket)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// PackPacket indicates an expected call of PackPacket
+func (mr *MockPackerMockRecorder) PackPacket() *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PackPacket", reflect.TypeOf((*MockPacker)(nil).PackPacket))
+}
+
+// PackRetransmission mocks base method
+func (m *MockPacker) PackRetransmission(arg0 *ackhandler.Packet) ([]*packedPacket, error) {
+	ret := m.ctrl.Call(m, "PackRetransmission", arg0)
+	ret0, _ := ret[0].([]*packedPacket)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// PackRetransmission indicates an expected call of PackRetransmission
+func (mr *MockPackerMockRecorder) PackRetransmission(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PackRetransmission", reflect.TypeOf((*MockPacker)(nil).PackRetransmission), arg0)
+}
+
+// QueueControlFrame mocks base method
+func (m *MockPacker) QueueControlFrame(arg0 wire.Frame) {
+	m.ctrl.Call(m, "QueueControlFrame", arg0)
+}
+
+// QueueControlFrame indicates an expected call of QueueControlFrame
+func (mr *MockPackerMockRecorder) QueueControlFrame(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueueControlFrame", reflect.TypeOf((*MockPacker)(nil).QueueControlFrame), arg0)
+}
+
+// SetMaxPacketSize mocks base method
+func (m *MockPacker) SetMaxPacketSize(arg0 protocol.ByteCount) {
+	m.ctrl.Call(m, "SetMaxPacketSize", arg0)
+}
+
+// SetMaxPacketSize indicates an expected call of SetMaxPacketSize
+func (mr *MockPackerMockRecorder) SetMaxPacketSize(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetMaxPacketSize", reflect.TypeOf((*MockPacker)(nil).SetMaxPacketSize), arg0)
+}
+
+// SetOmitConnectionID mocks base method
+func (m *MockPacker) SetOmitConnectionID() {
+	m.ctrl.Call(m, "SetOmitConnectionID")
+}
+
+// SetOmitConnectionID indicates an expected call of SetOmitConnectionID
+func (mr *MockPackerMockRecorder) SetOmitConnectionID() *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetOmitConnectionID", reflect.TypeOf((*MockPacker)(nil).SetOmitConnectionID))
+}