Show IDN in error screens (DNS failure etc.)

chrome/browser/errorpage_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 "base/bind.h"
+#include "base/prefs/pref_service.h"
+#include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/google/google_util.h"
 #include "chrome/browser/net/url_request_mock_util.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_navigation_observer.h"
 #include "content/test/net/url_request_failed_job.h"
 #include "content/test/net/url_request_mock_http_job.h"
 #include "net/base/net_errors.h"
+#include "net/base/net_util.h"
 #include "net/url_request/url_request_filter.h"
 #include "net/url_request/url_request_job_factory.h"
 
 using content::BrowserThread;
 using content::NavigationController;
 using content::URLRequestFailedJob;
 
 namespace {
 
 class ErrorPageTest : public InProcessBrowserTest {
@@ skipping 211 matching lines @@
 
 // Checks that the Link Doctor is not loaded when we receive an actual 404 page.
 IN_PROC_BROWSER_TEST_F(ErrorPageTest, Page404) {
   NavigateToURLAndWaitForTitle(
       content::URLRequestMockHTTPJob::GetMockUrl(
           base::FilePath(FILE_PATH_LITERAL("page404.html"))),
       "SUCCESS",
       1);
 }
 
+// Returns JavaScrips code that executes plain text search for the page.

[mmenke, 2013/09/27 14:44:05]

nit:  Javascript

[yuusuke, 2013/09/28 10:42:50]

Done.

+// Pass into content::ExecuteScriptAndExtractBool as |script| parameter.
+std::string GetTextContentContainsStringScript(
+    const std::string& value_to_search) {
+  return base::StringPrintf(
+      "var textContent = document.body.textContent;"
+      "var hasError = textContent.indexOf('%s') >= 0;"
+      "domAutomationController.send(hasError);", value_to_search.c_str());

[mmenke, 2013/09/27 14:44:05]

nit:  To make the second parameter stand out a bit more, suggest putting it on
its own line.

[yuusuke, 2013/09/28 10:42:50]

Done.

+}
+
 // Protocol handler that fails all requests with net::ERR_ADDRESS_UNREACHABLE.
 class AddressUnreachableProtocolHandler
     : public net::URLRequestJobFactory::ProtocolHandler {
  public:
   AddressUnreachableProtocolHandler() {}
   virtual ~AddressUnreachableProtocolHandler() {}
 
   // net::URLRequestJobFactory::ProtocolHandler:
   virtual net::URLRequestJob* MaybeCreateJob(
       net::URLRequest* request,
@@ skipping 55 matching lines @@
   ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(
       browser(),
       URLRequestFailedJob::GetMockHttpUrl(net::ERR_NAME_NOT_RESOLVED),
       2);
 
   // Verify that the expected error page is being displayed.  Do this by making
   // sure the original error code (ERR_NAME_NOT_RESOLVED) is displayed.
   bool result = false;
   EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
       browser()->tab_strip_model()->GetActiveWebContents(),
-      "var textContent = document.body.textContent;"
-      "var hasError = textContent.indexOf('ERR_NAME_NOT_RESOLVED') >= 0;"
-      "domAutomationController.send(hasError);",
+      GetTextContentContainsStringScript("ERR_NAME_NOT_RESOLVED"),
+      &result));
+  EXPECT_TRUE(result);
+}
+
+// A test fixture that substitutes kMockHostname with IDN.

[mmenke, 2013/09/27 14:44:05]

nit:  Think it's ambiguous what "kMockHostname" is.  Maybe "A test fixture that
simulates failing requests for an IDN domain name."?

[yuusuke, 2013/09/28 10:42:50]

Done.

+class ErrorPageForIDNTest : public InProcessBrowserTest {
+ public:
+  // InProcessBrowserTest:
+  virtual void SetUpOnMainThread() OVERRIDE {
+    // Clear AcceptLanguages to force punycode decoding.
+    browser()->profile()->GetPrefs()->SetString(prefs::kAcceptLanguages,
+                                                std::string());
+    BrowserThread::PostTask(
+        BrowserThread::IO, FROM_HERE,
+        base::Bind(&ErrorPageForIDNTest::AddFilters));
+  }
+
+  virtual void CleanUpOnMainThread() OVERRIDE {
+    BrowserThread::PostTask(
+        BrowserThread::IO, FROM_HERE,
+        base::Bind(&ErrorPageForIDNTest::RemoveFilters));
+  }
+
+  // Target hostname in ASCII (punycode).
+  static std::string GetHostname() {
+    return "xn--d1abbgf6aiiy.xn--p1ai";
+  }
+
+  static std::string GetHostnameAsJSUnicode() {
+    return "\\u043f\\u0440\\u0435\\u0437\\u0438\\u0434\\u0435\\u043d"
+           "\\u0442.\\u0440\\u0444";
+  }

[mmenke, 2013/09/27 14:44:05]

optional nit:  Suggest just making these consts, it's the much more common style
in Chrome.  Just something like:

static const char kHostnamePunycode[] = "xn--d1abbgf6aiiy.xn--p1ai";
static const char kHostnameJSUnicode[] = ....

[yuusuke, 2013/09/28 10:42:50]

Done.

+
+ private:
+  static void AddFilters() {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    content::URLRequestFailedJob::AddUrlHandler(GetHostname());
+  }
+
+  static void RemoveFilters() {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+    net::URLRequestFilter::GetInstance()->ClearHandlers();
+  }
+};
+
+// Make sure error page shows correct unicode for IDN.
+IN_PROC_BROWSER_TEST_F(ErrorPageForIDNTest, IDN) {
+  ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(

[mmenke, 2013/09/27 14:44:05]

Think it's worth a comment that ERR_UNSAFE_PORT will not trigger the link
doctor.

[yuusuke, 2013/09/28 10:42:50]

Done.

+      browser(),
+      URLRequestFailedJob::GetMockHttpUrl(net::ERR_UNSAFE_PORT,
+                                          GetHostname()),
+      1);
+
+  bool result = false;
+  EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
+      browser()->tab_strip_model()->GetActiveWebContents(),
+      GetTextContentContainsStringScript(GetHostnameAsJSUnicode()),
       &result));
   EXPECT_TRUE(result);
 }
 
 }  // namespace