Set width and height attributes for <img> if image is loaded from srcset

third_party/WebKit/Source/web/tests/WebFrameSerializerTest.cpp

 /*
  * Copyright (C) 2011 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 125 matching lines @@
     URLTestHelpers::registerMockedURLLoad(parsedURL, fileName,
                                           "frameserialization/", "text/html");
     FrameTestHelpers::loadFrame(mainFrameImpl(), url.utf8().data());
     SingleLinkRewritingDelegate delegate(parsedURL, WebString("local"));
     SimpleWebFrameSerializerClient serializerClient;
     WebFrameSerializer::serialize(mainFrameImpl(), &serializerClient,
                                   &delegate);
     return serializerClient.toString();
   }
 
+  WebViewImpl* webView() { return m_helper.webView(); }
+
   WebLocalFrameImpl* mainFrameImpl() {
     return m_helper.webView()->mainFrameImpl();
   }
 
  private:
   FrameTestHelpers::WebViewHelper m_helper;
 };
 
 TEST_F(WebFrameSerializerTest, URLAttributeValues) {
   registerMockedImageURL("javascript:\"");
@@ skipping 121 matching lines @@
 }
 
 // Regression test for crbug.com/678893, where in some cases serializing an
 // image document could cause code to pick an element from an empty container.
 TEST_F(WebFrameSerializerSanitizationTest, FromBrokenImageDocument) {
   String mhtml = generateMHTMLParts("http://www.test.com", "broken-image.png",
                                     "image/png");
   EXPECT_TRUE(mhtml.isEmpty());
 }
 
+TEST_F(WebFrameSerializerSanitizationTest, ImageLoadedFromSrcsetForHiDPI) {
+  URLTestHelpers::registerMockedURLLoad(
+      KURL(ParsedURLString, "http://www.test.com/1x.png"),
+      "frameserialization/1x.png");
+  URLTestHelpers::registerMockedURLLoad(
+      KURL(ParsedURLString, "http://www.test.com/2x.png"),
+      "frameserialization/2x.png");
+
+  // Set high DPR in order to load image from srcset, instead of src.
+  webView()->setDeviceScaleFactor(2.0f);
+
+  String mhtml = generateMHTMLParts("http://www.test.com", "img_srcset.html");
+
+  // srcset attribute should be skipped.
+  EXPECT_EQ(WTF::kNotFound, mhtml.find("srcset="));
+
+  // Width and height attributes should be set when none is present in <img>.
+  EXPECT_NE(WTF::kNotFound,
+            mhtml.find("id=3D\"i1\" width=3D\"6\" height=3D\"6\">"));
+
+  // Height attribute should not be set if width attribute is already present in
+  // <img>
+  EXPECT_NE(WTF::kNotFound, mhtml.find("id=3D\"i2\" width=3D\"8\">"));
+}
+
+TEST_F(WebFrameSerializerSanitizationTest, ImageLoadedFromSrcForNormalDPI) {
+  URLTestHelpers::registerMockedURLLoad(
+      KURL(ParsedURLString, "http://www.test.com/1x.png"),
+      "frameserialization/1x.png");
+  URLTestHelpers::registerMockedURLLoad(
+      KURL(ParsedURLString, "http://www.test.com/2x.png"),
+      "frameserialization/2x.png");
+
+  String mhtml = generateMHTMLParts("http://www.test.com", "img_srcset.html");
+
+  // srcset attribute should be skipped.
+  EXPECT_EQ(WTF::kNotFound, mhtml.find("srcset="));
+
+  // New width and height attributes should not be set.
+  EXPECT_NE(WTF::kNotFound, mhtml.find("id=3D\"i1\">"));
+  EXPECT_NE(WTF::kNotFound, mhtml.find("id=3D\"i2\" width=3D\"8\">"));
+}
+
 }  // namespace blink