Create net::FormatOriginForDisplay helper function

net/base/net_util_icu_unittest.cc

 // Copyright 2014 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 "net/base/net_util.h"
 
 #include <string.h>
 
 #include <vector>
 
@@ skipping 345 matching lines @@
 struct UrlTestData {
   const char* description;
   const char* input;
   const char* languages;
   FormatUrlTypes format_types;
   UnescapeRule::Type escape_rules;
   const wchar_t* output;  // Use |wchar_t| to handle Unicode constants easily.
   size_t prefix_len;
 };
 
+struct OriginTestData {
+  const char* description;
+  const char* input;
+  const char* languages;
+  const bool always_show_scheme;
+  const wchar_t* output;
+};
+
 // A helper for IDN*{Fast,Slow}.
 // Append "::<language list>" to |expected| and |actual| to make it
 // easy to tell which sub-case fails without debugging.
 void AppendLanguagesToOutputs(const char* languages,
                               base::string16* expected,
                               base::string16* actual) {
   base::string16 to_append = ASCIIToUTF16("::") + ASCIIToUTF16(languages);
   expected->append(to_append);
   actual->append(to_append);
 }
@@ skipping 685 matching lines @@
                        UnescapeRule::NORMAL, omit_http_start_with_ftp_offsets);
 
   const size_t omit_all_offsets[] = {
     0, kNpos, kNpos, kNpos, kNpos, kNpos, kNpos, 0, kNpos, kNpos, kNpos, kNpos,
     0, 1, 2, 3, 4, 5, 6, 7
   };
   CheckAdjustedOffsets("http://user@foo.com/", "en", kFormatUrlOmitAll,
                        UnescapeRule::NORMAL, omit_all_offsets);
 }
 
+TEST(NetUtilTest, FormatOriginForDisplay) {
+  const OriginTestData tests[] = {
+    {"Empty URL", "", "", false, L""},
+    {"HTTP URL, forcing bool",
+     "http://www.google.com/", "", true, L"http://www.google.com"},
+    {"HTTP URL, no forcing bool",
+     "http://www.google.com/", "", false, L"http://www.google.com"},
+    {"HTTPS URL, forcing bool",
+     "https://www.google.com/", "", true, L"https://www.google.com"},
+    {"HTTPS URL, no forcing bool",
+     "https://www.google.com/", "", false, L"www.google.com"},
+    {"Standard HTTP port",
+     "http://www.google.com:80/", "", false, L"http://www.google.com"},
+    {"Standard HTTPS port",
+     "https://www.google.com:443/", "", false, L"www.google.com"},
+    {"Non-standard HTTP port",
+     "http://www.google.com:9000/", "", false, L"http://www.google.com:9000"},
+    {"Non-standard HTTPS port",
+     "https://www.google.com:9000/", "", false, L"www.google.com:9000"},
+    {"File URI, forcing bool",
+     "file://usr/example/file.html", "", true, L"file://usr/example/file.html"},
+    {"File URI, no forcing bool",
+     "file://usr/example/file.html", "", true, L"file://usr/example/file.html"},
+    {"HTTP URL with path",
+     "http://www.google.com/test.html", "", false, L"http://www.google.com"},
+    {"HTTPS URL with path",
+     "https://www.google.com/test.html", "", false, L"www.google.com"},
+    {"Unusual secure scheme (wss)",
+     "wss://www.google.com/", "", false, L"www.google.com"},
+    {"Unusual non-secure scheme (gopher)",
+     "gopher://www.google.com/", "", false, L"gopher://www.google.com"},
+    {"Unlisted scheme (chrome)",
+     "chrome://version", "", false, L"chrome://version"},
+    {"HTTP IP address",
+     "http://173.194.65.103", "", false, L"http://173.194.65.103"},
+    {"HTTPS IP address, no forcing bool",
+     "https://173.194.65.103", "", false, L"173.194.65.103"},
+    {"HTTPS IP address, forcing bool",
+     "https://173.194.65.103", "", true, L"https://173.194.65.103"},
+  };
+  for (size_t i = 0; i < arraysize(tests); ++i) {
+    base::string16 formatted = FormatOriginForDisplay(
+        GURL(tests[i].input), tests[i].languages, tests[i].always_show_scheme);
+    EXPECT_EQ(WideToUTF16(tests[i].output), formatted) << tests[i].description;
+  }
+}
+
 }  // namespace net