| Index: net/base/net_test_suite.h
|
| diff --git a/net/base/net_test_suite.h b/net/base/net_test_suite.h
|
| index 8ebc4e79beb71de22713c1ad74b7e982e74b7b28..274d0670c9fe5226eb8c2e97983b497973b34d2e 100644
|
| --- a/net/base/net_test_suite.h
|
| +++ b/net/base/net_test_suite.h
|
| @@ -6,53 +6,33 @@
|
| #define NET_BASE_NET_TEST_SUITE_H_
|
| #pragma once
|
|
|
| -#include "base/message_loop.h"
|
| #include "base/ref_counted.h"
|
| #include "base/test/test_suite.h"
|
| #include "build/build_config.h"
|
| #include "net/base/mock_host_resolver.h"
|
| -#if defined(USE_NSS)
|
| -#include "net/ocsp/nss_ocsp.h"
|
| -#endif
|
| +
|
| +class MessageLoop;
|
| +
|
| +namespace net {
|
| +class NetworkChangeNotifier;
|
| +}
|
|
|
| class NetTestSuite : public base::TestSuite {
|
| public:
|
| - NetTestSuite(int argc, char** argv) : TestSuite(argc, argv) {
|
| - }
|
| + NetTestSuite(int argc, char** argv);
|
| + virtual ~NetTestSuite();
|
| +
|
| + virtual void Initialize();
|
|
|
| - virtual void Initialize() {
|
| - TestSuite::Initialize();
|
| - InitializeTestThread();
|
| - }
|
| + virtual void Shutdown();
|
| +
|
| + protected:
|
|
|
| // Called from within Initialize(), but separate so that derived classes
|
| // can initialize the NetTestSuite instance only and not
|
| // TestSuite::Initialize(). TestSuite::Initialize() performs some global
|
| // initialization that can only be done once.
|
| - void 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());
|
| - }
|
| -
|
| - virtual void 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 InitializeTestThread();
|
|
|
| private:
|
| scoped_ptr<net::NetworkChangeNotifier> network_change_notifier_;
|
|
|