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());
    }
    
}