Strip unwanted classNames from all nodes

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 374 matching lines @@
         for (int i = 0; i < mBody.getChildCount(); i++) {
             DomUtil.stripImageElements(mBody.getChild(i));
         }
         assertEquals(expected, mBody.getInnerHTML());
 
         mBody.setInnerHTML(html);
         DomUtil.stripImageElements(mBody);
         assertEquals(expected, mBody.getInnerHTML());
     }
 
+    public void testStripUnwantedClassNames() {
+        String html =
+            "<br class=\"iscaptiontxt\">" +
+            "<br id=\"a\">" +
+            "<br class=\"\">" +
+            "<div class=\"tion cap\">" +
+                "<br class=\"hascaption\">" +
+                "<br class=\"not_me\">" +
+            "</div>";
+
+        final String expected =
+            "<br class=\"caption\">" +
+            "<br id=\"a\">" +
+            "<br>" +
+            "<div>" +
+                "<br class=\"caption\">" +
+                "<br>" +
+            "</div>";
+
+        mBody.setInnerHTML(html);
+        for (int i = 0; i < mBody.getChildCount(); i++) {
+            DomUtil.stripUnwantedClassNames(Element.as(mBody.getChild(i)));
+        }
+        assertEquals(expected, mBody.getInnerHTML());
+
+        mBody.setInnerHTML(html);
+        DomUtil.stripUnwantedClassNames(mBody);
+        assertEquals(expected, mBody.getInnerHTML());
+    }
+
     public void testIsVisibleByOffsetParentDisplayNone() {
         String html =
             "<div style=\"display: none;\">" +
                 "<div></div>" +
             "</div>";
         mBody.setInnerHTML(html);
         Element child = mBody.getFirstChildElement().getFirstChildElement();
         assertFalse(DomUtil.isVisibleByOffset(child));
     }
 
@@ skipping 220 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));
     }
 }