Add more data to net-internals view of DNS cache

chrome/browser/ui/webui/net_internals/net_internals_ui_browsertest.cc

 // Copyright (c) 2012 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 "chrome/browser/ui/webui/net_internals/net_internals_ui_browsertest.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 17 matching lines @@
 #include "chrome/common/channel_info.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/net_log/chrome_net_log.h"
 #include "components/prefs/pref_service.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui_message_handler.h"
 #include "net/base/address_list.h"
 #include "net/base/net_errors.h"
+#include "net/base/network_change_notifier.h"
 #include "net/dns/host_cache.h"
 #include "net/dns/host_resolver.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/http/http_network_session.h"
 #include "net/http/http_transaction_factory.h"
 #include "net/log/net_log.h"
 #include "net/log/net_log_event_type.h"
 #include "net/log/net_log_source_type.h"
 #include "net/log/net_log_with_source.h"
 #include "net/log/write_to_file_net_log_observer.h"
@@ skipping 62 matching lines @@
   // which must already have been started.
   void GetTestServerURL(const base::ListValue* list_value);
 
   // Called on UI thread.  Adds an entry to the cache for the specified
   // hostname by posting a task to the IO thread.  Takes the host name,
   // ip address, net error code, and expiration time in days from now
   // as parameters.  If the error code indicates failure, the ip address
   // must be an empty string.
   void AddCacheEntry(const base::ListValue* list_value);
 
+  // Simulates a network change.
+  void ChangeNetwork(const base::ListValue* list_value);
+
   // Opens the given URL in a new tab.
   void LoadPage(const base::ListValue* list_value);
 
   // Opens a page in a new tab that prerenders the given URL.
   void PrerenderPage(const base::ListValue* list_value);
 
   // Navigates to the prerender in the background tab. This assumes that
   // there is a "Click()" function in the background tab which will navigate
   // there, and that the background tab exists at slot 1.
   void NavigateToPrerender(const base::ListValue* list_value);
@@ skipping 27 matching lines @@
       incognito_browser_(NULL) {
 }
 
 void NetInternalsTest::MessageHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("getTestServerURL",
       base::Bind(&NetInternalsTest::MessageHandler::GetTestServerURL,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("addCacheEntry",
       base::Bind(&NetInternalsTest::MessageHandler::AddCacheEntry,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback(
+      "changeNetwork",
+      base::Bind(&NetInternalsTest::MessageHandler::ChangeNetwork,
+                 base::Unretained(this)));
   web_ui()->RegisterMessageCallback("loadPage",
       base::Bind(&NetInternalsTest::MessageHandler::LoadPage,
                   base::Unretained(this)));
   web_ui()->RegisterMessageCallback("prerenderPage",
       base::Bind(&NetInternalsTest::MessageHandler::PrerenderPage,
                   base::Unretained(this)));
   web_ui()->RegisterMessageCallback("navigateToPrerender",
       base::Bind(&NetInternalsTest::MessageHandler::NavigateToPrerender,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("createIncognitoBrowser",
@@ skipping 40 matching lines @@
   ASSERT_TRUE(browser());
 
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(&AddCacheEntryOnIOThread,
                  base::RetainedRef(browser()->profile()->GetRequestContext()),
                  hostname, ip_literal, static_cast<int>(net_error),
                  static_cast<int>(expire_days_from_now)));
 }
 
+void NetInternalsTest::MessageHandler::ChangeNetwork(
+    const base::ListValue* list_value) {
+  net::NetworkChangeNotifier::NotifyObserversOfIPAddressChangeForTests();
+}
+
 void NetInternalsTest::MessageHandler::LoadPage(
     const base::ListValue* list_value) {
   std::string url;
   ASSERT_TRUE(list_value->GetString(0, &url));
   LOG(WARNING) << "url: [" << url << "]";
   ui_test_utils::NavigateToURLWithDisposition(
       browser(), GURL(url), WindowOpenDisposition::NEW_BACKGROUND_TAB,
       ui_test_utils::BROWSER_TEST_NONE);
 }
 
@@ skipping 126 matching lines @@
   if (test_server_started_)
     return true;
   test_server_started_ = embedded_test_server()->Start();
 
   // Sample domain for SDCH-view test. Dictionaries for localhost/127.0.0.1
   // are forbidden.
   host_resolver()->AddRule("testdomain.com", "127.0.0.1");
   host_resolver()->AddRule("sub.testdomain.com", "127.0.0.1");
   return test_server_started_;
 }