Extract image URLs in srcset as well

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;
 import com.google.gwt.dom.client.Element;
+import com.google.gwt.dom.client.ImageElement;
 import com.google.gwt.dom.client.Node;
 import com.google.gwt.dom.client.NodeList;
 
 import java.util.Map;
 import java.util.List;
 
 public class DomUtilTest extends DomDistillerJsTestCase {
     public void testGetAttributes() {
         Element e = Document.get().createDivElement();
         e.setInnerHTML("<div style=\"width:50px; height:100px\" id=\"f\" class=\"sdf\"></div>");
@@ skipping 244 matching lines @@
         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 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 326 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));
     }
 }