Index: net/base/net_test_suite.cc |
diff --git a/net/base/net_test_suite.cc b/net/base/net_test_suite.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a37bc5b283066a863ce5bfa1ff67ffa9061a18d2 |
--- /dev/null |
+++ b/net/base/net_test_suite.cc |
@@ -0,0 +1,46 @@ |
+// 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/base/net_test_suite.h" |
+ |
+#include "base/message_loop.h" |
+#if defined(USE_NSS) |
+#include "net/ocsp/nss_ocsp.h" |
+#endif |
+ |
+NetTestSuite::NetTestSuite(int argc, char** argv) |
+ : TestSuite(argc, argv) { |
+} |
+ |
+NetTestSuite::~NetTestSuite() {} |
+ |
+void NetTestSuite::Initialize() { |
+ TestSuite::Initialize(); |
+ InitializeTestThread(); |
+} |
+ |
+void NetTestSuite::Shutdown() { |
+#if defined(USE_NSS) |
+ net::ShutdownOCSP(); |
+#endif |
+ |
+ // We want to destroy this here before the TestSuite continues to tear down |
+ // the environment. |
+ message_loop_.reset(); |
+ |
+ TestSuite::Shutdown(); |
+} |
+ |
+void NetTestSuite::InitializeTestThread() { |
+ network_change_notifier_.reset(net::NetworkChangeNotifier::CreateMock()); |
+ |
+ host_resolver_proc_ = new net::RuleBasedHostResolverProc(NULL); |
+ scoped_host_resolver_proc_.Init(host_resolver_proc_.get()); |
+ // In case any attempts are made to resolve host names, force them all to |
+ // be mapped to localhost. This prevents DNS queries from being sent in |
+ // the process of running these unit tests. |
+ host_resolver_proc_->AddRule("*", "127.0.0.1"); |
+ |
+ message_loop_.reset(new MessageLoopForIO()); |
+} |