Unit tests for stop and route

src/test/java/model/DataManagerTest.java

@@ -4,12 +4,13 @@ package model;/**
 
 import static org.junit.Assert.*;
 
+import javafx.collections.FXCollections;
 import org.junit.Test;
 
 public class DataManagerTest {
 
     @Test
-    public void AddRide(){
+    public void addRide(){
         DataManager dataManager = new DataManager();
         Ride ride = new Ride("Nissan March", "Blue", "HB9999", 2008, 4);
         dataManager.addRides(ride);
@@ -23,4 +24,43 @@ public class DataManagerTest {
         assertTrue(dataManager.getRides().size() == 0);
     }
 
+    @Test
+    public void addStop(){
+        DataManager dataManager = new DataManager();
+        Stop stop1 = new Stop("Stop 1");
+        Stop stop2 = new Stop("Stop 2");
+        Stop stop3 = new Stop("Stop 1");
+        dataManager.addStop(stop1);
+        assertTrue(dataManager.getStops().size() == 1);
+        assertTrue(!dataManager.addStop(stop1));
+        assertTrue(!dataManager.addStop(stop3));
+        assertTrue(dataManager.addStop(stop2));
+        assertTrue(dataManager.getStops().size() == 2);
+    }
+
+    @Test
+    public void addRoute(){
+        DataManager dataManager = new DataManager();
+        Stop stop1 = new Stop("Stop 1");
+        Stop stop2 = new Stop("Stop 2");
+        Stop stop3 = new Stop("Stop 3");
+        Stop stop4 = new Stop("Stop 4");
+        Stop stop5 = new Stop("Stop 5");
+        Stop[] stops = {stop1, stop2, stop3, stop4, stop5};
+        Route route = new Route("Route1", FXCollections.observableArrayList(stops));
+        dataManager.addRoute(route);
+        assertTrue(dataManager.getRoutes().size() == 1);
+        //does not add duplicate routes
+        assertTrue(!dataManager.addRoute(route));
+        assertTrue(dataManager.getRoutes().size() == 1);
+        Route route2 = new Route("Route 2", FXCollections.observableArrayList(stops));
+        assertTrue(!dataManager.addRoute(route2));
+        assertTrue(dataManager.getRoutes().size() == 1);
+        //can acutally add another route
+        Stop[] stops1 = {stop1, stop2};
+        Route route3 = new Route("Route 3", FXCollections.observableArrayList(stops1));
+        assertTrue(dataManager.addRoute(route3));
+        assertTrue(dataManager.getRoutes().size() == 2);
+    }
+
 }