Adjust timeouts on DnsTransactionTest for valgrind

net/dns/dns_transaction_unittest.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/dns/dns_transaction.h"
 
 #include <deque>
 #include <vector>
 
 #include "base/bind.h"
+#include "base/test/test_timeouts.h"
 #include "base/time.h"
 #include "net/dns/dns_protocol.h"
 #include "net/dns/dns_query.h"
 #include "net/dns/dns_response.h"
 #include "net/dns/dns_session.h"
 #include "net/dns/dns_test_util.h"
 #include "net/socket/socket_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 namespace {
 
 // A mock for RandIntCallback that always returns 0.
 int ReturnZero(int min, int max) {
   return 0;
 }
 
 class DnsTransactionTest : public testing::Test {
  protected:
   virtual void SetUp() OVERRIDE {
-    DnsConfig config;
+    qname_ = std::string(kT0DnsName, arraysize(kT0DnsName));
+    // Use long timeout to prevent timing out on slow bots.
+    ConfigureSession(base::TimeDelta::FromMilliseconds(
+        TestTimeouts::action_timeout_ms()));
+    callback_ = base::Bind(&DnsTransactionTest::OnTransactionComplete,
+                           base::Unretained(this));
+  }
 
+  void ConfigureSession(const base::TimeDelta& timeout) {
     IPEndPoint dns_server;
     bool rv = CreateDnsAddress(kDnsIp, kDnsPort, &dns_server);
     ASSERT_TRUE(rv);
+
+    DnsConfig config;
     config.nameservers.push_back(dns_server);
     config.attempts = 3;
-    config.timeout = base::TimeDelta::FromMilliseconds(20);
-
-    qname_ = std::string(kT0DnsName, arraysize(kT0DnsName));
+    config.timeout = timeout;
 
     session_ = new DnsSession(config,
                               new MockClientSocketFactory(),
                               base::Bind(&ReturnZero),
                               NULL /* NetLog */);
-
-    callback_ = base::Bind(&DnsTransactionTest::OnTransactionComplete,
-                           base::Unretained(this));
   }
 
   void StartTransaction() {
     transaction_.reset(new DnsTransaction(session_.get(),
                                           qname_,
                                           kT0Qtype,
                                           callback_,
                                           BoundNetLog()));
 
     int rv0 = transaction_->Start();
@@ skipping 84 matching lines @@
     MockRead(true, reinterpret_cast<const char*>(kT0ResponseDatagram),
              arraysize(kT0ResponseDatagram))
   };
 
   scoped_refptr<DelayedSocketData> socket0_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
   scoped_refptr<DelayedSocketData> socket1_data(
       new DelayedSocketData(0, reads0, arraysize(reads0),
                             writes0, arraysize(writes0)));
 
+  // Use short timeout to speed up the test.
+  ConfigureSession(base::TimeDelta::FromMilliseconds(
+      TestTimeouts::tiny_timeout_ms()));
   factory().AddSocketDataProvider(socket0_data.get());
   factory().AddSocketDataProvider(socket1_data.get());
 
   StartTransaction();
 
   MessageLoop::current()->Run();
 
   EXPECT_EQ(OK, rv());
 
   EXPECT_TRUE(socket0_data->at_read_eof());
@@ skipping 18 matching lines @@
   };
 
   scoped_refptr<DelayedSocketData> socket0_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
   scoped_refptr<DelayedSocketData> socket1_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
   scoped_refptr<DelayedSocketData> socket2_data(
       new DelayedSocketData(0, reads0, arraysize(reads0),
                             writes0, arraysize(writes0)));
 
+  // Use short timeout to speed up the test.
+  ConfigureSession(base::TimeDelta::FromMilliseconds(
+      TestTimeouts::tiny_timeout_ms()));
   factory().AddSocketDataProvider(socket0_data.get());
   factory().AddSocketDataProvider(socket1_data.get());
   factory().AddSocketDataProvider(socket2_data.get());
 
   StartTransaction();
 
   MessageLoop::current()->Run();
 
   EXPECT_EQ(OK, rv());
 
@@ skipping 15 matching lines @@
               arraysize(kT0QueryDatagram))
   };
 
   scoped_refptr<DelayedSocketData> socket0_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
   scoped_refptr<DelayedSocketData> socket1_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
   scoped_refptr<DelayedSocketData> socket2_data(
       new DelayedSocketData(2, NULL, 0, writes0, arraysize(writes0)));
 
+  // Use short timeout to speed up the test.
+  ConfigureSession(base::TimeDelta::FromMilliseconds(
+      TestTimeouts::tiny_timeout_ms()));
   factory().AddSocketDataProvider(socket0_data.get());
   factory().AddSocketDataProvider(socket1_data.get());
   factory().AddSocketDataProvider(socket2_data.get());
 
   StartTransaction();
 
   MessageLoop::current()->Run();
 
   EXPECT_EQ(ERR_DNS_TIMED_OUT, rv());
 
   EXPECT_TRUE(socket0_data->at_read_eof());
   EXPECT_TRUE(socket0_data->at_write_eof());
   EXPECT_TRUE(socket1_data->at_read_eof());
   EXPECT_TRUE(socket1_data->at_write_eof());
   EXPECT_TRUE(socket2_data->at_read_eof());
   EXPECT_TRUE(socket2_data->at_write_eof());
   EXPECT_EQ(3u, factory().udp_client_sockets().size());
 }
 
 }  // namespace
 
 }  // namespace net