samples/visitor/21-clientprovider/src-test/org/apidesign/test/visitor/PrintTest.java
1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/samples/visitor/21-clientprovider/src-test/org/apidesign/test/visitor/PrintTest.java Sat Jun 14 09:59:16 2008 +0200
1.3 @@ -0,0 +1,49 @@
1.4 +package org.apidesign.test.visitor;
1.5 +
1.6 +import static junit.framework.Assert.*;
1.7 +import org.apidesign.visitor.notevolutionready.Language.Expression;
1.8 +import org.apidesign.visitor.notevolutionready.Language.Number;
1.9 +import org.apidesign.visitor.notevolutionready.Language.Plus;
1.10 +import org.apidesign.visitor.notevolutionready.Language.Visitor;
1.11 +import org.junit.Test;
1.12 +
1.13 +public class PrintTest {
1.14 +
1.15 + // BEGIN: visitor.notevolutionready.print
1.16 + public static class PrintVisitor implements Visitor {
1.17 + StringBuffer sb = new StringBuffer();
1.18 +
1.19 + public void visitPlus(Plus s) {
1.20 + s.getFirst().visit(this);
1.21 + sb.append(" + ");
1.22 + s.getSecond().visit(this);
1.23 + }
1.24 +
1.25 + public void visitNumber(Number n) {
1.26 + sb.append (n.getValue());
1.27 + }
1.28 + }
1.29 +
1.30 + @Test public void printOnePlusOne() {
1.31 + Number one = new Number(1);
1.32 + Expression plus = new Plus(one, one);
1.33 +
1.34 + PrintVisitor print = new PrintVisitor();
1.35 + plus.visit(print);
1.36 +
1.37 + assertEquals("1 + 1", print.sb.toString());
1.38 + }
1.39 + // END: visitor.notevolutionready.print
1.40 +
1.41 + @Test public void printOnePlusTwoPlusThree() {
1.42 + Number one = new Number(1);
1.43 + Number two = new Number(2);
1.44 + Number three = new Number(3);
1.45 + Expression plus = new Plus(one, new Plus(two, three));
1.46 +
1.47 + PrintVisitor print = new PrintVisitor();
1.48 + plus.visit(print);
1.49 +
1.50 + assertEquals("1 + 2 + 3", print.sb.toString());
1.51 + }
1.52 +}