Add basic unit test that we get ConnectionInformation in callbacks

OwnZones · May 10, 2024 · 012277e · 012277e
1 parent 37ef9ba
commit 012277e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/test/TestSrt.cpp b/test/TestSrt.cpp
@@ -80,12 +80,14 @@ class TestSRTFixture : public ::testing::Test {
         // notice when client connects to server
         mServer.clientConnected = [&](struct sockaddr& sin, SRTSOCKET newSocket,
                                       std::shared_ptr<SRTNet::NetworkConnection>& ctx,
-                                      const SRTNet::ConnectionInformation&) {
+                                      const SRTNet::ConnectionInformation& connectionInformation) {
             {
                 std::lock_guard<std::mutex> lock(mConnectedMutex);
                 mConnected = true;
             }
             mConnectedCondition.notify_one();
+            EXPECT_NE(connectionInformation.mPeerSrtVersion, SRTNet::ConnectionInformation().mPeerSrtVersion);
+            EXPECT_NE(connectionInformation.mNegotiatedLatency, SRTNet::ConnectionInformation().mNegotiatedLatency);
             return mConnectionCtx;
         };
 
@@ -98,6 +100,12 @@ class TestSRTFixture : public ::testing::Test {
             EXPECT_EQ(ctx, mConnectionCtx);
         };
 
+        mClient.connectedToServer = [&](std::shared_ptr<SRTNet::NetworkConnection>& ctx,
+                SRTSOCKET newSocket,
+                const SRTNet::ConnectionInformation& connectionInformation) {
+            EXPECT_NE(connectionInformation.mPeerSrtVersion, SRTNet::ConnectionInformation().mPeerSrtVersion);
+            EXPECT_NE(connectionInformation.mNegotiatedLatency, SRTNet::ConnectionInformation().mNegotiatedLatency);
+        };
     }
 
     bool waitForClientToConnect(std::chrono::seconds timeout) {