• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "net/base/net_test_suite.h"
6 
7 #include "base/message_loop.h"
8 #if defined(USE_NSS)
9 #include "net/ocsp/nss_ocsp.h"
10 #endif
11 
NetTestSuite(int argc,char ** argv)12 NetTestSuite::NetTestSuite(int argc, char** argv)
13     : TestSuite(argc, argv) {
14 }
15 
~NetTestSuite()16 NetTestSuite::~NetTestSuite() {}
17 
Initialize()18 void NetTestSuite::Initialize() {
19   TestSuite::Initialize();
20   InitializeTestThread();
21 }
22 
Shutdown()23 void NetTestSuite::Shutdown() {
24 #if defined(USE_NSS)
25   net::ShutdownOCSP();
26 #endif
27 
28   // We want to destroy this here before the TestSuite continues to tear down
29   // the environment.
30   message_loop_.reset();
31 
32   TestSuite::Shutdown();
33 }
34 
InitializeTestThread()35 void NetTestSuite::InitializeTestThread() {
36   network_change_notifier_.reset(net::NetworkChangeNotifier::CreateMock());
37 
38   host_resolver_proc_ = new net::RuleBasedHostResolverProc(NULL);
39   scoped_host_resolver_proc_.Init(host_resolver_proc_.get());
40   // In case any attempts are made to resolve host names, force them all to
41   // be mapped to localhost.  This prevents DNS queries from being sent in
42   // the process of running these unit tests.
43   host_resolver_proc_->AddRule("*", "127.0.0.1");
44 
45   message_loop_.reset(new MessageLoopForIO());
46 }
47