refactor the map of sessions into a separate struct

mock_session_handler_test.go

@@ -0,0 +1,78 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: github.com/lucas-clemente/quic-go (interfaces: SessionHandler)
+
+// Package quic is a generated GoMock package.
+package quic
+
+import (
+	reflect "reflect"
+
+	gomock "github.com/golang/mock/gomock"
+	protocol "github.com/lucas-clemente/quic-go/internal/protocol"
+)
+
+// MockSessionHandler is a mock of SessionHandler interface
+type MockSessionHandler struct {
+	ctrl     *gomock.Controller
+	recorder *MockSessionHandlerMockRecorder
+}
+
+// MockSessionHandlerMockRecorder is the mock recorder for MockSessionHandler
+type MockSessionHandlerMockRecorder struct {
+	mock *MockSessionHandler
+}
+
+// NewMockSessionHandler creates a new mock instance
+func NewMockSessionHandler(ctrl *gomock.Controller) *MockSessionHandler {
+	mock := &MockSessionHandler{ctrl: ctrl}
+	mock.recorder = &MockSessionHandlerMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use
+func (m *MockSessionHandler) EXPECT() *MockSessionHandlerMockRecorder {
+	return m.recorder
+}
+
+// Add mocks base method
+func (m *MockSessionHandler) Add(arg0 protocol.ConnectionID, arg1 packetHandler) {
+	m.ctrl.Call(m, "Add", arg0, arg1)
+}
+
+// Add indicates an expected call of Add
+func (mr *MockSessionHandlerMockRecorder) Add(arg0, arg1 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Add", reflect.TypeOf((*MockSessionHandler)(nil).Add), arg0, arg1)
+}
+
+// Close mocks base method
+func (m *MockSessionHandler) Close() {
+	m.ctrl.Call(m, "Close")
+}
+
+// Close indicates an expected call of Close
+func (mr *MockSessionHandlerMockRecorder) Close() *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockSessionHandler)(nil).Close))
+}
+
+// Get mocks base method
+func (m *MockSessionHandler) Get(arg0 protocol.ConnectionID) (packetHandler, bool) {
+	ret := m.ctrl.Call(m, "Get", arg0)
+	ret0, _ := ret[0].(packetHandler)
+	ret1, _ := ret[1].(bool)
+	return ret0, ret1
+}
+
+// Get indicates an expected call of Get
+func (mr *MockSessionHandlerMockRecorder) Get(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockSessionHandler)(nil).Get), arg0)
+}
+
+// Remove mocks base method
+func (m *MockSessionHandler) Remove(arg0 protocol.ConnectionID) {
+	m.ctrl.Call(m, "Remove", arg0)
+}
+
+// Remove indicates an expected call of Remove
+func (mr *MockSessionHandlerMockRecorder) Remove(arg0 interface{}) *gomock.Call {
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Remove", reflect.TypeOf((*MockSessionHandler)(nil).Remove), arg0)
+}