Add support for figure element

javatests/org/chromium/distiller/EmbedExtractorTest.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 com.google.gwt.dom.client.AnchorElement;
 import com.google.gwt.dom.client.Style;
+import org.chromium.distiller.webdocument.WebElement;
 import org.chromium.distiller.webdocument.WebEmbed;
+import org.chromium.distiller.webdocument.WebFigure;
 import org.chromium.distiller.webdocument.WebImage;
 import org.chromium.distiller.extractors.embeds.EmbedExtractor;
 import org.chromium.distiller.extractors.embeds.TwitterExtractor;
 import org.chromium.distiller.extractors.embeds.VimeoExtractor;
 import org.chromium.distiller.extractors.embeds.YouTubeExtractor;
 import org.chromium.distiller.extractors.embeds.ImageExtractor;
 
 import com.google.gwt.dom.client.Document;
 import com.google.gwt.dom.client.Element;
 import com.google.gwt.dom.client.IFrameElement;
@@ skipping 184 matching lines @@
         result = (WebEmbed) extractor.extract(nontweet);
         assertNull(result);
     }
 
     public void testRenderedTwitterExtractor() {
         IFrameElement twitter = TestUtil.createIframe();
         // Add iframe to body so its document is generated.
         mBody.appendChild(twitter);
 
         // This string represents a very simplified version of the twitter iframe embed structure.
-        String iframeStructure = 
+        String iframeStructure =
                 "<div class=\"media-forward root standalone-tweet ltr\"" +
                     "data-iframe-title=\"Embedded Tweet\"" +
                     "data-scribe=\"page:tweet\">" +
                     "<blockquote data-tweet-id=\"1234567890\"" +
                         "data-scribe=\"section:subject\">" +
                         "<div class=\"cards-base cards-multimedia customisable-border\"" +
                             "data-scribe=\"component:card\"" +
                             "data-video-content-id=\"0987654321\">" +
                         "</div>" +
                         "<div class=\"header\">" +
@@ skipping 163 matching lines @@
         mBody.appendChild(image);
 
         mHead.setInnerHTML("<base href=\"http://example.com/\">");
 
         EmbedExtractor extractor = new ImageExtractor();
         WebImage result = (WebImage) extractor.extract(image);
         assertNotNull(result);
         assertEquals("<img src=\"http://example.com/image.png\">", result.generateOutput(false));
     }
 
