package org.codehaus.groovy.antlr.treewalker;
import junit.framework.TestCase;
import org.codehaus.groovy.antlr.parser.GroovyTokenTypes;
import org.custommonkey.xmlunit.Diff;
import org.custommonkey.xmlunit.XMLUnit;
/**
* Testcases for the antlr AST visitor that prints groovy source code nodes as Freemind compatible XML.
*/
public class MindMapPrinterTest extends TestCase implements GroovyTokenTypes{
private static final String HEADER = "<map version='0.7.1'><node TEXT='AST'>";
private static final String FOOTER = "</node></map>";
protected void setUp() throws Exception {
XMLUnit.setIgnoreWhitespace(true);
}
public void testAbstract() throws Exception {
assertXmlEquals(
HEADER +
"<node TEXT='MODIFIERS <" + MODIFIERS +">' POSITION='right' COLOR=\"#000000\">" +
" <node TEXT='public <" + LITERAL_public +">' POSITION='right' COLOR=\"#17178B\"></node>" +
"</node>" +
"<node TEXT='abstract <" + ABSTRACT +">' POSITION='right' COLOR=\"#006699\"></node>" +
"<node TEXT='CLASS_DEF <" + CLASS_DEF +">' POSITION='right' COLOR=\"#17178B\"></node>" +
"<node TEXT='an identifier <" + IDENT +"> : Foo' POSITION='right' COLOR=\"#006699\"></node>" +
"<node TEXT='EXTENDS_CLAUSE <" + EXTENDS_CLAUSE +">' POSITION='right' COLOR=\"#17178B\"></node>" +
"<node TEXT='IMPLEMENTS_CLAUSE <" + IMPLEMENTS_CLAUSE +">' POSITION='right' COLOR=\"#17178B\"></node>" +
"<node TEXT='OBJBLOCK <" + OBJBLOCK +">' POSITION='right' COLOR=\"#006699\"></node>" +
FOOTER,
pretty("public abstract class Foo{}"));
}
public void testArrayDeclarator() throws Exception {
assertXmlEquals(
HEADER +
"<node TEXT='VARIABLE_DEF <" + VARIABLE_DEF +"> : primes' POSITION='right' COLOR=\"#000000\">" +
" <node TEXT='TYPE <" + TYPE +">' POSITION='right' COLOR=\"#17178B\">" +
" <node TEXT='ARRAY_DECLARATOR <" + ARRAY_DECLARATOR +"> : [' POSITION='right' COLOR=\"#000000\">" +
" <node TEXT='int <" + LITERAL_int +">' POSITION='right' COLOR=\"#17178B\"></node>" +
" </node>" +
" </node>" +
" <node TEXT='an identifier <" + IDENT +"> : primes' POSITION='right' COLOR=\"#006699\"></node>" +
" <node TEXT='= <" + ASSIGN +">' POSITION='right' COLOR=\"#000000\">" +
" <node TEXT='new <" + LITERAL_new +">' POSITION='right' COLOR=\"#17178B\">" +
" <node TEXT='int <" + LITERAL_int +">' POSITION='right' COLOR=\"#17178B\"></node>" +
" <node TEXT='ARRAY_DECLARATOR <" + ARRAY_DECLARATOR +"> : [' POSITION='right' COLOR=\"#000000\">" +
" <node TEXT='a numeric literal <" + NUM_INT +"> : 5' POSITION='right' COLOR=\"#006699\"></node>" +
" </node>" +
" </node>" +
" </node>" +
"</node>" +
FOOTER,
pretty("int[] primes = new int[5]"));
}
public void testRegexMatch() throws Exception {
assertXmlEquals(
HEADER +
"<node TEXT='if <" + LITERAL_if +">' POSITION='right' COLOR=\"#17178B\">" +
" <node TEXT='an identifier <" + IDENT +"> : foo' POSITION='right' COLOR=\"#006699\"></node>" +
" <node TEXT='==~ <" + REGEX_MATCH +">' POSITION='right' COLOR=\"#000000\"></node>" +
" <node TEXT='a string literal <" + STRING_LITERAL +"> : bar' POSITION='right' COLOR=\"#008000\"></node>" +
"</node>" +
"<node TEXT='SLIST <" + SLIST +"> : {' POSITION='right' COLOR=\"#006699\"></node>" +
FOOTER,
pretty("if (foo==~\"bar\"){}"));
}
private void assertXmlEquals(String expected, String actual) throws Exception {
Diff diff = new Diff(expected, actual);
assertTrue(diff.toString(), diff.similar());
}
private String pretty(String input) throws Exception {
TraversalTestHelper traverser = new TraversalTestHelper();
return traverser.traverse(input, MindMapPrinter.class);
}
}