| Index: chrome/test/unit/chrome_test_suite.h
|
| ===================================================================
|
| --- chrome/test/unit/chrome_test_suite.h (revision 20760)
|
| +++ chrome/test/unit/chrome_test_suite.h (working copy)
|
| @@ -28,15 +28,17 @@
|
| #include "chrome/common/mac_app_names.h"
|
| #endif
|
| #include "chrome/test/testing_browser_process.h"
|
| -#include "net/base/host_resolver_unittest.h"
|
| +#include "net/base/mock_host_resolver.h"
|
| #include "net/base/net_util.h"
|
|
|
| // In many cases it may be not obvious that a test makes a real DNS lookup.
|
| // We generally don't want to rely on external DNS servers for our tests,
|
| -// so this mapper catches external queries.
|
| -class WarningHostMapper : public net::HostMapper {
|
| +// so this host resolver procedure catches external queries.
|
| +class WarningHostResolverProc : public net::HostResolverProc {
|
| public:
|
| - virtual std::string Map(const std::string& host) {
|
| + WarningHostResolverProc() : HostResolverProc(NULL) {}
|
| +
|
| + virtual int Resolve(const std::string& host, net::AddressList* addrlist) {
|
| const char* kLocalHostNames[] = {"localhost", "127.0.0.1"};
|
| bool local = false;
|
|
|
| @@ -51,11 +53,11 @@
|
| }
|
|
|
| // Make the test fail so it's harder to ignore.
|
| - // If you really need to make real DNS query, use net::RuleBasedHostMapper
|
| - // and its AllowDirectLookup method.
|
| + // If you really need to make real DNS query, use
|
| + // net::RuleBasedHostResolverProc and its AllowDirectLookup method.
|
| EXPECT_TRUE(local) << "Making external DNS lookup of " << host;
|
|
|
| - return MapUsingPrevious(host);
|
| + return ResolveUsingPrevious(host, addrlist);
|
| }
|
| };
|
|
|
| @@ -71,8 +73,8 @@
|
|
|
| TestSuite::Initialize();
|
|
|
| - host_mapper_ = new WarningHostMapper();
|
| - scoped_host_mapper_.Init(host_mapper_.get());
|
| + host_resolver_proc_ = new WarningHostResolverProc();
|
| + scoped_host_resolver_proc_.Init(host_resolver_proc_.get());
|
|
|
| chrome::RegisterPathProvider();
|
| app::RegisterPathProvider();
|
| @@ -140,8 +142,8 @@
|
|
|
| StatsTable* stats_table_;
|
| ScopedOleInitializer ole_initializer_;
|
| - scoped_refptr<WarningHostMapper> host_mapper_;
|
| - net::ScopedHostMapper scoped_host_mapper_;
|
| + scoped_refptr<WarningHostResolverProc> host_resolver_proc_;
|
| + net::ScopedDefaultHostResolverProc scoped_host_resolver_proc_;
|
| };
|
|
|
| #endif // CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_
|
|
|