From 758a425bbc2e40cf789c521024533574ca1770e5 Mon Sep 17 00:00:00 2001
From: Kevin Krakauer <krakauer@google.com>
Date: Tue, 14 Jun 2022 11:58:26 -0700
Subject: [PATCH] shrink DataTransferStressTest

PiperOrigin-RevId: 454921228
---
 test/syscalls/linux/socket_generic_stress.cc | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test/syscalls/linux/socket_generic_stress.cc b/test/syscalls/linux/socket_generic_stress.cc
index c08e4d99e..51901a3a0 100644
--- a/test/syscalls/linux/socket_generic_stress.cc
+++ b/test/syscalls/linux/socket_generic_stress.cc
@@ -193,9 +193,6 @@ INSTANTIATE_TEST_SUITE_P(
 using DataTransferStressTest = SocketPairTest;
 
 TEST_P(DataTransferStressTest, BigDataTransfer) {
-  // TODO(b/165912341): These are too slow on KVM platform with nested virt.
-  SKIP_IF(GvisorPlatform() == Platform::kKVM);
-
   const std::unique_ptr<SocketPair> sockets =
       ASSERT_NO_ERRNO_AND_VALUE(NewSocketPair());
   int client_fd = sockets->first_fd();
@@ -218,9 +215,12 @@ TEST_P(DataTransferStressTest, BigDataTransfer) {
     ASSERT_THAT(shutdown(server_fd, SHUT_WR), SyscallSucceeds());
   });
 
+  // Tests can be prohibitively slow on the KVM platform with nested virt.
+  const int kShift = GvisorPlatform() == Platform::kKVM ? 10 : 20;
+
   const std::string chunk = "Though this upload be but little, it is fierce.";
   std::string big_string;
-  while (big_string.size() < 31 << 20) {
+  while (big_string.size() < 31 << kShift) {
     big_string += chunk;
   }
   absl::string_view data = big_string;
@@ -236,7 +236,7 @@ TEST_P(DataTransferStressTest, BigDataTransfer) {
   });
 
   std::string buf;
-  buf.resize(1 << 20);
+  buf.resize(1 << kShift);
   while (!data.empty()) {
     ssize_t n = read(client_fd, buf.data(), buf.size());
     ASSERT_GE(n, 0);