| Index: net/dns/dns_transaction_unittest.cc
|
| diff --git a/net/dns/dns_transaction_unittest.cc b/net/dns/dns_transaction_unittest.cc
|
| index 46c68b3158fca1f53005193de27093d29eed2907..ac701d829cd5fb34317ffc55f1f7a50568f24314 100644
|
| --- a/net/dns/dns_transaction_unittest.cc
|
| +++ b/net/dns/dns_transaction_unittest.cc
|
| @@ -8,6 +8,7 @@
|
| #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"
|
| @@ -29,24 +30,28 @@ int ReturnZero(int min, int max) {
|
| 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() {
|
| @@ -151,6 +156,9 @@ TEST_F(DnsTransactionTest, FirstTimeoutTest) {
|
| 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());
|
|
|
| @@ -189,6 +197,9 @@ TEST_F(DnsTransactionTest, SecondTimeoutTest) {
|
| 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());
|
| @@ -224,6 +235,9 @@ TEST_F(DnsTransactionTest, ThirdTimeoutTest) {
|
| 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());
|
|
|