source: tmcsimulator/trunk/test/atmsdriver/model/LoadHighwaysTest.java @ 343

package atmsdriver.model;



import java.io.FileWriter;

import java.io.PrintWriter;

import java.nio.file.FileSystems;

import java.nio.file.Files;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.List;

import junit.framework.TestCase;



/**

 *

 * @author jdalbey

 */

public class LoadHighwaysTest extends TestCase {



    public LoadHighwaysTest(String testName) {

        super(testName);

    }



    @Override

    protected void setUp() throws Exception {

        super.setUp();

        PrintWriter writer = null;

        try {

            writer = new PrintWriter(new FileWriter("test/atmsdriver/model/lds_loadhighways_sample.txt"));

            

            writer.println("2");

            writer.println("32 0 2");

            writer.println("1210831 1 5 S 0.9 4 CALAFIA");

            writer.println("1210832 ML ML_1");

            writer.println("1210833 ML ML_2");

            writer.println("1210834 ML ML_3");

            writer.println("1210835 ML ML_4");

            writer.println("1210845 2 5 S 1.49 4 EL CAMINO REAL");

            writer.println("1210846 ML ML_1");

            writer.println("1210847 ML ML_2");

            writer.println("1210848 ML ML_3");

            writer.println("1210849 ML ML_4");

            writer.println("74 0 1");

            writer.println("1204203 2 5 N 1.26 4 MAGDALENA");

            writer.println("1204212 ML ML_1");

            writer.println("1204213 ML ML_2");

            writer.println("1204214 ML ML_3");

            writer.println("1204215 ML ML_4");

            writer.close();

        } catch (Exception e) {

            e.printStackTrace();

        }

    }



    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

        Path path = FileSystems.getDefault().getPath("test/atmsdriver/model", "lds_loadhighways_sample.txt");

        Files.delete(path);

    }



    /**

     * Test of Highways constructor 

     */

    public void testLoadHighways() {

        System.out.println("toXML");

        Highways highways = new Highways(

                "test/atmsdriver/model/lds_loadhighways_sample.txt",

                "localhost", 8080);



        // Test for correct number of highways

        List<Highway> result = (ArrayList) highways.highways;

        assertEquals(1, result.size());

        

        // Test 5 was loaded

        Highway fiveS = result.get(0);

        assertEquals(new Integer(5), fiveS.routeNumber);

        

        // Test for correct number of stations

        assertEquals(new Integer(3), new Integer(fiveS.stations.size()));

        

        // Test 5 stations are sorted by postmile

        List<Station> stations = (ArrayList) fiveS.stations;

        ArrayList<Double> stationsPostmiles = new ArrayList<>();

        for(Station station : stations)

        {

            stationsPostmiles.add(station.postmile);

        }

        

        // Create expected station postmile list (sorted)

        ArrayList<Double> expectedStationsPostmiles = new ArrayList<>();

        expectedStationsPostmiles.add(new Double(0.9));

        expectedStationsPostmiles.add(new Double(1.26));

        expectedStationsPostmiles.add(new Double(1.49));



        for(int i = 0; i < 3; i++)

        {

            assertEquals(expectedStationsPostmiles.get(i), 

                    stationsPostmiles.get(i));

        }

        

        // Test for correct number of loops

        assertEquals(new Integer(12), new Integer(fiveS.stations.get(0).loops.size()

            + fiveS.stations.get(1).loops.size() + fiveS.stations.get(2).loops.size()));

    }

}