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.Style;
 import org.chromium.distiller.webdocument.WebEmbed;
 import org.chromium.distiller.webdocument.WebImage;
 import org.chromium.distiller.extractors.embeds.EmbedExtractor;
@@ skipping 192 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 169 matching lines @@
         assertNotNull(result);
         assertEquals("<img src=\"http://example.com/image.png\">", result.generateOutput(false));
     }
 
     public void testImageExtractorLazy() {
         extractLazilyLoadedImage("data-src");
         extractLazilyLoadedImage("datasrc");
         extractLazilyLoadedImage("data-original");
         extractLazilyLoadedImage("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 testFigureWithCaption() {
+        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();
+        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\">" +
+                    "<figcaption>This is a caption</figcaption>" +
+                "</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);
+    }
 }