package org.apidesign.test.visitor;

import org.apidesign.test.visitor.PrintTest.PrintVisitor;

import static junit.framework.Assert.*;

import org.apidesign.visitor.Language.Expression;

import org.apidesign.visitor.Language.Minus;

import org.apidesign.visitor.Language.Number;

import org.apidesign.visitor.Language.Plus;

import org.apidesign.visitor.Language.Real;

import org.apidesign.visitor.Language.Visitor;

import org.junit.Test;

public class PrintRealTest {

    @Test public void printOneMinusTwo() {

        Real one = newReal(1);

        Real two = newReal(2);

        Expression plus = PrintOfMinusStructureTest.newMinus(one, two);

        PrintVisitor print = new PrintVisitor();

        plus.visit(print.dispatch);

        if (Boolean.getBoolean("no.failures")) {

            assertEquals("unknownunknownunknown", print.sb.toString());

            return;

        }

        assertEquals("1.0 - 2.0", print.sb.toString());

    }

    @Test public void visitorReadyForVersion3_0() {

        class PrintVisitor3_0 implements Visitor.Version3_0 {

            StringBuffer sb = new StringBuffer();

            Visitor dispatch = Visitor.create(this);

            public boolean visitUnknown(Expression exp) {

                sb.append("unknown");

                return true;

            }

            public void visitPlus(Plus s) {

                s.getFirst().visit(dispatch);

                sb.append(" + ");

                s.getSecond().visit(dispatch);

            }

            public void visitMinus(Minus m) {

                m.getFirst().visit(dispatch);

                sb.append(" - ");

                m.getSecond().visit(dispatch);

            }

            public void visitReal(Real r) {

                sb.append(r.getValue());

            }

        }

        Number one = PrintTest.newNumber(1);

        Number two = PrintTest.newNumber(2);

        Expression plus = PrintOfMinusStructureTest.newMinus(one, two);

        PrintVisitor3_0 print = new PrintVisitor3_0();

        plus.visit(print.dispatch);

        assertEquals("1.0 - 2.0", print.sb.toString());

        Real five = newReal(5);

        Real three = newReal(3);

        Expression realPlus = PrintOfMinusStructureTest.newMinus(five, three);

        PrintVisitor3_0 printReal = new PrintVisitor3_0();

        realPlus.visit(printReal.dispatch);

        assertEquals("5.0 - 3.0", printReal.sb.toString());

    }

    public static Real newReal(final double value) {

        return new Real() {

            public double getValue() {

                return value;

            }

            public void visit(Visitor v) {

                v.dispatchReal(this);

            }

        };

    }

}