SERVER-26674 transport::Session objects should be shared_ptr managed

1 files changed, 17 insertions, 13 deletions

diff --git a/src/mongo/transport/service_entry_point_test_suite.h b/src/mongo/transport/service_entry_point_test_suite.h
index ba1d60fd027..d1d3042cc5e 100644
--- a/src/mongo/transport/service_entry_point_test_suite.h
+++ b/src/mongo/transport/service_entry_point_test_suite.h
@@ -92,10 +92,10 @@ private:
     class MockTicket : public transport::TicketImpl {
     public:
         // Source constructor
-        MockTicket(const transport::Session& session, Message* message, Date_t expiration);
+        MockTicket(const transport::SessionHandle& session, Message* message, Date_t expiration);
 
         // Sink constructor
-        MockTicket(const transport::Session& session, Date_t expiration);
+        MockTicket(const transport::SessionHandle& session, Date_t expiration);
 
         MockTicket(MockTicket&&) = default;
         MockTicket& operator=(MockTicket&&) = default;
@@ -121,19 +121,19 @@ private:
         MockTLHarness();
 
         transport::Ticket sourceMessage(
-            transport::Session& session,
+            const transport::SessionHandle& session,
             Message* message,
             Date_t expiration = transport::Ticket::kNoExpirationDate) override;
         transport::Ticket sinkMessage(
-            transport::Session& session,
+            const transport::SessionHandle& session,
             const Message& message,
             Date_t expiration = transport::Ticket::kNoExpirationDate) override;
         Status wait(transport::Ticket&& ticket) override;
         void asyncWait(transport::Ticket&& ticket, TicketCallback callback) override;
-        SSLPeerInfo getX509PeerInfo(const transport::Session& session) const override;
-        void registerTags(const transport::Session& session) override;
+        SSLPeerInfo getX509PeerInfo(const transport::ConstSessionHandle& session) const override;
+        void registerTags(const transport::ConstSessionHandle& session) override;
         Stats sessionStats() override;
-        void end(transport::Session& session) override;
+        void end(const transport::SessionHandle& session) override;
         void endAllSessions(transport::Session::TagMask tags) override;
         Status start() override;
         void shutdown() override;
@@ -141,11 +141,13 @@ private:
         ServiceEntryPointTestSuite::MockTicket* getMockTicket(const transport::Ticket& ticket);
 
         // Mocked method hooks
-        stdx::function<transport::Ticket(transport::Session&, Message*, Date_t)> _sourceMessage;
-        stdx::function<transport::Ticket(transport::Session&, const Message&, Date_t)> _sinkMessage;
+        stdx::function<transport::Ticket(const transport::SessionHandle&, Message*, Date_t)>
+            _sourceMessage;
+        stdx::function<transport::Ticket(const transport::SessionHandle&, const Message&, Date_t)>
+            _sinkMessage;
         stdx::function<Status(transport::Ticket)> _wait;
         stdx::function<void(transport::Ticket, TicketCallback)> _asyncWait;
-        stdx::function<void(const transport::Session&)> _end;
+        stdx::function<void(const transport::SessionHandle&)> _end;
         stdx::function<void(transport::Session& session)> _destroy_hook;
         stdx::function<void(transport::Session::TagMask tags)> _endAllSessions =
             [](transport::Session::TagMask tags) {};
@@ -153,9 +155,11 @@ private:
         stdx::function<void(void)> _shutdown = [] {};
 
         // Pre-set hook methods
-        transport::Ticket _defaultSource(transport::Session& s, Message* m, Date_t d);
-        transport::Ticket _defaultSink(transport::Session& s, const Message&, Date_t d);
-        transport::Ticket _sinkThenErrorOnWait(transport::Session& s, const Message& m, Date_t d);
+        transport::Ticket _defaultSource(const transport::SessionHandle& s, Message* m, Date_t d);
+        transport::Ticket _defaultSink(const transport::SessionHandle& s, const Message&, Date_t d);
+        transport::Ticket _sinkThenErrorOnWait(const transport::SessionHandle& s,
+                                               const Message& m,
+                                               Date_t d);
 
         Status _defaultWait(transport::Ticket ticket);
         Status _waitError(transport::Ticket ticket);