Refs #20589: Readapt test

Signed-off-by: Jesus Perez <jesusperez@eprosima.com>
eProsima · Mar 5, 2024 · 371a955 · 371a955
1 parent 5aa3aca
commit 371a955
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/test/unittest/transport/TCPv4Tests.cpp b/test/unittest/transport/TCPv4Tests.cpp
@@ -1523,11 +1523,14 @@ TEST_F(TCPv4Tests, secure_non_blocking_send)
 
     // Send the message with no header. Since TCP actually allocates twice the size of the buffer requested
     // and we want to guarantee that the buffer might be full, we send the message more than twice.
+    size_t bytes_sent = 0;
     for (int i = 0; i < 5; i++)
     {
-        sender_channel_resource->send(nullptr, 0, data, size, ec);
+        bytes_sent += sender_channel_resource->send(nullptr, 0, data, size, ec);
     }
 
+    ASSERT_EQ(bytes_sent, size * 2);
+
     secure_socket->lowest_layer().close(ec);
 }
 #endif // ifndef _WIN32
@@ -2022,11 +2025,14 @@ TEST_F(TCPv4Tests, non_blocking_send)
 
     // Send the message with no header. Since TCP actually allocates twice the size of the buffer requested
     // and we want to guarantee that the buffer might be full, we send the message more than twice.
+    size_t bytes_sent = 0;
     for (int i = 0; i < 5; i++)
     {
-        sender_channel_resource->send(nullptr, 0, data, size, ec);
+        bytes_sent += sender_channel_resource->send(nullptr, 0, data, size, ec);
     }
 
+    ASSERT_EQ(bytes_sent, size * 2);
+
     socket.shutdown(asio::ip::tcp::socket::shutdown_both);
     socket.cancel();
     socket.close();

diff --git a/test/unittest/transport/TCPv6Tests.cpp b/test/unittest/transport/TCPv6Tests.cpp
@@ -373,11 +373,14 @@ TEST_F(TCPv6Tests, non_blocking_send)
 
     // Send the message with no header. Since TCP actually allocates twice the size of the buffer requested
     // and we want to guarantee that the buffer might be full, we send the message more than twice.
+    size_t bytes_sent = 0;
     for (int i = 0; i < 5; i++)
     {
-        sender_channel_resource->send(nullptr, 0, data, size, ec);
+        bytes_sent += sender_channel_resource->send(nullptr, 0, data, size, ec);
     }
 
+    ASSERT_EQ(bytes_sent, size * 2);
+
     socket.shutdown(asio::ip::tcp::socket::shutdown_both);
     socket.cancel();
     socket.close();