Support <picture> in image extraction

javatests/org/chromium/distiller/DomUtilTest.java

 // Copyright 2014 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.
 
 package org.chromium.distiller;
 
 import org.chromium.distiller.webdocument.WebTable;
 
 import com.google.gwt.core.client.JsArray;
 import com.google.gwt.dom.client.Document;
@@ skipping 232 matching lines @@
 
         assertEquals(11, contentNodes.size());
     }
 
     public void testMakeAllLinksAbsolute() {
         final String html =
             "<a href=\"link\"></a>" +
             "<img src=\"image\" srcset=\"image200 200w, image400 400w\">" +
             "<img src=\"image2\">" +
             "<video src=\"video\" poster=\"poster\">" +
-                "<source src=\"source\">" +
+                "<source src=\"source\" srcset=\"s2, s3\">" +
                 "<track src=\"track\">" +
             "</video>";
 
         final String expected =
             "<a href=\"http://example.com/link\"></a>" +
             "<img src=\"http://example.com/image\" " +
               "srcset=\"http://example.com/image200 200w, http://example.com/image400 400w\">" +
             "<img src=\"http://example.com/image2\">" +
             "<video src=\"http://example.com/video\" poster=\"http://example.com/poster\">" +
-                "<source src=\"http://example.com/source\">" +
+                "<source src=\"http://example.com/source\" " +
+                    "srcset=\"http://example.com/s2, http://example.com/s3\">" +
                 "<track src=\"http://example.com/track\">" +
             "</video>";
 
         mHead.setInnerHTML("<base href=\"http://example.com/\">");
         mBody.setInnerHTML(html);
 
         mBody.setInnerHTML(html);
         for (int i = 0; i < mBody.getChildCount(); i++) {
             DomUtil.makeAllLinksAbsolute(mBody.getChild(i));
         }
         assertEquals(expected, mBody.getInnerHTML());
     }
 
     public void testGetSrcSetUrls() {
         String html =
             "<img src=\"http://example.com/image\" " +
               "srcset=\"http://example.com/image200 200w, http://example.com/image400 400w\">";
 
         mBody.setInnerHTML(html);
         List<String> list = DomUtil.getSrcSetUrls((ImageElement) mBody.getChild(0));
         assertEquals(2, list.size());
         assertEquals("http://example.com/image200", list.get(0));
         assertEquals("http://example.com/image400", list.get(1));
     }
 
+    public void testGetAllSrcSetUrls() {
+        String html =
+            "<picture>" +
+                "<source srcset=\"image200 200w, //example.org/image400 400w\">" +
+                "<source srcset=\"image100 100w, //example.org/image300 300w\">" +
+                "<img>" +
+            "</picture>";
+        Element container = Document.get().createDivElement();
+        container.setInnerHTML(html);
+        List<String> urls = DomUtil.getAllSrcSetUrls(container);
+        assertEquals(4, urls.size());
+        assertEquals("image200", urls.get(0));
+        assertEquals("//example.org/image400", urls.get(1));
+        assertEquals("image100", urls.get(2));
+        assertEquals("//example.org/image300", urls.get(3));
+    }
+
     public void testStripTableBackgroundColorAttributes() {
         String tableWithBGColorHTML =
             "<table bgcolor=\"red\">" +
                 "<tbody>" +
                     "<tr bgcolor=\"red\">" +
                         "<th bgcolor=\"red\">text</th>" +
                         "<th bgcolor=\"red\">text</th>" +
                     "</tr><tr bgcolor=\"red\">" +
                         "<td bgcolor=\"red\">text</td>" +
                         "<td bgcolor=\"red\">text</td>" +
@@ skipping 356 matching lines @@
         element = element.getNextSiblingElement();
         assertEquals(300*200, DomUtil.getArea(element));
 
         element = element.getNextSiblingElement();
         assertEquals(400*100, DomUtil.getArea(element));
 
         element = element.getFirstChildElement();
         assertEquals(400*100, DomUtil.getArea(element));
     }
 }