Add support for media attribute on link

third_party/WebKit/Source/core/loader/LinkHeaderTest.cpp

 // Copyright 2015 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 "core/loader/LinkHeader.h"
 
 #include "testing/gtest/include/gtest/gtest.h"
 #include <base/macros.h>
 
 namespace blink {
 namespace {
 
 TEST(LinkHeaderTest, Empty)
 {
     String nullString;
     LinkHeaderSet nullHeaderSet(nullString);
     ASSERT_EQ(nullHeaderSet.size(), unsigned(0));
     String emptyString("");
     LinkHeaderSet emptyHeaderSet(emptyString);
     ASSERT_EQ(emptyHeaderSet.size(), unsigned(0));
 }
 
 struct SingleTestCase {
     const char* headerValue;
     bool valid;
     const char* url;
     const char* rel;
     const char* as;
+    const char* media;
 } singleTestCases[] = {
-    {"</images/cat.jpg>; rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>;rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>   ;rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"< /images/cat.jpg>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg >   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg wutwut>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg wutwut  \t >   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; Rel=prefetch   ", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; Rel=PReFetCh   ", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; rel=prefetch; rel=somethingelse", true, "/images/cat.jpg", "prefetch", ""},
-    {"  </images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", ""},
-    {"\t  </images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>\t\t ; \trel=prefetch \t  ", true, "/images/cat.jpg", "prefetch", ""},
+    {"</images/cat.jpg>; rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>;rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>   ;rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"< /images/cat.jpg>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg >   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg wutwut>   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg wutwut  \t >   ;   rel=prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; Rel=prefetch   ", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; Rel=PReFetCh   ", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; rel=prefetch; rel=somethingelse", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"  </images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"\t  </images/cat.jpg>; rel=prefetch   ", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>\t\t ; \trel=prefetch \t  ", true, "/images/cat.jpg", "prefetch", "", ""},
     {"\f</images/cat.jpg>\t\t ; \trel=prefetch \t  ", false},
-    {"</images/cat.jpg>; rel= prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"<../images/cat.jpg?dog>; rel= prefetch", true, "../images/cat.jpg?dog", "prefetch", ""},
-    {"</images/cat.jpg>; rel =prefetch", true, "/images/cat.jpg", "prefetch", ""},
+    {"</images/cat.jpg>; rel= prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"<../images/cat.jpg?dog>; rel= prefetch", true, "../images/cat.jpg?dog", "prefetch", "", ""},
+    {"</images/cat.jpg>; rel =prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
     {"</images/cat.jpg>; rel pel=prefetch", false},
-    {"< /images/cat.jpg>", true, "/images/cat.jpg", "", ""},
+    {"< /images/cat.jpg>", true, "/images/cat.jpg", "", "", ""},
     {"</images/cat.jpg>; rel =", false},
-    {"</images/cat.jpg>; wut=sup; rel =prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; wut=sup ; rel =prefetch", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; wut=sup ; rel =prefetch  \t  ;", true, "/images/cat.jpg", "prefetch", ""},
+    {"</images/cat.jpg>; wut=sup; rel =prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; wut=sup ; rel =prefetch", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; wut=sup ; rel =prefetch  \t  ;", true, "/images/cat.jpg", "prefetch", "", ""},
     {"</images/cat.jpg> wut=sup ; rel =prefetch  \t  ;", false},
     {"<   /images/cat.jpg", false},
-    {"<   http://wut.com/  sdfsdf ?sd>; rel=dns-prefetch", true, "http://wut.com/", "dns-prefetch", ""},
-    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=dns-prefetch", true, "http://wut.com/%20%20%3dsdfsdf?sd", "dns-prefetch", ""},
-    {"<   http://wut.com/dfsdf?sdf=ghj&wer=rty>; rel=prefetch", true, "http://wut.com/dfsdf?sdf=ghj&wer=rty", "prefetch", ""},
-    {"<   http://wut.com/dfsdf?sdf=ghj&wer=rty>;;;;; rel=prefetch", true, "http://wut.com/dfsdf?sdf=ghj&wer=rty", "prefetch", ""},
-    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=preload;as=image", true, "http://wut.com/%20%20%3dsdfsdf?sd", "preload", "image"},
-    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=preload;as=whatever", true, "http://wut.com/%20%20%3dsdfsdf?sd", "preload", "whatever"},
+    {"<   http://wut.com/  sdfsdf ?sd>; rel=dns-prefetch", true, "http://wut.com/", "dns-prefetch", "", ""},
+    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=dns-prefetch", true, "http://wut.com/%20%20%3dsdfsdf?sd", "dns-prefetch", "", ""},
+    {"<   http://wut.com/dfsdf?sdf=ghj&wer=rty>; rel=prefetch", true, "http://wut.com/dfsdf?sdf=ghj&wer=rty", "prefetch", "", ""},
+    {"<   http://wut.com/dfsdf?sdf=ghj&wer=rty>;;;;; rel=prefetch", true, "http://wut.com/dfsdf?sdf=ghj&wer=rty", "prefetch", "", ""},
+    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=preload;as=image", true, "http://wut.com/%20%20%3dsdfsdf?sd", "preload", "image", ""},
+    {"<   http://wut.com/%20%20%3dsdfsdf?sd>; rel=preload;as=whatever", true, "http://wut.com/%20%20%3dsdfsdf?sd", "preload", "whatever", ""},
     {"</images/cat.jpg>; anchor=foo; rel=prefetch;", false},
     {"</images/cat.jpg>; rel=prefetch;anchor=foo ", false},
     {"</images/cat.jpg>; anchor='foo'; rel=prefetch;", false},
     {"</images/cat.jpg>; rel=prefetch;anchor='foo' ", false},
     {"</images/cat.jpg>; rel=prefetch;anchor='' ", false},
-    {"</images/cat.jpg>; rel=prefetch;", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/cat.jpg>; rel=prefetch    ;", true, "/images/cat.jpg", "prefetch", ""},
-    {"</images/ca,t.jpg>; rel=prefetch    ;", true, "/images/ca,t.jpg", "prefetch", ""},
-    {"<simple.css>; rel=stylesheet; title=\"title with a DQUOTE and backslash\"", true, "simple.css", "stylesheet", ""},
+    {"</images/cat.jpg>; rel=prefetch;", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/cat.jpg>; rel=prefetch    ;", true, "/images/cat.jpg", "prefetch", "", ""},
+    {"</images/ca,t.jpg>; rel=prefetch    ;", true, "/images/ca,t.jpg", "prefetch", "", ""},
+    {"<simple.css>; rel=stylesheet; title=\"title with a DQUOTE and backslash\"", true, "simple.css", "stylesheet", "", ""},
     {"<simple.css>; rel=stylesheet; title=\"title with a DQUOTE \\\" and backslash: \\\"", false},
-    {"<simple.css>; title=\"title with a DQUOTE \\\" and backslash: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", ""},
-    {"<simple.css>; title=\'title with a DQUOTE \\\' and backslash: \'; rel=stylesheet; ", true, "simple.css", "stylesheet", ""},
-    {"<simple.css>; title=\"title with a DQUOTE \\\" and ;backslash,: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", ""},
-    {"<simple.css>; title=\"title with a DQUOTE \' and ;backslash,: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", ""},
-    {"<simple.css>; title=\"\"; rel=stylesheet; ", true, "simple.css", "stylesheet", ""},
-    {"<simple.css>; title=\"\"; rel=\"stylesheet\"; ", true, "simple.css", "stylesheet", ""},
+    {"<simple.css>; title=\"title with a DQUOTE \\\" and backslash: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", "", ""},
+    {"<simple.css>; title=\'title with a DQUOTE \\\' and backslash: \'; rel=stylesheet; ", true, "simple.css", "stylesheet", "", ""},
+    {"<simple.css>; title=\"title with a DQUOTE \\\" and ;backslash,: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", "", ""},
+    {"<simple.css>; title=\"title with a DQUOTE \' and ;backslash,: \"; rel=stylesheet; ", true, "simple.css", "stylesheet", "", ""},
+    {"<simple.css>; title=\"\"; rel=stylesheet; ", true, "simple.css", "stylesheet", "", ""},
+    {"<simple.css>; title=\"\"; rel=\"stylesheet\"; ", true, "simple.css", "stylesheet", "", ""},
     {"<simple.css>; rel=stylesheet; title=\"", false},
-    {"<simple.css>; rel=stylesheet; title=\"\"", true, "simple.css", "stylesheet", ""},
+    {"<simple.css>; rel=stylesheet; title=\"\"", true, "simple.css", "stylesheet", "", ""},
     {"<simple.css>; rel=\"stylesheet\"; title=\"", false},
     {"<simple.css>; rel=\";style,sheet\"; title=\"", false},
     {"<simple.css>; rel=\"bla'sdf\"; title=\"", false},
-    {"<simple.css>; rel=\"\"; title=\"\"", true, "simple.css", "", ""},
-    {"<simple.css>; rel=''; title=\"\"", true, "simple.css", "", ""},
+    {"<simple.css>; rel=\"\"; title=\"\"", true, "simple.css", "", "", ""},
+    {"<simple.css>; rel=''; title=\"\"", true, "simple.css", "", "", ""},
     {"<simple.css>; rel=''; title=", false},
     {"<simple.css>; rel=''; title", false},
     {"<simple.css>; rel=''; media", false},
     {"<simple.css>; rel=''; hreflang", false},
     {"<simple.css>; rel=''; type", false},
     {"<simple.css>; rel=''; rev", false},
-    {"<simple.css>; rel=''; bla", true, "simple.css", "", ""},
+    {"<simple.css>; rel=''; bla", true, "simple.css", "", "", ""},
     {"<simple.css>; rel='prefetch", false},
     {"<simple.css>; rel=\"prefetch", false},
     {"<simple.css>; rel=\"", false},
-    {"<http://whatever.com>; rel=preconnect; valid!", true, "http://whatever.com", "preconnect", ""},
-    {"<http://whatever.com>; rel=preconnect; valid$", true, "http://whatever.com", "preconnect", ""},
+    {"<http://whatever.com>; rel=preconnect; valid!", true, "http://whatever.com", "preconnect", "", ""},
+    {"<http://whatever.com>; rel=preconnect; valid$", true, "http://whatever.com", "preconnect", "", ""},
     {"<http://whatever.com>; rel=preconnect; invalid@", false},
     {"<http://whatever.com>; rel=preconnect; invalid*", false},
+    {"</images/cat.jpg>; rel=prefetch;media='(max-width: 5000px)'", true, "/images/cat.jpg", "prefetch", "", "(max-width: 5000px)"},
+    {"</images/cat.jpg>; rel=prefetch;media=\"(max-width: 5000px)\"", true, "/images/cat.jpg", "prefetch", "", "(max-width: 5000px)"},
+    {"</images/cat.jpg>; rel=prefetch;media=(max-width:5000px)", true, "/images/cat.jpg", "prefetch", "", "(max-width:5000px)"},
 };
 
 void PrintTo(const SingleTestCase& test, std::ostream* os)
 {
     *os << ::testing::PrintToString(test.headerValue);
 }
 
 class SingleLinkHeaderTest : public ::testing::TestWithParam<SingleTestCase> {};
 
 // Test the cases with a single header
 TEST_P(SingleLinkHeaderTest, Single)
 {
     const SingleTestCase testCase = GetParam();
     LinkHeaderSet headerSet(testCase.headerValue);
     ASSERT_EQ(1u, headerSet.size());
     LinkHeader& header = headerSet[0];
     EXPECT_EQ(testCase.valid, header.valid());
     if (testCase.valid) {
         EXPECT_STREQ(testCase.url, header.url().ascii().data());
         EXPECT_STREQ(testCase.rel, header.rel().ascii().data());
         EXPECT_STREQ(testCase.as, header.as().ascii().data());
+        EXPECT_STREQ(testCase.media, header.media().ascii().data());
     }
 }
 
 INSTANTIATE_TEST_CASE_P(LinkHeaderTest, SingleLinkHeaderTest, testing::ValuesIn(singleTestCases));
 
 struct DoubleTestCase {
     const char* headerValue;
     const char* url;
     const char* rel;
     bool valid;
@@ skipping 68 matching lines @@
     EXPECT_STREQ(testCase.url, header.url().ascii().data());
     EXPECT_STREQ(testCase.rel, header.rel().ascii().data());
     EXPECT_EQ(testCase.crossorigin, header.crossOrigin());
     EXPECT_EQ(testCase.valid, header.valid());
 }
 
 INSTANTIATE_TEST_CASE_P(LinkHeaderTest, CrossOriginLinkHeaderTest, testing::ValuesIn(crossOriginTestCases));
 
 } // namespace
 } // namespace blink