Prevent making real DNS lookups by chrome tests.

chrome/browser/extensions/extension_view_unittest.cc

 // Copyright (c) 2006-2008 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 "base/message_loop.h"
+#include "base/ref_counted.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/extensions/extension_error_reporter.h"
 #include "chrome/browser/extensions/extension_view.h"
 #include "chrome/browser/extensions/extensions_service.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/test/in_process_browser_test.h"
 #include "chrome/test/ui_test_utils.h"
+#include "net/base/host_resolver_unittest.h"
 
 namespace {
 
 // How long to wait for the extension to put up a javascript alert before giving
 // up.
 const int kAlertTimeoutMs = 20000;
 
 // How long to wait for the extension to load before giving up.
 const int kLoadTimeoutMs = 10000;
 
@@ skipping 66 matching lines @@
   }
 
   NotificationRegistrar registrar_;
   Extension* extension_;
 };
 
 }  // namespace
 
 class ExtensionViewTest : public InProcessBrowserTest {
  public:
+  ExtensionViewTest() {
+    host_mapper_ = new net::RuleBasedHostMapper();
+    // TODO(aa): Don't make a real dns lookup here or simulate a failing lookup.
+    // But if it's really needed then remove the TODO.
+    host_mapper_->AllowDirectLookup("*.google.com");
+    scoped_host_mapper_.Init(host_mapper_.get());
+  }
+
   virtual void SetUp() {
     // Initialize the error reporter here, otherwise BrowserMain will create it
     // with the wrong MessageLoop.
     ExtensionErrorReporter::Init(false);
 
     // Use single-process in an attempt to speed it up and make it less flaky.
     //EnableSingleProcess();
 
     InProcessBrowserTest::SetUp();
   }
+
+ private:
+  scoped_refptr<net::RuleBasedHostMapper> host_mapper_;
+  net::ScopedHostMapper scoped_host_mapper_;
 };
 
 // Tests that ExtensionView starts an extension process and runs the script
 // contained in the extension's "index.html" file.
 IN_PROC_BROWSER_TEST_F(ExtensionViewTest, Index) {
   // Create an observer first to be sure we have the notification registered
   // before it's sent.
   ExtensionLoadedObserver observer;
 
   // Get the path to our extension.
@@ skipping 10 matching lines @@
 
   // Now wait for it to load, and grab a pointer to it.
   Extension* extension = observer.WaitForExtension();
   ASSERT_TRUE(extension);
   GURL url = Extension::GetResourceURL(extension->url(), "toolstrip1.html");
 
   // Start the extension process and wait for it to show a javascript alert.
   MockExtensionView view(extension, url, profile);
   EXPECT_TRUE(view.got_message());
 }