blob: a35a30c90e3b765a3f2d8e1a93e057c4676e2ffd [file] [log] [blame]
// 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 <memory>
#include "base/memory/ref_counted.h"
#include "base/test/scoped_task_environment.h"
#include "base/test/test_suite.h"
#include "build/build_config.h"
#include "net/dns/mock_host_resolver.h"
namespace base {
namespace test {
class ScopedTaskEnvironment;
namespace net {
class NetworkChangeNotifier;
class NetTestSuite : public base::TestSuite {
NetTestSuite(int argc, char** argv);
~NetTestSuite() override;
void Initialize() override;
void Shutdown() override;
// Returns the base::test::ScopedTaskEnvironment initialized by the current
// NetTestSuite.
static base::test::ScopedTaskEnvironment* GetScopedTaskEnvironment();
// Sets the global ScopedTaskEnvironment with a new environment of main thread
// type, |type|. For example, one might want to use a MOCK_TIME
// ScopedTaskEnvironment.
static void SetScopedTaskEnvironment(
base::test::ScopedTaskEnvironment::MainThreadType type);
// Sets the global ScopedTaskEnvironment to a new environment of the default
// type used by NetTestSuite.
static void ResetScopedTaskEnvironment();
// 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();
std::unique_ptr<net::NetworkChangeNotifier> network_change_notifier_;
std::unique_ptr<base::test::ScopedTaskEnvironment> scoped_task_environment_;
scoped_refptr<net::RuleBasedHostResolverProc> host_resolver_proc_;
net::ScopedDefaultHostResolverProc scoped_host_resolver_proc_;