DataHandler angepasst
[staff/due1/sed-hs15-srs-purple.git] / src / main / java / ch / bfh / ti / srs / datahandler / DataHandlerForTests.java
index 65199a6..78d9126 100644 (file)
  */
 package ch.bfh.ti.srs.datahandler;
 
+import java.util.ArrayList;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
+import java.util.UUID;
 
 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(){
-               
+public class DataHandlerForTests implements DataHandler {
+       private Set<User> userList = new HashSet<User>();
+       private Set<Resource> resourceList = new HashSet<Resource>();
+       private Set<Reservation> reservationList = new HashSet<Reservation>();
+
+       public DataHandlerForTests() {
+
        }
 
-       /* (non-Javadoc)
-        * @see ch.bfh.ti.srs.dataschema.DataHandler#findUser(java.lang.String)
+       /**
+        * {@inheritDoc}
         */
        @Override
        public User findUser(String userName) {
                for (User user : userList) {
-                       if(user.getUserName().equals(userName)) return user;
+                       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)
+       /**
+        * {@inheritDoc}
         */
        @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!");
+               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)
+       /**
+        * {@inheritDoc}
         */
        @Override
        public Resource findResource(String resourceName) {
                for (Resource resource : resourceList) {
-                       if(resource.getName().equals(resourceName)) return resource;
+                       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)
+
+       /**
+        * {@inheritDoc}
         */
        @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!");
+               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)
+       /**
+        * {@inheritDoc}
         */
        @Override
-       public Reservation findReservation(Resource resource, DateRange daterange) {
+       public Reservation findReservation(UUID id) {
                for (Reservation reservation : reservationList) {
-                       if(reservation.getResource().equals(resource) && reservation.getDateRange().equals(daterange)) return reservation;
+                       if (reservation.getId().equals(id))
+                               return reservation;
                }
                return null;
        }
-       /* (non-Javadoc)
-        * @see ch.bfh.ti.srs.dataschema.DataHandler#addReservation(ch.bfh.ti.srs.dataschema.Reservation)
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public List<Reservation> findGroupReservations(UUID groupId) {
+               if (groupId == null)
+                       throw new IllegalArgumentException("DateRange object must not be null!");
+               List<Reservation> groupReservations = new ArrayList<Reservation>();
+               for (Reservation reservation : reservationList) {
+                       if (reservation.getGroupId().equals(groupId))
+                               groupReservations.add(reservation);
+               }
+               return groupReservations;
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public List<Reservation> userReservations(User user) {
+               if (user == null)
+                       throw new IllegalArgumentException("User object must not be null!");
+               List<Reservation> allUserReservations = new ArrayList<Reservation>();
+               for (Reservation reservation : reservationList) {
+                       if (reservation.getUser().equals(user))
+                               allUserReservations.add(reservation);
+               }
+               return allUserReservations;
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public List<Reservation> overlaps(DateRange range) {
+               if (range == null)
+                       throw new IllegalArgumentException("DateRange object must not be null!");
+               List<Reservation> overlappingReservations = new ArrayList<Reservation>();
+               for (Reservation reservation : reservationList) {
+                       if (reservation.getDateRange().overlaps(range))
+                               overlappingReservations.add(reservation);
+               }
+               return overlappingReservations;
+       }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public List<Reservation> findAllReservations() {
+               List<Reservation> allReservations = new ArrayList<Reservation>();
+               for (Reservation reservation : reservationList) {
+                       allReservations.add(reservation);
+               }
+               return allReservations;
+       }
+
+       /**
+        * {@inheritDoc}
         */
        @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);       
+               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);
        }
+
 }