Add tests for UDPSocket copy constructor.

Add more explicit assertions for TCPSocket copy constructor too.
Kilemonn · Jul 8, 2024 · cfeb031 · cfeb031
1 parent 3e01d03
commit cfeb031
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/tests/socket/TCPSocketTest.cpp b/tests/socket/TCPSocketTest.cpp
@@ -77,6 +77,13 @@ namespace kt
     {
         TCPSocket server = serverSocket.acceptTCPConnection();
         TCPSocket copiedSocket(socket);
+
+        ASSERT_EQ(socket.getHostname(), copiedSocket.getHostname());
+        ASSERT_EQ(socket.getPort(), copiedSocket.getPort());
+        ASSERT_EQ(socket.getInternetProtocolVersion(), copiedSocket.getInternetProtocolVersion());
+        kt::SocketAddress initialAddress = socket.getSocketAddress();
+        kt::SocketAddress copiedAddress = copiedSocket.getSocketAddress();
+        ASSERT_EQ(0, std::memcmp(&initialAddress, &copiedAddress, sizeof(initialAddress)));
 
         const std::string testString = "Test";
         ASSERT_TRUE(copiedSocket.send(testString));

diff --git a/tests/socket/UDPSocketTest.cpp b/tests/socket/UDPSocketTest.cpp
@@ -36,6 +36,20 @@ namespace kt
         ASSERT_EQ(kt::InternetProtocolVersion::Any, socket.getInternetProtocolVersion());
     }
 
+    TEST_F(UDPSocketTest, UDPCopyConstructors)
+    {
+        socket.bind();
+        UDPSocket copiedSocket(socket);
+
+        ASSERT_EQ(socket.isUdpBound(), copiedSocket.isUdpBound());
+        ASSERT_NE(std::nullopt, socket.getListeningPort());
+        ASSERT_NE(std::nullopt, copiedSocket.getListeningPort());
+        ASSERT_EQ(socket.getListeningPort().value(), copiedSocket.getListeningPort().value());
+        ASSERT_EQ(socket.getInternetProtocolVersion(), copiedSocket.getInternetProtocolVersion());
+
+        copiedSocket.close();
+    }
+
     /*
      * Ensure that multiple calls to UDPSocket.bind() fails if another socket is already bound to that port.
      */