diff options
Diffstat (limited to 'src/mongo/transport/service_entry_point_test_suite.h')
| -rw-r--r-- | src/mongo/transport/service_entry_point_test_suite.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/mongo/transport/service_entry_point_test_suite.h b/src/mongo/transport/service_entry_point_test_suite.h index c9b9c20702e..ba1d60fd027 100644 --- a/src/mongo/transport/service_entry_point_test_suite.h +++ b/src/mongo/transport/service_entry_point_test_suite.h @@ -146,6 +146,7 @@ private: stdx::function<Status(transport::Ticket)> _wait; stdx::function<void(transport::Ticket, TicketCallback)> _asyncWait; stdx::function<void(const transport::Session&)> _end; + stdx::function<void(transport::Session& session)> _destroy_hook; stdx::function<void(transport::Session::TagMask tags)> _endAllSessions = [](transport::Session::TagMask tags) {}; stdx::function<Status(void)> _start = [] { return Status::OK(); }; @@ -162,6 +163,9 @@ private: // Reset all hooks to their defaults void _resetHooks(); + + private: + void _destroy(transport::Session& session) override; }; std::unique_ptr<MockTLHarness> _tl; |