From 358db26704b9095982131bc64c4bc8521a6e08f3 Mon Sep 17 00:00:00 2001
From: kiran1552 <kiran@1552>
Date: Tue, 20 Sep 2016 18:46:10 +0530
Subject: [PATCH] Kata/Collections/practice completed all test cases

---
 .../collections/katas/BookingResponse.java    |  4 ++
 .../vetclinic/collections/katas/PetHotel.java | 22 ++++++--
 .../katas/WhenBookingPetsIntoAPetHotel.java   | 51 ++++++++++++++++++-
 3 files changed, 73 insertions(+), 4 deletions(-)

diff --git a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/BookingResponse.java b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/BookingResponse.java
index 19e87674..2d4f3b57 100644
--- a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/BookingResponse.java
+++ b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/BookingResponse.java
@@ -14,4 +14,8 @@ public BookingResponse(boolean confirmedStatus) {
     public boolean isConfirmed() {
         return confirmedStatus;
     }
+
+    public boolean isOnWaitingList() {
+        return !confirmedStatus;
+    }
 }
diff --git a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/PetHotel.java b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/PetHotel.java
index 60363b06..a8b6ee73 100644
--- a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/PetHotel.java
+++ b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/PetHotel.java
@@ -11,15 +11,31 @@ public class PetHotel {
 
     private Collection<Pet> pets= new TreeSet<>(Comparator.comparing(Pet::getName));
     public static int DEFAULT_MAXIMUM_CAPACITY = 20;
+    private Queue<Pet> waitingList = new LinkedList<>();
 
     public List<Pet> getPets() {
         return new ArrayList(pets);
     }
 
     public BookingResponse checkIn(Pet pet) {
+        boolean confirmStatus= false;
+        if (pets.size() <DEFAULT_MAXIMUM_CAPACITY) {
+            pets.add(pet);
+            confirmStatus=true;
+        }
+        else
+            waitingList.add(pet);
+        return new BookingResponse(confirmStatus);
+    }
+
+    public List<Pet> getWaitingList() {
+        return new ArrayList<>(waitingList);
+    }
 
-        if (pets.size() <DEFAULT_MAXIMUM_CAPACITY)
-           pets.add(pet);
-        return new BookingResponse(true);
+    public void checkOut(Pet pet) {
+        pets.remove(pet);
+        if (!waitingList.isEmpty()) {
+            checkIn(waitingList.poll());
+        }
     }
 }
diff --git a/src/test/java/serenitylabs/tutorials/vetclinic/collections/katas/WhenBookingPetsIntoAPetHotel.java b/src/test/java/serenitylabs/tutorials/vetclinic/collections/katas/WhenBookingPetsIntoAPetHotel.java
index 3e696b87..5061c6c1 100644
--- a/src/test/java/serenitylabs/tutorials/vetclinic/collections/katas/WhenBookingPetsIntoAPetHotel.java
+++ b/src/test/java/serenitylabs/tutorials/vetclinic/collections/katas/WhenBookingPetsIntoAPetHotel.java
@@ -52,7 +52,7 @@ public void should_not_be_able_to_check_in_the_same_pet_twice() throws Exception
         PetHotel petHotel=new PetHotel();
         Pet fido= Pet.dog().named("Fido");
         Pet felix= Pet.cat().named("Felix");
-        
+
         petHotel.checkIn(fido);
         petHotel.checkIn(felix);
 
@@ -110,20 +110,69 @@ public void should_not_be_able_to_check_in_pets_beyond_hotel_capacity() throws E
 
     @Test
     public void should_notify_owner_that_the_hotel_is_full() throws Exception {
+
+        // Given
+        PetHotel petHotel=APetHotel.with(20).petsCheckIn();
+        Pet fido= Pet.dog().named("Fido");
+        // When
+        BookingResponse bookingResponse= petHotel.checkIn(fido);
+        // Then
+        assertThat(bookingResponse.isConfirmed(), is(false));
+        assertThat(bookingResponse.isOnWaitingList(), is(true));
+
+
     }
 
 
     @Test
     public void should_place_pets_on_a_waiting_list_when_the_hotel_is_full() throws Exception {
+
+        // Given
+        PetHotel petHotel=APetHotel.with(20).petsCheckIn();
+        Pet fido= Pet.dog().named("Fido");
+        // When
+         petHotel.checkIn(fido);
+        // Then
+        assertThat(petHotel.getWaitingList(), hasItem(fido));
+
     }
 
     @Test
     public void pets_on_the_waiting_list_should_be_added_to_the_hotel_when_a_place_is_freed() throws Exception {
+
+        // Given
+        PetHotel petHotel=APetHotel.with(19).petsCheckIn();
+        Pet fido= Pet.dog().named("Fido");
+        Pet felix= Pet.cat().named("Felix");
+        // Pet roger= Pet.rabbit().named("Roger");
+
+        petHotel.checkIn(fido);
+        petHotel.checkIn(felix);
+        // petHotel.checkIn(roger);
+        petHotel.checkOut(fido);
+        // Then
+        assertThat(petHotel.getPets(), hasItem(felix));
+
     }
 
 
     @Test
     public void pets_on_the_waiting_list_should_be_admitted_on_a_first_come_first_served_basis() throws Exception {
+
+        PetHotel hotel = APetHotel.with(19).petsCheckIn();
+        Pet felix = Pet.cat().named("FELIX");
+        Pet fido = Pet.dog().named("FIDO");
+        Pet roger = Pet.rabbit().named("ROGER");
+
+        hotel.checkIn(felix);
+        hotel.checkIn(fido);
+        hotel.checkIn(roger);
+
+        //WHEN
+        hotel.checkOut(felix);
+
+        //Then
+        assertThat(hotel.getPets(), hasItem(fido));
     }
 
 }