DataHandler updated
[staff/due1/sed-hs15-srs-purple.git] / src / main / java / ch / bfh / ti / srs / datahandler / DataHandlerForTests.java
index dcd7579..65199a6 100644 (file)
@@ -1,41 +1,87 @@
-/*\r
- * Copyright (c) 2015 Berner Fachhochschule, Switzerland.\r
- *\r
- * Project Smart Reservation System.\r
- *\r
- * Distributable under GPL license. See terms of license at gnu.org.\r
- */\r
-package ch.bfh.ti.srs.datahandler;\r
-\r
-import java.util.HashSet;\r
-import java.util.Set;\r
-\r
-import ch.bfh.ti.srs.dataschema.User;\r
-\r
-public class DataHandlerForTests implements DataHandler{\r
-       Set<User> userList = new HashSet<User>();\r
-       \r
-       public DataHandlerForTests(){\r
-               \r
-       }\r
-\r
-       /* (non-Javadoc)\r
-        * @see ch.bfh.ti.srs.core.DataHandler#findUser(java.lang.String)\r
-        */\r
-       @Override\r
-       public User findUser(String userName) {\r
-               for (User user : userList) {\r
-                       if(user.getUserName().equals(userName)) return user;\r
-               }\r
-               return null;\r
-       }\r
-\r
-       /* (non-Javadoc)\r
-        * @see ch.bfh.ti.srs.core.DataHandler#addUser(ch.bfh.ti.srs.core.User)\r
-        */\r
-       @Override\r
-       public void addUser(User user) throws Exception {\r
-               if(user == null) throw new IllegalArgumentException("User object must not be null!");\r
-               userList.add(user);\r
-       }\r
-}\r
+/*
+ * Copyright (c) 2015 Berner Fachhochschule, Switzerland.
+ *
+ * Project Smart Reservation System.
+ *
+ * Distributable under GPL license. See terms of license at gnu.org.
+ */
+package ch.bfh.ti.srs.datahandler;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import ch.bfh.ti.daterange.DateRange;
+import ch.bfh.ti.srs.dataschema.Reservation;
+import ch.bfh.ti.srs.dataschema.Resource;
+import ch.bfh.ti.srs.dataschema.User;
+
+public class DataHandlerForTests implements DataHandler{
+       Set<User> userList = new HashSet<User>();
+       Set<Resource> resourceList = new HashSet<Resource>();
+       Set<Reservation> reservationList = new HashSet<Reservation>();
+       
+       public DataHandlerForTests(){
+               
+       }
+
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#findUser(java.lang.String)
+        */
+       @Override
+       public User findUser(String userName) {
+               for (User user : userList) {
+                       if(user.getUserName().equals(userName)) return user;
+               }
+               return null;
+       }
+
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#addUser(ch.bfh.ti.srs.dataschema.User)
+        */
+       @Override
+       public void addUser(User user) throws Exception {
+               if(user == null) throw new IllegalArgumentException("User object must not be null!");
+               if(userList.contains(user)) throw new IllegalArgumentException("User already  exists!");
+               userList.add(user);
+       }
+
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#findResource(java.lang.String)
+        */
+       @Override
+       public Resource findResource(String resourceName) {
+               for (Resource resource : resourceList) {
+                       if(resource.getName().equals(resourceName)) return resource;
+               }
+               return null;
+       }
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#addResource(ch.bfh.ti.srs.dataschema.Resource)
+        */
+       @Override
+       public void addResource(Resource resource) throws Exception {
+               if(resource == null) throw new IllegalArgumentException("Resource object must not be null!");
+               if(resourceList.contains(resource)) throw new IllegalArgumentException("Resource already  exists!");
+               resourceList.add(resource);
+       }
+
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#findReservation(java.lang.String)
+        */
+       @Override
+       public Reservation findReservation(Resource resource, DateRange daterange) {
+               for (Reservation reservation : reservationList) {
+                       if(reservation.getResource().equals(resource) && reservation.getDateRange().equals(daterange)) return reservation;
+               }
+               return null;
+       }
+       /* (non-Javadoc)
+        * @see ch.bfh.ti.srs.dataschema.DataHandler#addReservation(ch.bfh.ti.srs.dataschema.Reservation)
+        */
+       @Override
+       public void addReservation(Reservation reservation) throws Exception {
+               if(reservation == null) throw new IllegalArgumentException("Reservation object must not be null!");
+               if(resourceList.contains(reservation)) throw new IllegalArgumentException("Reservation already  exists!");
+               reservationList.add(reservation);       
+       }
+}