Upgrade: Send a 'Prefer' header for feature detection.

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 63 matching lines @@
 
     void expectUpgrade(const char* input, WebURLRequest::RequestContext requestContext, WebURLRequest::FrameType frameType, const char* expected)
     {
         KURL inputURL(ParsedURLString, input);
         KURL expectedURL(ParsedURLString, expected);
 
         FetchRequest fetchRequest = FetchRequest(ResourceRequest(inputURL), FetchInitiatorInfo());
         fetchRequest.mutableResourceRequest().setRequestContext(requestContext);
         fetchRequest.mutableResourceRequest().setFrameType(frameType);
 
-        fetcher->maybeUpgradeInsecureRequestURL(fetchRequest);
+        fetcher->upgradeInsecureRequest(fetchRequest);
 
         EXPECT_STREQ(expectedURL.string().utf8().data(), fetchRequest.resourceRequest().url().string().utf8().data());
         EXPECT_EQ(expectedURL.protocol(), fetchRequest.resourceRequest().url().protocol());
         EXPECT_EQ(expectedURL.host(), fetchRequest.resourceRequest().url().host());
         EXPECT_EQ(expectedURL.port(), fetchRequest.resourceRequest().url().port());
         EXPECT_EQ(expectedURL.hasPort(), fetchRequest.resourceRequest().url().hasPort());
         EXPECT_EQ(expectedURL.path(), fetchRequest.resourceRequest().url().path());
     }
 
+    void expectPreferHeader(const char* input, WebURLRequest::FrameType frameType, bool shouldPrefer)
+    {
+        KURL inputURL(ParsedURLString, input);
+
+        FetchRequest fetchRequest = FetchRequest(ResourceRequest(inputURL), FetchInitiatorInfo());
+        fetchRequest.mutableResourceRequest().setRequestContext(WebURLRequest::RequestContextScript);
+        fetchRequest.mutableResourceRequest().setFrameType(frameType);
+
+        fetcher->upgradeInsecureRequest(fetchRequest);
+
+        EXPECT_STREQ(shouldPrefer ? "return=secure-representation" : "",
+            fetchRequest.resourceRequest().httpHeaderField("Prefer").utf8().data());
+    }
+
     KURL secureURL;
     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;
 };
@@ skipping 83 matching lines @@
 
     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)
+{
+    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);
+    }
+}
+
 } // namespace