use gomock to mock crypto.AEAD in the crypto setup tests

internal/mocks/crypto/aead.go

@@ -0,0 +1,71 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: github.com/lucas-clemente/quic-go/internal/crypto (interfaces: AEAD)
+
+package mockcrypto
+
+import (
+	reflect "reflect"
+
+	gomock "github.com/golang/mock/gomock"
+	protocol "github.com/lucas-clemente/quic-go/internal/protocol"
+)
+
+// MockAEAD is a mock of AEAD interface
+type MockAEAD struct {
+	ctrl     *gomock.Controller
+	recorder *MockAEADMockRecorder
+}
+
+// MockAEADMockRecorder is the mock recorder for MockAEAD
+type MockAEADMockRecorder struct {
+	mock *MockAEAD
+}
+
+// NewMockAEAD creates a new mock instance
+func NewMockAEAD(ctrl *gomock.Controller) *MockAEAD {
+	mock := &MockAEAD{ctrl: ctrl}
+	mock.recorder = &MockAEADMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use
+func (_m *MockAEAD) EXPECT() *MockAEADMockRecorder {
+	return _m.recorder
+}
+
+// Open mocks base method
+func (_m *MockAEAD) Open(_param0 []byte, _param1 []byte, _param2 protocol.PacketNumber, _param3 []byte) ([]byte, error) {
+	ret := _m.ctrl.Call(_m, "Open", _param0, _param1, _param2, _param3)
+	ret0, _ := ret[0].([]byte)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Open indicates an expected call of Open
+func (_mr *MockAEADMockRecorder) Open(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "Open", reflect.TypeOf((*MockAEAD)(nil).Open), arg0, arg1, arg2, arg3)
+}
+
+// Overhead mocks base method
+func (_m *MockAEAD) Overhead() int {
+	ret := _m.ctrl.Call(_m, "Overhead")
+	ret0, _ := ret[0].(int)
+	return ret0
+}
+
+// Overhead indicates an expected call of Overhead
+func (_mr *MockAEADMockRecorder) Overhead() *gomock.Call {
+	return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "Overhead", reflect.TypeOf((*MockAEAD)(nil).Overhead))
+}
+
+// Seal mocks base method
+func (_m *MockAEAD) Seal(_param0 []byte, _param1 []byte, _param2 protocol.PacketNumber, _param3 []byte) []byte {
+	ret := _m.ctrl.Call(_m, "Seal", _param0, _param1, _param2, _param3)
+	ret0, _ := ret[0].([]byte)
+	return ret0
+}
+
+// Seal indicates an expected call of Seal
+func (_mr *MockAEADMockRecorder) Seal(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "Seal", reflect.TypeOf((*MockAEAD)(nil).Seal), arg0, arg1, arg2, arg3)
+}

internal/mocks/gen.go

@@ -2,5 +2,6 @@ package mocks
 
 //go:generate sh -c "./mockgen_internal.sh mocks stream_flow_controller.go github.com/lucas-clemente/quic-go/internal/flowcontrol StreamFlowController"
 //go:generate sh -c "./mockgen_internal.sh mocks connection_flow_controller.go github.com/lucas-clemente/quic-go/internal/flowcontrol ConnectionFlowController"
+//go:generate sh -c "./mockgen_internal.sh mockcrypto crypto/aead.go github.com/lucas-clemente/quic-go/internal/crypto AEAD"
 //go:generate sh -c "./mockgen_stream.sh mocks stream.go github.com/lucas-clemente/quic-go StreamI"
 //go:generate sh -c "goimports -w ."