Home | History | Annotate | Download | only in base 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "net/base/url_util.h"
      6 
      7 #include "testing/gtest/include/gtest/gtest.h"
      8 #include "url/gurl.h"
      9 
     10 namespace net {
     11 namespace {
     12 
     13 TEST(UrlUtilTest, AppendQueryParameter) {
     14   // Appending a name-value pair to a URL without a query component.
     15   EXPECT_EQ("http://example.com/path?name=value",
     16             AppendQueryParameter(GURL("http://example.com/path"),
     17                                  "name", "value").spec());
     18 
     19   // Appending a name-value pair to a URL with a query component.
     20   // The original component should be preserved, and the new pair should be
     21   // appended with '&'.
     22   EXPECT_EQ("http://example.com/path?existing=one&name=value",
     23             AppendQueryParameter(GURL("http://example.com/path?existing=one"),
     24                                  "name", "value").spec());
     25 
     26   // Appending a name-value pair with unsafe characters included. The
     27   // unsafe characters should be escaped.
     28   EXPECT_EQ("http://example.com/path?existing=one&na+me=v.alue%3D",
     29             AppendQueryParameter(GURL("http://example.com/path?existing=one"),
     30                                  "na me", "v.alue=").spec());
     31 
     32 }
     33 
     34 TEST(UrlUtilTest, AppendOrReplaceQueryParameter) {
     35   // Appending a name-value pair to a URL without a query component.
     36   EXPECT_EQ("http://example.com/path?name=value",
     37             AppendOrReplaceQueryParameter(GURL("http://example.com/path"),
     38                                  "name", "value").spec());
     39 
     40   // Appending a name-value pair to a URL with a query component.
     41   // The original component should be preserved, and the new pair should be
     42   // appended with '&'.
     43   EXPECT_EQ("http://example.com/path?existing=one&name=value",
     44       AppendOrReplaceQueryParameter(
     45           GURL("http://example.com/path?existing=one"),
     46           "name", "value").spec());
     47 
     48   // Appending a name-value pair with unsafe characters included. The
     49   // unsafe characters should be escaped.
     50   EXPECT_EQ("http://example.com/path?existing=one&na+me=v.alue%3D",
     51       AppendOrReplaceQueryParameter(
     52           GURL("http://example.com/path?existing=one"),
     53           "na me", "v.alue=").spec());
     54 
     55   // Replace value of an existing paramater.
     56   EXPECT_EQ("http://example.com/path?existing=one&name=new",
     57       AppendOrReplaceQueryParameter(
     58           GURL("http://example.com/path?existing=one&name=old"),
     59           "name", "new").spec());
     60 
     61   // Replace a name-value pair with unsafe characters included. The
     62   // unsafe characters should be escaped.
     63   EXPECT_EQ("http://example.com/path?na+me=n.ew%3D&existing=one",
     64       AppendOrReplaceQueryParameter(
     65           GURL("http://example.com/path?na+me=old&existing=one"),
     66           "na me", "n.ew=").spec());
     67 
     68   // Replace the value of first parameter with this name only.
     69   EXPECT_EQ("http://example.com/path?name=new&existing=one&name=old",
     70       AppendOrReplaceQueryParameter(
     71           GURL("http://example.com/path?name=old&existing=one&name=old"),
     72           "name", "new").spec());
     73 
     74   // Preserve the content of the original params regarless of our failure to
     75   // interpret them correctly.
     76   EXPECT_EQ("http://example.com/path?bar&name=new&left=&"
     77             "=right&=&&name=again",
     78       AppendOrReplaceQueryParameter(
     79           GURL("http://example.com/path?bar&name=old&left=&"
     80                 "=right&=&&name=again"),
     81           "name", "new").spec());
     82 }
     83 
     84 TEST(UrlUtilTest, GetValueForKeyInQuery) {
     85   GURL url("http://example.com/path?name=value&boolParam&"
     86            "url=http://test.com/q?n1%3Dv1%26n2");
     87   std::string value;
     88 
     89   // False when getting a non-existent query param.
     90   EXPECT_FALSE(GetValueForKeyInQuery(url, "non-exist", &value));
     91 
     92   // True when query param exist.
     93   EXPECT_TRUE(GetValueForKeyInQuery(url, "name", &value));
     94   EXPECT_EQ("value", value);
     95 
     96   EXPECT_TRUE(GetValueForKeyInQuery(url, "boolParam", &value));
     97   EXPECT_EQ("", value);
     98 
     99   EXPECT_TRUE(GetValueForKeyInQuery(url, "url", &value));
    100   EXPECT_EQ("http://test.com/q?n1=v1&n2", value);
    101 }
    102 
    103 TEST(UrlUtilTest, GetValueForKeyInQueryInvalidURL) {
    104   GURL url("http://%01/?test");
    105   std::string value;
    106 
    107   // Always false when parsing an invalid URL.
    108   EXPECT_FALSE(GetValueForKeyInQuery(url, "test", &value));
    109 }
    110 
    111 }  // namespace
    112 }  // namespace net
    113