+    private void extractLazilyLoadedFigure(String attr) {
+        ImageElement image = TestUtil.createImage();
+        image.setAttribute(attr, "image.png");
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+
+        mBody.appendChild(figure);
+
+        mHead.setInnerHTML("<base href=\"http://example.com/\">");
+
+        EmbedExtractor extractor = new ImageExtractor();
+        WebFigure result = (WebFigure) extractor.extract(figure);
+        assertNotNull(result);
+        assertEquals("<figure><img src=\"http://example.com/image.png\"></figure>",
+                result.generateOutput(false));
+    }
+
     public void testImageExtractorLazy() {
         extractLazilyLoadedImage("data-src");
         extractLazilyLoadedImage("datasrc");
         extractLazilyLoadedImage("data-original");
         extractLazilyLoadedImage("data-url");
+
+        extractLazilyLoadedFigure("data-src");
+        extractLazilyLoadedFigure("datasrc");
+        extractLazilyLoadedFigure("data-original");
+        extractLazilyLoadedFigure("data-url");
+    }
+
+    public void testFigureWithoutCaption() {
+        ImageElement image = TestUtil.createImage();
+        image.setSrc("http://wwww.example.com/image.jpeg");
+        image.setAttribute("width", "100");
+        image.setAttribute("height", "100");
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+        mBody.appendChild(figure);
+
+        EmbedExtractor extractor = new ImageExtractor();
+        WebImage result = (WebImage) extractor.extract(figure);
+        String got = result.generateOutput(false);
+        String expected =
+            "<figure>" +
+                "<img src=\"http://wwww.example.com/image.jpeg\"" +
+                    " width=\"100\" height=\"100\">" +
+            "</figure>";
+        assertNotNull(result);
+        assertEquals(100, result.getHeight());
+        assertEquals(100, result.getWidth());
+        assertEquals(expected, got);
+    }
+
+    public void testFigureWithoutImageAndCaption() {
+        Element figure = Document.get().createElement("FIGURE");
+        mBody.appendChild(figure);
+
+        EmbedExtractor extractor = new ImageExtractor();
+        WebImage result = (WebImage) extractor.extract(figure);
+        assertNull(result);
+    }
+
+    public void testFigureCaptionTextOnly() {
+        ImageElement image = TestUtil.createImage();
+        image.setSrc("http://wwww.example.com/image.jpeg");
+        image.setAttribute("width", "100");
+        image.setAttribute("height", "100");
+
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+        Element figcaption = Document.get().createElement("FIGCAPTION");
+        figcaption.setInnerHTML("This is a caption");
+        figure.appendChild(figcaption);
+        mBody.appendChild(figure);
+
+        EmbedExtractor extractor = new ImageExtractor();
+        WebElement result = extractor.extract(figure);
+        assertEquals("This is a caption", result.generateOutput(true));
+    }
+
+    public void testFigureCaptionWithAnchor() {
+        mHead.setInnerHTML("<base href=\"http://example.com/\">");
+
+        ImageElement image = TestUtil.createImage();
+        image.setSrc("http://wwww.example.com/image.jpeg");
+        image.setAttribute("width", "100");
+        image.setAttribute("height", "100");
+
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+        figure.setAttribute("class", "test");
+        figure.setAttribute("attribute", "value");
+
+        Element figcaption = Document.get().createElement("FIGCAPTION");
+        AnchorElement anchor = Document.get().createAnchorElement();
+        anchor.setHref("link_page.html");
+        anchor.setInnerHTML("caption link");
+        figcaption.appendChild(TestUtil.createText("This is a "));
+        figcaption.appendChild(anchor);
+        figure.appendChild(figcaption);
+        mBody.appendChild(figure);
+        String expected =
+            "<figure>" +
+                "<img src=\"http://wwww.example.com/image.jpeg\"" +
+                    " width=\"100\" height=\"100\">" +
+                "<figcaption>This is a " +
+                    "<a href=\"http://example.com/link_page.html\">caption link</a>" +
+                "</figcaption>" +
+            "</figure>";
+        EmbedExtractor extractor = new ImageExtractor();
+        WebElement result = extractor.extract(figure);
+        assertEquals(expected, TestUtil.removeAllDirAttributes(
+                result.generateOutput(false)));
+    }
+
+    public void testFigureWithCaptionWithoutAnchor() {
+        ImageElement image = TestUtil.createImage();
+        image.setSrc("http://wwww.example.com/image.jpeg");
+        image.setAttribute("width", "100");
+        image.setAttribute("height", "100");
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+        Element figcaption = Document.get().createElement("FIGCAPTION");
+        figcaption.setInnerHTML("<div><span>This is a caption</span></div>");
+        figure.appendChild(figcaption);
+        mBody.appendChild(figure);
+
+        EmbedExtractor extractor = new ImageExtractor();
+        WebImage result = (WebImage) extractor.extract(figure);
+        String expected =
+            "<figure>" +
+                "<img src=\"http://wwww.example.com/image.jpeg\"" +
+                    " width=\"100\" height=\"100\">" +
+                "<figcaption>This is a caption</figcaption>" +
+            "</figure>";
+        assertNotNull(result);
+        assertEquals(100, result.getHeight());
+        assertEquals(100, result.getWidth());
+        assertEquals(expected, TestUtil.removeAllDirAttributes(
+                result.generateOutput(false)));
+    }
+
+    public void testDivCaption() {
+        ImageElement image = TestUtil.createImage();
+        image.setSrc("http://wwww.example.com/image.jpeg");
+        image.setAttribute("width", "100");
+        image.setAttribute("height", "100");
+
+        Element figure = Document.get().createElement("FIGURE");
+        figure.appendChild(image);
+        Element div = Document.get().createElement("DIV");
+        div.setInnerHTML("This is a caption");
+        figure.appendChild(div);
+        mBody.appendChild(figure);
+        String expected =
+            "<figure>" +
+                "<img src=\"http://wwww.example.com/image.jpeg\"" +
+                    " width=\"100\" height=\"100\">" +
+                "<figcaption>This is a caption</figcaption>" +
+            "</figure>";
+        EmbedExtractor extractor = new ImageExtractor();
+        WebElement result = extractor.extract(figure);
+        assertEquals(expected, TestUtil.removeAllDirAttributes(
+                result.generateOutput(false)));
     }
 }