Minor refactor of MoQSessionTest

Summary: So we can use setupMoQSession in subsequent tests

Reviewed By: NEUDitao

Differential Revision: D66563096

fbshipit-source-id: 1aa0b3e4d34a73ded90e2e4e1b94b454be038eac

moxygen/test/MoQSessionTest.cpp

@@ -151,6 +151,8 @@ class MoQSessionTest : public testing::Test,
               .asUint64 = kTestMaxSubscribeId}}}});
   }
 
+  void setupMoQSession();
+
  protected:
   folly::EventBase eventBase_;
   std::unique_ptr<proxygen::test::FakeSharedWebTransport> clientWt_;
@@ -162,9 +164,8 @@ class MoQSessionTest : public testing::Test,
   uint64_t negotiatedVersion_ = kVersionDraftCurrent;
   bool failServerSetup_{false};
 };
-} // namespace
 
-TEST_F(MoQSessionTest, Setup) {
+void MoQSessionTest::setupMoQSession() {
   clientSession_->start();
   serverSession_->start();
   eventBase_.loopOnce();
@@ -180,7 +181,7 @@ TEST_F(MoQSessionTest, Setup) {
         serverSetup.params.at(0).key,
         folly::to_underlying(SetupKey::MAX_SUBSCRIBE_ID));
     EXPECT_EQ(serverSetup.params.at(0).asUint64, kTestMaxSubscribeId);
-    clientSession->close();
+    clientSession->getEventBase()->terminateLoopSoon();
   }(clientSession_)
                                                        .scheduleOn(&eventBase_)
                                                        .start();
@@ -192,6 +193,12 @@ TEST_F(MoQSessionTest, Setup) {
       .start();
   eventBase_.loop();
 }
+} // namespace
+
+TEST_F(MoQSessionTest, Setup) {
+  setupMoQSession();
+  clientSession_->close();
+}
 
 // receive bidi stream on client