Avoid sending mixed-content requests for ImageSet contexts

third_party/WebKit/Source/core/html/parser/HTMLPreloadScannerTest.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/html/parser/HTMLPreloadScanner.h"
 
 #include <memory>
 #include "core/MediaTypeNames.h"
 #include "core/css/MediaValuesCached.h"
 #include "core/frame/Settings.h"
@@ skipping 40 matching lines @@
   // shouldn't be checked (and no network request should be created).
   const char* expected_referrer;
 };
 
 struct NonceTestCase {
   const char* base_url;
   const char* input_html;
   const char* nonce;
 };
 
+struct ContextTestCase {
+  const char* base_url;
+  const char* input_html;
+  const char* preloaded_url;  // Or nullptr if no preload is expected.
+  bool is_image_set;
+};
+
 class MockHTMLResourcePreloader : public ResourcePreloader {
  public:
   void PreloadRequestVerification(Resource::Type type,
                                   const char* url,
                                   const char* base_url,
                                   int width,
                                   const ClientHintsPreferences& preferences) {
     if (!url) {
       EXPECT_FALSE(preload_request_);
       return;
@@ skipping 49 matching lines @@
   }
 
   void NonceRequestVerification(const char* nonce) {
     ASSERT_TRUE(preload_request_.get());
     if (strlen(nonce))
       EXPECT_EQ(nonce, preload_request_->Nonce());
     else
       EXPECT_TRUE(preload_request_->Nonce().IsEmpty());
   }
 
+  void ContextVerification(bool is_image_set) {
+    ASSERT_TRUE(preload_request_.get());
+    EXPECT_EQ(preload_request_->IsImageSetForTestingOnly(), is_image_set);
+  }
+
  protected:
   void Preload(std::unique_ptr<PreloadRequest> preload_request,
                const NetworkHintsInterface&) override {
     preload_request_ = std::move(preload_request);
   }
 
  private:
   std::unique_ptr<PreloadRequest> preload_request_;
 };
 
@@ skipping 99 matching lines @@
   void Test(NonceTestCase test_case) {
     MockHTMLResourcePreloader preloader;
     KURL base_url(kParsedURLString, test_case.base_url);
     scanner_->AppendToEnd(String(test_case.input_html));
     PreloadRequestStream requests = scanner_->Scan(base_url, nullptr);
     preloader.TakeAndPreload(requests);
 
     preloader.NonceRequestVerification(test_case.nonce);
   }
 
+  void Test(ContextTestCase test_case) {
+    MockHTMLResourcePreloader preloader;
+    KURL base_url(kParsedURLString, test_case.base_url);
+    scanner_->AppendToEnd(String(test_case.input_html));
+    PreloadRequestStream requests = scanner_->Scan(base_url, nullptr);
+    preloader.TakeAndPreload(requests);
+
+    preloader.ContextVerification(test_case.is_image_set);
+  }
+
  private:
   std::unique_ptr<DummyPageHolder> dummy_page_holder_;
   std::unique_ptr<HTMLPreloadScanner> scanner_;
 };
 
 TEST_F(HTMLPreloadScannerTest, testImages) {
   TestCase test_cases[] = {
       {"http://example.test", "<img src='bla.gif'>", "bla.gif",
        "http://example.test/", Resource::kImage, 0},
       {"http://example.test", "<img srcset='bla.gif 320w, blabla.gif 640w'>",
@@ skipping 376 matching lines @@
        "<picture><source media='(max-width: 400px)' type='image/webp' "
        "srcset='srcset_bla.gif'><img sizes='50vw' srcset='bla.gif "
        "500w'></picture>",
        "bla.gif", "http://example.test/", Resource::kImage, 250},
   };
 
   for (const auto& test_case : test_cases)
     Test(test_case);
 }
 
+TEST_F(HTMLPreloadScannerTest, testContext) {
+  ContextTestCase test_cases[] = {
+      {"http://example.test",
+       "<picture><source srcset='srcset_bla.gif'><img src='bla.gif'></picture>",
+       "srcset_bla.gif", true},
+      {"http://example.test", "<img src='bla.gif'>", "bla.gif", false},
+      {"http://example.test", "<img srcset='bla.gif'>", "bla.gif", true},
+  };
+
+  for (const auto& test_case : test_cases)
+    Test(test_case);
+}
+
 TEST_F(HTMLPreloadScannerTest, testReferrerPolicy) {
   ReferrerPolicyTestCase test_cases[] = {
       {"http://example.test", "<img src='bla.gif'/>", "bla.gif",
        "http://example.test/", Resource::kImage, 0, kReferrerPolicyDefault},
       {"http://example.test", "<img referrerpolicy='origin' src='bla.gif'/>",
        "bla.gif", "http://example.test/", Resource::kImage, 0,
        kReferrerPolicyOrigin, nullptr},
       {"http://example.test",
        "<meta name='referrer' content='not-a-valid-policy'><img "
        "src='bla.gif'/>",
@@ skipping 243 matching lines @@
       "sheet.css",
       "http://example.test/",
       Resource::kCSSStyleSheet,
       0,
       kReferrerPolicyAlways,
       "http://whatever.test/"};
   Test(test_case);
 }
 
 }  // namespace blink