package org.openlcb;

import junit.framework.Assert;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;

/**
 * @author  Bob Jacobsen   Copyright 2009
 * @version $Revision: 2175 $
 */
public class GatewayTest extends TestCase {
    protected Gateway getGateway() {
        return new Gateway();
    }
    
    public void testCtor() {
        getGateway();
    }

    public void testGet() {
        g = getGateway();
        Connection cE = g.getEastConnection();
        Connection cW = g.getWestConnection();
    }
    
    protected boolean resultE;
    protected boolean resultW;
    protected Gateway g;
    protected TestListener tE;
    protected TestListener tW;
    protected Connection cE;
    protected Connection cW;
    
    abstract class TestListener extends AbstractConnection {
        public void put(Message m, Connection n) {
            setResult();
        }
        abstract void setResult();
    }

    protected void buildGateway() {
        g = getGateway();
        resultE = false;
        resultW = false;
        tE = new TestListener() {
            void setResult() { resultE = true;}
        };
        tW = new TestListener() {
            void setResult() { resultW = true;}
        };
        g.registerWest(tW);
        g.registerEast(tE);
        cE = g.getEastConnection();
        cW = g.getWestConnection();
    }

    public void testEastToWest() {
        buildGateway();
        Message m = new Message(new NodeID(new byte[]{1,2,3,4,5,6}))
            {public int getMTI() {return 0; }};

        cE.put(m, tE);
        
        checkMovedEastToWestOnly();
    }
    
    protected void checkMovedEastToWestOnly() {
        Assert.assertTrue(!resultE);
        Assert.assertTrue(resultW);
        resultE = false;
        resultW = false;
    }
    
    public void testWestToEast() {
        buildGateway();
        Message m = new Message(new NodeID(new byte[]{1,2,3,4,5,6}))
            {public int getMTI() {return 0; }};

        cW.put(m, tW);
        
        checkMovedWestToEastOnly();
    }
    
    protected void checkMovedWestToEastOnly() {
        Assert.assertTrue(resultE);
        Assert.assertTrue(!resultW);
        resultE = false;
        resultW = false;
    }
        
    // from here down is testing infrastructure
    
    public GatewayTest(String s) {
        super(s);
    }

    // Main entry point
    static public void main(String[] args) {
        String[] testCaseName = {GatewayTest.class.getName()};
        junit.swingui.TestRunner.main(testCaseName);
    }

    // test suite from all defined tests
    public static Test suite() {
        TestSuite suite = new TestSuite(GatewayTest.class);
        return suite;
    }
}