TransformXMLTest.java

package edu.stanford.nlp.process;

import java.io.ByteArrayInputStream;
import java.io.StringWriter;

import junit.framework.TestCase;
import edu.stanford.nlp.util.StringUtils;
import java.util.function.Function;
import edu.stanford.nlp.objectbank.IdentityFunction;


/** @author Christopher Manning */
public class TransformXMLTest extends TestCase {

  private String testCase =
          "<doc><el arg=\"funny&'>"stuff\">yo! C&C! </el></doc>";

  private String expectedAnswer =
          "<doc> <el arg=\"funny&'>"stuff\"> yo! C&C! </el> </doc>";

  private String expectedAnswer2 =
          "<doc> <el arg=\"funny&'>"stuff\"> yo! C&C!yo! C&C! </el> </doc>";

  private Function<String,String> duplicate = new Function<String,String>() {
    public String apply(String in) {
      return in + in;
    }
  };

  public void testTransformXML1() {
    TransformXML<String> tx = new TransformXML<String>();
    StringWriter sw = new StringWriter();
    tx.transformXML(StringUtils.EMPTY_STRING_ARRAY, 
                    new IdentityFunction<String>(),
                    new ByteArrayInputStream(testCase.getBytes()),
                    sw);
    String answer = sw.toString().replaceAll("\\s+", " ").trim();
    assertEquals("Bad XML transform", expectedAnswer, answer);
    sw = new StringWriter();
    tx.transformXML(new String[] {"el"},
                    duplicate,
                    new ByteArrayInputStream(testCase.getBytes()),
                    sw);
    String answer2 = sw.toString().replaceAll("\\s+", " ").trim();
    assertEquals("Bad XML transform", expectedAnswer2, answer2);
  }

}