diff --git a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/APetHotel.java b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/APetHotel.java index 6a5c242b..a9b06f7b 100644 --- a/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/APetHotel.java +++ b/src/main/java/serenitylabs/tutorials/vetclinic/collections/katas/APetHotel.java @@ -11,6 +11,7 @@ * A utility class to generate pet hotels with pets already booked */ public class APetHotel { + public static PetAdder with(int petCount) { return new PetAdder(petCount); } @@ -38,5 +39,13 @@ private Breed someBreed() { private String someName(int petCount) { return PET_NAMES.get(random.nextInt(PET_NAMES.size())) + " " + petCount; } + + public PetHotel petsCheckIn() { + PetHotel petHotel= new PetHotel(); + for (int i=0;i pets= new TreeSet<>(Comparator.comparing(Pet::getName)); + public static int DEFAULT_MAXIMUM_CAPACITY = 20; + + public List getPets() { + return new ArrayList(pets); + } + + public BookingResponse checkIn(Pet pet) { + + if (pets.size()