mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 14:26:09 +03:00
66 lines
2.1 KiB
C++
66 lines
2.1 KiB
C++
// 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 <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 {
|
|
public:
|
|
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);
|
|
|
|
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();
|
|
|
|
// 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:
|
|
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_;
|
|
};
|
|
|
|
#endif // NET_TEST_NET_TEST_SUITE_H_
|