source: tmcsimulator/trunk/test/tmcsim/client/cadclientgui/data/CADDataTest.java @ 476

package tmcsim.client.cadclientgui.data;



import java.util.Vector;

import javax.swing.table.DefaultTableModel;

import junit.framework.TestCase;



/**

 *

 * @author jdalbey

 */

public class CADDataTest extends TestCase {

    private Vector<Unit> units;

    private Unit alpha;

    public CADDataTest(String testName) {

        super(testName);

    }

    

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        units = new Vector<Unit>();

        alpha = new Unit("1");

        units.add(alpha);

    }

    

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }



    /**

     * Test of clearData method, of class CADData.

     */

    public void testClearData() {

        System.out.println("clearData");

        CADData instance = new CADData();

        instance.setUnitsFromXML(units);

        assertEquals(alpha,instance.getUnit("1"));

        instance.clearData();

        assertFalse(instance.checkForValidId(1));

    }



    /**

     * Test of setUnitsFromXML method, of class CADData.

     */

    public void testSetUnitsFromXML() {

        System.out.println("setUnitsFromXML");

        CADData instance = new CADData();

        instance.setUnitsFromXML(units);

        assertEquals(alpha,instance.getUnit("1"));

    }





    /**

     * Test of tableForUnitStatus method, of class CADData.

     */

    public void testTableForUnitStatus() {

        System.out.println("tableForUnitStatus");

        CADData instance = new CADData();

        instance.setUnitsFromXML(units);

        DefaultTableModel result = instance.tableForUnitStatus();

        assertEquals(1, result.getRowCount());

    }



    /**

     * Test of unitAvailable method, of class CADData.

     */

    public void testUnitAvailable() {

        System.out.println("unitAvailable");

        String unitNum = "1";

        CADData instance = new CADData();

        instance.setUnitsFromXML(units);

        instance.addIncident(new Incident(1,"Desc", 10L));

        instance.unitAssignedToIncident(unitNum, 1, true);

        instance.unitAvailable(unitNum);

        assertEquals("10-98",instance.getUnit("1").getStatus());

    }

    

}