Added basic client hints support.

Source/core/fetch/ResourceFetcherTest.cpp

 /*
  * Copyright (c) 2013, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 19 matching lines @@
 
 #include "config.h"
 #include "core/fetch/ResourceFetcher.h"
 
 #include <gtest/gtest.h>
 #include "core/fetch/FetchInitiatorInfo.h"
 #include "core/fetch/FetchRequest.h"
 #include "core/fetch/ImageResource.h"
 #include "core/fetch/MemoryCache.h"
 #include "core/fetch/ResourcePtr.h"
+#include "core/frame/FrameView.h"
 #include "core/html/HTMLDocument.h"
 #include "core/loader/DocumentLoader.h"
+#include "core/page/Page.h"
+#include "core/testing/DummyPageHolder.h"
 #include "platform/network/ResourceRequest.h"
 #include "platform/weborigin/KURL.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "public/platform/WebURLRequest.h"
 
 namespace blink {
 
-class ResourceFetcherTest : public ::testing::Test {
+class ResourceFetcherUpgradeTest : public ::testing::Test {
 public:
-    ResourceFetcherTest()
+    ResourceFetcherUpgradeTest()
         : secureURL(ParsedURLString, "https://secureorigin.test/image.png")
         , exampleOrigin(SecurityOrigin::create(KURL(ParsedURLString, "https://example.test/")))
         , secureOrigin(SecurityOrigin::create(secureURL))
     {
     }
 
 protected:
     virtual void SetUp()
     {
         // Create a ResourceFetcher that has a real DocumentLoader and Document, but is not attached to a LocalFrame.
@@ skipping 47 matching lines @@
     RefPtr<SecurityOrigin> exampleOrigin;
     RefPtr<SecurityOrigin> secureOrigin;
 
     // We don't use the DocumentLoader directly in any tests, but need to keep it around as long
     // as the ResourceFetcher and Document live due to indirect usage.
     RefPtr<DocumentLoader> documentLoader;
     RefPtrWillBePersistent<ResourceFetcher> fetcher;
     RefPtrWillBePersistent<Document> document;
 };
 
-TEST_F(ResourceFetcherTest, StartLoadAfterFrameDetach)
+TEST_F(ResourceFetcherUpgradeTest, StartLoadAfterFrameDetach)
 {
     EXPECT_EQ(fetcher->frame(), static_cast<LocalFrame*>(0));
 
     // Try to request a url. The request should fail, no resource should be returned,
     // and no resource should be present in the cache.
     FetchRequest fetchRequest = FetchRequest(ResourceRequest(secureURL), FetchInitiatorInfo());
     ResourcePtr<ImageResource> image = fetcher->fetchImage(fetchRequest);
     EXPECT_EQ(image.get(), static_cast<ImageResource*>(0));
     EXPECT_EQ(memoryCache()->resourceForURL(secureURL), static_cast<Resource*>(0));
 }
 
-TEST_F(ResourceFetcherTest, UpgradeInsecureResourceRequests)
+TEST_F(ResourceFetcherUpgradeTest, UpgradeInsecureResourceRequests)
 {
     struct TestCase {
         const char* original;
         const char* upgraded;
     } tests[] = {
         { "http://example.test/image.png", "https://example.test/image.png" },
         { "http://example.test:80/image.png", "https://example.test:443/image.png" },
         { "http://example.test:1212/image.png", "https://example.test:1212/image.png" },
 
         { "https://example.test/image.png", "https://example.test/image.png" },
@@ skipping 23 matching lines @@
         expectUpgrade(test.original, WebURLRequest::RequestContextForm, WebURLRequest::FrameTypeTopLevel, test.upgraded);
         expectUpgrade(test.original, WebURLRequest::RequestContextForm, WebURLRequest::FrameTypeAuxiliary, test.upgraded);
 
         // Or unless the host of the document matches the host of the resource:
         document->setSecurityOrigin(exampleOrigin);
         expectUpgrade(test.original, WebURLRequest::RequestContextScript, WebURLRequest::FrameTypeTopLevel, test.upgraded);
         expectUpgrade(test.original, WebURLRequest::RequestContextScript, WebURLRequest::FrameTypeAuxiliary, test.upgraded);
     }
 }
 
-TEST_F(ResourceFetcherTest, DoNotUpgradeInsecureResourceRequests)
+TEST_F(ResourceFetcherUpgradeTest, DoNotUpgradeInsecureResourceRequests)
 {
     document->setSecurityOrigin(secureOrigin);
     document->setInsecureContentPolicy(SecurityContext::InsecureContentDoNotUpgrade);
 
     expectUpgrade("http://example.test/image.png", "http://example.test/image.png");
     expectUpgrade("http://example.test:80/image.png", "http://example.test:80/image.png");
     expectUpgrade("http://example.test:1212/image.png", "http://example.test:1212/image.png");
 
     expectUpgrade("https://example.test/image.png", "https://example.test/image.png");
     expectUpgrade("https://example.test:80/image.png", "https://example.test:80/image.png");
     expectUpgrade("https://example.test:1212/image.png", "https://example.test:1212/image.png");
 
     expectUpgrade("ftp://example.test/image.png", "ftp://example.test/image.png");
     expectUpgrade("ftp://example.test:21/image.png", "ftp://example.test:21/image.png");
     expectUpgrade("ftp://example.test:1212/image.png", "ftp://example.test:1212/image.png");
 }
 
-TEST_F(ResourceFetcherTest, MonitorInsecureResourceRequests)
+TEST_F(ResourceFetcherUpgradeTest, MonitorInsecureResourceRequests)
 {
     document->setSecurityOrigin(secureOrigin);
     document->setInsecureContentPolicy(SecurityContext::InsecureContentMonitor);
 
     expectUpgrade("http://example.test/image.png", "http://example.test/image.png");
     expectUpgrade("http://example.test:80/image.png", "http://example.test:80/image.png");
     expectUpgrade("http://example.test:1212/image.png", "http://example.test:1212/image.png");
 
     expectUpgrade("https://example.test/image.png", "https://example.test/image.png");
     expectUpgrade("https://example.test:80/image.png", "https://example.test:80/image.png");
     expectUpgrade("https://example.test:1212/image.png", "https://example.test:1212/image.png");
 
     expectUpgrade("ftp://example.test/image.png", "ftp://example.test/image.png");
     expectUpgrade("ftp://example.test:21/image.png", "ftp://example.test:21/image.png");
     expectUpgrade("ftp://example.test:1212/image.png", "ftp://example.test:1212/image.png");
 }
 
-TEST_F(ResourceFetcherTest, SendPreferHeader)
+TEST_F(ResourceFetcherUpgradeTest, SendPreferHeader)
 {
     struct TestCase {
         const char* toRequest;
         WebURLRequest::FrameType frameType;
         bool shouldPrefer;
     } tests[] = {
         { "http://example.test/page.html", WebURLRequest::FrameTypeAuxiliary, true },
         { "http://example.test/page.html", WebURLRequest::FrameTypeNested, true },
         { "http://example.test/page.html", WebURLRequest::FrameTypeNone, false },
         { "http://example.test/page.html", WebURLRequest::FrameTypeTopLevel, true },
         { "https://example.test/page.html", WebURLRequest::FrameTypeAuxiliary, false },
         { "https://example.test/page.html", WebURLRequest::FrameTypeNested, false },
         { "https://example.test/page.html", WebURLRequest::FrameTypeNone, false },
         { "https://example.test/page.html", WebURLRequest::FrameTypeTopLevel, false }
     };
 
     for (auto test : tests) {
         document->setInsecureContentPolicy(SecurityContext::InsecureContentDoNotUpgrade);
         expectPreferHeader(test.toRequest, test.frameType, test.shouldPrefer);
 
         document->setInsecureContentPolicy(SecurityContext::InsecureContentUpgrade);
         expectPreferHeader(test.toRequest, test.frameType, test.shouldPrefer);
 
         document->setInsecureContentPolicy(SecurityContext::InsecureContentMonitor);
         expectPreferHeader(test.toRequest, test.frameType, test.shouldPrefer);
     }
 }
 
+class ResourceFetcherHintsTest : public ::testing::Test {
+public:
+    ResourceFetcherHintsTest() { }
+
+protected:
+    virtual void SetUp()
+    {
+        dummyPageHolder = DummyPageHolder::create(IntSize(500, 500));
+        dummyPageHolder->page().setDeviceScaleFactor(1.0);
+        documentLoader = DocumentLoader::create(&dummyPageHolder->frame(), ResourceRequest("http://www.example.com"), SubstituteData());
+        document = toHTMLDocument(&dummyPageHolder->document());
+        fetcher = documentLoader->fetcher();
+        fetcher->setDocument(document.get());
+    }
+
+    void expectHeader(const char* input, const char* headerName, bool isPresent, const char* headerValue)
+    {
+        KURL inputURL(ParsedURLString, input);
+        FetchRequest fetchRequest = FetchRequest(ResourceRequest(inputURL), FetchInitiatorInfo());
+        fetcher->addClientHintsIfNeccessary(fetchRequest);
+
+        EXPECT_STREQ(isPresent ? headerValue : "",
+            fetchRequest.resourceRequest().httpHeaderField(headerName).utf8().data());
+    }
+
+    OwnPtr<DummyPageHolder> dummyPageHolder;
+    // We don't use the DocumentLoader directly in any tests, but need to keep it around as long
+    // as the ResourceFetcher and Document live due to indirect usage.
+    RefPtr<DocumentLoader> documentLoader;
+    RefPtrWillBePersistent<Document> document;
+    RefPtrWillBePersistent<ResourceFetcher> fetcher;
+};
+
+TEST_F(ResourceFetcherHintsTest, MonitorDPRHints)
+{
+    expectHeader("http://www.example.com/1.gif", "DPR", false, "");
+    dummyPageHolder->frame().setShouldSendDPRHint();
+    expectHeader("http://www.example.com/1.gif", "DPR", true, "1");
+    dummyPageHolder->page().setDeviceScaleFactor(2.5);
+    expectHeader("http://www.example.com/1.gif", "DPR", true, "2.5");
+    expectHeader("http://www.example.com/1.gif", "RW", false, "");
+}
+
+TEST_F(ResourceFetcherHintsTest, MonitorRWHints)
+{
+    expectHeader("http://www.example.com/1.gif", "RW", false, "");
+    dummyPageHolder->frame().setShouldSendRWHint();
+    expectHeader("http://www.example.com/1.gif", "RW", true, "500");
+    dummyPageHolder->frameView().setLayoutSizeFixedToFrameSize(false);
+    dummyPageHolder->frameView().setLayoutSize(IntSize(800, 800));
+    expectHeader("http://www.example.com/1.gif", "RW", true, "800");
+    expectHeader("http://www.example.com/1.gif", "DPR", false, "");
+}
+
+TEST_F(ResourceFetcherHintsTest, MonitorBothHints)
+{
+    expectHeader("http://www.example.com/1.gif", "DPR", false, "");
+    expectHeader("http://www.example.com/1.gif", "RW", false, "");
+
+    dummyPageHolder->frame().setShouldSendDPRHint();
+    dummyPageHolder->frame().setShouldSendRWHint();
+    expectHeader("http://www.example.com/1.gif", "DPR", true, "1");
+    expectHeader("http://www.example.com/1.gif", "RW", true, "500");
+}
+
 } // namespace