1 files changed, 53 insertions, 0 deletions
diff --git a/net/test/net_test_suite.h b/net/test/net_test_suite.h
new file mode 100644
index 0000000..b2fed70
--- /dev/null
+++ b/net/test/net_test_suite.h
@@ -0,0 +1,53 @@
+// 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.
+
+#ifndef NET_TEST_NET_TEST_SUITE_H_
+#define NET_TEST_NET_TEST_SUITE_H_
+
+#include "base/memory/ref_counted.h"
+#include "base/test/test_suite.h"
+#include "build/build_config.h"
+#include "net/base/mock_host_resolver.h"
+
+class MessageLoop;
+
+namespace net {
+class NetworkChangeNotifier;
+}
+
+class NetTestSuite : public base::TestSuite {
+ public:
+  NetTestSuite(int argc, char** argv);
+  virtual ~NetTestSuite();
+
+  virtual void Initialize() OVERRIDE;
+
+  virtual void Shutdown() OVERRIDE;
+
+ protected:
+  // This constructor is only accessible to specialized net test
+  // implementations which need to control the creation of an AtExitManager
+  // instance for the duration of the test.
+  NetTestSuite(int argc, char** argv, bool create_at_exit_manager);
+
+  // 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();
+
+  // Same as above, except it does not create a mock
+  // NetworkChangeNotifier.  Use this if your test needs to create and
+  // manage its own mock NetworkChangeNotifier, or if your test uses
+  // the production NetworkChangeNotifier.
+  void InitializeTestThreadNoNetworkChangeNotifier();
+
+ private:
+  scoped_ptr<net::NetworkChangeNotifier> network_change_notifier_;
+  scoped_ptr<MessageLoop> message_loop_;
+  scoped_refptr<net::RuleBasedHostResolverProc> host_resolver_proc_;
+  net::ScopedDefaultHostResolverProc scoped_host_resolver_proc_;
+};
+
+#endif  // NET_TEST_NET_TEST_SUITE_H_