Move folly/Hash.h to folly/hash/

[folly.git] / folly / test / UriTest.cpp

diff --git a/folly/test/UriTest.cpp b/folly/test/UriTest.cpp
index 97b14637f5f53c470ed60537f78e3fea2bb97eae..1431d63da0e74389eb691d0536d41b366a737ec3 100644 (file)
--- a/folly/test/UriTest.cpp
+++ b/folly/test/UriTest.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright 2013 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -14,17 +14,15 @@
  * limitations under the License.
  */
 
-#include "folly/Uri.h"
+#include <folly/Uri.h>
+#include <folly/portability/GTest.h>
 
+#include <boost/algorithm/string.hpp>
 #include <glog/logging.h>
-#include <gtest/gtest.h>
+#include <map>
 
 using namespace folly;
 
-namespace {
-
-}  // namespace
-
 TEST(Uri, Simple) {
   {
     fbstring s("http://www.facebook.com/hello/world?query#fragment");
@@ -34,6 +32,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("www.facebook.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("www.facebook.com", u.authority());
     EXPECT_EQ("/hello/world", u.path());
     EXPECT_EQ("query", u.query());
     EXPECT_EQ("fragment", u.fragment());
@@ -48,6 +47,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("www.facebook.com", u.host());
     EXPECT_EQ(8080, u.port());
+    EXPECT_EQ("www.facebook.com:8080", u.authority());
     EXPECT_EQ("/hello/world", u.path());
     EXPECT_EQ("query", u.query());
     EXPECT_EQ("fragment", u.fragment());
@@ -62,6 +62,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("127.0.0.1", u.host());
     EXPECT_EQ(8080, u.port());
+    EXPECT_EQ("127.0.0.1:8080", u.authority());
     EXPECT_EQ("/hello/world", u.path());
     EXPECT_EQ("query", u.query());
     EXPECT_EQ("fragment", u.fragment());
@@ -75,13 +76,47 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.username());
     EXPECT_EQ("", u.password());
     EXPECT_EQ("[::1]", u.host());
+    EXPECT_EQ("::1", u.hostname());
     EXPECT_EQ(8080, u.port());
+    EXPECT_EQ("[::1]:8080", u.authority());
     EXPECT_EQ("/hello/world", u.path());
     EXPECT_EQ("query", u.query());
     EXPECT_EQ("fragment", u.fragment());
     EXPECT_EQ(s, u.fbstr());  // canonical
   }
 
+  {
+    fbstring s("http://[2401:db00:20:7004:face:0:29:0]:8080/hello/world?query");
+    Uri u(s);
+    EXPECT_EQ("http", u.scheme());
+    EXPECT_EQ("", u.username());
+    EXPECT_EQ("", u.password());
+    EXPECT_EQ("[2401:db00:20:7004:face:0:29:0]", u.host());
+    EXPECT_EQ("2401:db00:20:7004:face:0:29:0", u.hostname());
+    EXPECT_EQ(8080, u.port());
+    EXPECT_EQ("[2401:db00:20:7004:face:0:29:0]:8080", u.authority());
+    EXPECT_EQ("/hello/world", u.path());
+    EXPECT_EQ("query", u.query());
+    EXPECT_EQ("", u.fragment());
+    EXPECT_EQ(s, u.fbstr());  // canonical
+  }
+
+  {
+    fbstring s("http://[2401:db00:20:7004:face:0:29:0]/hello/world?query");
+    Uri u(s);
+    EXPECT_EQ("http", u.scheme());
+    EXPECT_EQ("", u.username());
+    EXPECT_EQ("", u.password());
+    EXPECT_EQ("[2401:db00:20:7004:face:0:29:0]", u.host());
+    EXPECT_EQ("2401:db00:20:7004:face:0:29:0", u.hostname());
+    EXPECT_EQ(0, u.port());
+    EXPECT_EQ("[2401:db00:20:7004:face:0:29:0]", u.authority());
+    EXPECT_EQ("/hello/world", u.path());
+    EXPECT_EQ("query", u.query());
+    EXPECT_EQ("", u.fragment());
+    EXPECT_EQ(s, u.fbstr());  // canonical
+  }
+
   {
     fbstring s("http://user:pass@host.com/");
     Uri u(s);
@@ -90,6 +125,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("pass", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("user:pass@host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -104,6 +140,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("user@host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -118,6 +155,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("user@host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -132,6 +170,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("pass", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ(":pass@host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -146,6 +185,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -160,6 +200,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("host.com", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("host.com", u.authority());
     EXPECT_EQ("/", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -174,6 +215,7 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("", u.authority());
     EXPECT_EQ("/etc/motd", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
@@ -188,10 +230,11 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("", u.authority());
     EXPECT_EQ("/etc/motd", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
-    EXPECT_EQ("file:///etc/motd", u.fbstr());
+    EXPECT_EQ("file:/etc/motd", u.fbstr());
   }
 
   {
@@ -202,12 +245,166 @@ TEST(Uri, Simple) {
     EXPECT_EQ("", u.password());
     EXPECT_EQ("etc", u.host());
     EXPECT_EQ(0, u.port());
+    EXPECT_EQ("etc", u.authority());
     EXPECT_EQ("/motd", u.path());
     EXPECT_EQ("", u.query());
     EXPECT_EQ("", u.fragment());
     EXPECT_EQ(s, u.fbstr());
   }
 
-  EXPECT_THROW({Uri("2http://www.facebook.com/");},
-               std::invalid_argument);
+  {
+    // test query parameters
+    fbstring s("http://localhost?&key1=foo&key2=&key3&=bar&=bar=&");
+    Uri u(s);
+    auto paramsList = u.getQueryParams();
+    std::map<fbstring, fbstring> params;
+    for (auto& param : paramsList) {
+      params[param.first] = param.second;
+    }
+    EXPECT_EQ(3, params.size());
+    EXPECT_EQ("foo", params["key1"]);
+    EXPECT_NE(params.end(), params.find("key2"));
+    EXPECT_EQ("", params["key2"]);
+    EXPECT_NE(params.end(), params.find("key3"));
+    EXPECT_EQ("", params["key3"]);
+  }
+
+  {
+    // test query parameters
+    fbstring s("http://localhost?&&&&&&&&&&&&&&&");
+    Uri u(s);
+    auto params = u.getQueryParams();
+    EXPECT_TRUE(params.empty());
+  }
+
+  {
+    // test query parameters
+    fbstring s("http://localhost?&=invalid_key&key2&key3=foo");
+    Uri u(s);
+    auto paramsList = u.getQueryParams();
+    std::map<fbstring, fbstring> params;
+    for (auto& param : paramsList) {
+      params[param.first] = param.second;
+    }
+    EXPECT_EQ(2, params.size());
+    EXPECT_NE(params.end(), params.find("key2"));
+    EXPECT_EQ("", params["key2"]);
+    EXPECT_EQ("foo", params["key3"]);
+  }
+
+  {
+    // test query parameters
+    fbstring s("http://localhost?&key1=====&&=key2&key3=");
+    Uri u(s);
+    auto paramsList = u.getQueryParams();
+    std::map<fbstring, fbstring> params;
+    for (auto& param : paramsList) {
+      params[param.first] = param.second;
+    }
+    EXPECT_EQ(1, params.size());
+    EXPECT_NE(params.end(), params.find("key3"));
+    EXPECT_EQ("", params["key3"]);
+  }
+
+  {
+    // test query parameters
+    fbstring s("http://localhost?key1=foo=bar&key2=foobar&");
+    Uri u(s);
+    auto paramsList = u.getQueryParams();
+    std::map<fbstring, fbstring> params;
+    for (auto& param : paramsList) {
+      params[param.first] = param.second;
+    }
+    EXPECT_EQ(1, params.size());
+    EXPECT_EQ("foobar", params["key2"]);
+  }
+
+  {
+    fbstring s("2http://www.facebook.com");
+
+    try {
+      Uri u(s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument& ex) {
+      EXPECT_TRUE(boost::algorithm::ends_with(ex.what(), s));
+    }
+  }
+
+  {
+    fbstring s("www[facebook]com");
+
+    try {
+      Uri u("http://" + s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument& ex) {
+      EXPECT_TRUE(boost::algorithm::ends_with(ex.what(), s));
+    }
+  }
+
+  {
+    fbstring s("http://[::1:8080/hello/world?query#fragment");
+
+    try {
+      Uri u(s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument&) {
+      // success
+    }
+  }
+
+  {
+    fbstring s("http://::1]:8080/hello/world?query#fragment");
+
+    try {
+      Uri u(s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument&) {
+      // success
+    }
+  }
+
+  {
+    fbstring s("http://::1:8080/hello/world?query#fragment");
+
+    try {
+      Uri u(s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument&) {
+      // success
+    }
+  }
+
+  {
+    fbstring s("http://2401:db00:20:7004:face:0:29:0/hello/world?query");
+
+    try {
+      Uri u(s);
+      CHECK(false) << "Control should not have reached here";
+    } catch (const std::invalid_argument&) {
+      // success
+    }
+  }
+
+  // No authority (no "//") is valid
+  {
+    fbstring s("this:is/a/valid/uri");
+    Uri u(s);
+    EXPECT_EQ("this", u.scheme());
+    EXPECT_EQ("is/a/valid/uri", u.path());
+    EXPECT_EQ(s, u.fbstr());
+  }
+  {
+    fbstring s("this:is:another:valid:uri");
+    Uri u(s);
+    EXPECT_EQ("this", u.scheme());
+    EXPECT_EQ("is:another:valid:uri", u.path());
+    EXPECT_EQ(s, u.fbstr());
+  }
+  {
+    fbstring s("this:is@another:valid:uri");
+    Uri u(s);
+    EXPECT_EQ("this", u.scheme());
+    EXPECT_EQ("is@another:valid:uri", u.path());
+    EXPECT_EQ(s, u.fbstr());
+  }
 }