OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 #include "net/base/network_change_notifier.h" | |
9 #include "net/http/http_stream_factory.h" | |
10 #include "net/spdy/spdy_session.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 #if defined(USE_NSS) || defined(OS_IOS) | |
14 #include "net/ocsp/nss_ocsp.h" | |
15 #endif | |
16 | |
17 class StaticReset : public ::testing::EmptyTestEventListener { | |
18 virtual void OnTestStart(const ::testing::TestInfo& test_info) OVERRIDE { | |
19 net::HttpStreamFactory::ResetStaticSettingsToInit(); | |
20 } | |
21 }; | |
22 | |
23 NetTestSuite::NetTestSuite(int argc, char** argv) | |
24 : TestSuite(argc, argv) { | |
25 } | |
26 | |
27 NetTestSuite::NetTestSuite(int argc, char** argv, | |
28 bool create_at_exit_manager) | |
29 : TestSuite(argc, argv, create_at_exit_manager) { | |
30 } | |
31 | |
32 NetTestSuite::~NetTestSuite() {} | |
33 | |
34 void NetTestSuite::Initialize() { | |
35 TestSuite::Initialize(); | |
36 ::testing::UnitTest::GetInstance()->listeners().Append(new StaticReset()); | |
37 InitializeTestThread(); | |
38 } | |
39 | |
40 void NetTestSuite::Shutdown() { | |
41 #if defined(USE_NSS) || defined(OS_IOS) | |
42 net::ShutdownNSSHttpIO(); | |
43 #endif | |
44 | |
45 // We want to destroy this here before the TestSuite continues to tear down | |
46 // the environment. | |
47 message_loop_.reset(); | |
48 | |
49 TestSuite::Shutdown(); | |
50 } | |
51 | |
52 void NetTestSuite::InitializeTestThread() { | |
53 network_change_notifier_.reset(net::NetworkChangeNotifier::CreateMock()); | |
54 | |
55 InitializeTestThreadNoNetworkChangeNotifier(); | |
56 } | |
57 | |
58 void NetTestSuite::InitializeTestThreadNoNetworkChangeNotifier() { | |
59 host_resolver_proc_ = new net::RuleBasedHostResolverProc(NULL); | |
60 scoped_host_resolver_proc_.Init(host_resolver_proc_.get()); | |
61 // In case any attempts are made to resolve host names, force them all to | |
62 // be mapped to localhost. This prevents DNS queries from being sent in | |
63 // the process of running these unit tests. | |
64 host_resolver_proc_->AddRule("*", "127.0.0.1"); | |
65 | |
66 message_loop_.reset(new MessageLoopForIO()); | |
67 } | |
OLD | NEW |