Move DNS resolution from websocket-to-TCP proxy to ExtensionFunction

chrome/browser/chromeos/web_socket_proxy_helper_unittest.cc

+// Copyright (c) 2011 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 <vector>
+#include "base/string_number_conversions.h"
+#include "chrome/browser/chromeos/web_socket_proxy_helper.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chromeos {
+
+class WebSocketProxyHelperTest : public testing::Test {
+ public:
+  void FetchAndTest(const std::string& input,
+                    const std::string& passport,
+                    const std::string& ip,
+                    const std::string& hostname,
+                    int port, bool success) {
+    std::string passport_out;
+    std::string ip_out;
+    std::string hostname_out;
+    int port_out;
+    bool result = WebSocketProxyHelper::FetchPassportAddrNamePort(
+        (uint8*)input.data(), (uint8*)input.data() + input.length(),
+        &passport_out, &ip_out, &hostname_out, &port_out);
+    ASSERT_EQ(success, result) << "Input was: " << input;
+    if (success) {
+      EXPECT_EQ(passport, passport_out);
+      EXPECT_EQ(ip, ip_out);
+      EXPECT_EQ(hostname, hostname_out);
+      EXPECT_EQ(port, port_out);
+    }
+  }
+};
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNamePortSuccess) {
+  std::vector<std::string> ips;
+  ips.push_back("127.0.0.1");
+  ips.push_back("[ab:ab:ab:00:ed:78]");
+  std::vector<std::string> hostnames = ips;
+  hostnames.push_back("www.site.com");
+  hostnames.push_back("localhost");
+  hostnames.push_back("ab:ab:ab:ab:ab:ab");
+  ips.push_back("");  // Also valid ip, but not hostname.
+  std::vector<int> ports;
+  ports.push_back(1);
+  ports.push_back(65535);
+  for (size_t i = 0; i < ips.size(); ++i) {
+    for (size_t j = 0; j < hostnames.size(); ++j) {
+      for (size_t k = 0; k < ports.size(); ++k) {
+        std::string input = "passport:" + ips[i] + ":" +
+            hostnames[j] + ":" + base::IntToString(ports[k]) + ":";
+        FetchAndTest(input, "passport", ips[i], hostnames[j], ports[k], true);
+      }
+    }
+  }
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNamePortEmptyPassport) {
+  FetchAndTest("::localhost:1:", "", "", "", 0, false);
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNamePortBadIpv6) {
+  FetchAndTest("passport:[12:localhost:1:", "", "", "", 0, false);
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNameEmptyHostname) {
+  FetchAndTest("passport:::1:", "", "", "", 0, false);
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNameSmallPort) {
+  FetchAndTest("passport:::0:", "", "", "", 0, false);
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNameBigPort) {
+  FetchAndTest("passport:::65536:", "", "", "", 0, false);
+}
+
+TEST_F(WebSocketProxyHelperTest, FetchPassportAddrNoLastColon) {
+  FetchAndTest("passport::localhost:1", "", "", "", 0, false);
+}
+
+}  // namespace chromeos