add a function to close the packet handler map

Close will close the underlying connection and wait until listen has
returned. While not strictly necessary in production use, this will fix
a few race conditions in our tests.
This commit is contained in:
Marten Seemann 2019-01-24 18:09:46 +07:00
parent 6dc4be9f4e
commit bb185a3ad2
5 changed files with 32 additions and 7 deletions

View file

@ -44,6 +44,18 @@ func (mr *MockPacketHandlerManagerMockRecorder) Add(arg0, arg1 interface{}) *gom
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Add", reflect.TypeOf((*MockPacketHandlerManager)(nil).Add), arg0, arg1)
}
// Close mocks base method
func (m *MockPacketHandlerManager) Close() error {
ret := m.ctrl.Call(m, "Close")
ret0, _ := ret[0].(error)
return ret0
}
// Close indicates an expected call of Close
func (mr *MockPacketHandlerManagerMockRecorder) Close() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockPacketHandlerManager)(nil).Close))
}
// CloseServer mocks base method
func (m *MockPacketHandlerManager) CloseServer() {
m.ctrl.Call(m, "CloseServer")