Sockets-Chantal

lib/hotel.rb

@@ -0,0 +1,50 @@
+require 'set'
+
+require_relative 'reservation'
+
+class Hotel
+  NUM_ROOMS = 20
+
+  # rooms 1-20, reservations
+  def initialize()
+    @reservations = []
+  end
+
+  # Access the list of all of the rooms in the hotel.
+  def get_rooms
+    (1..NUM_ROOMS).to_set
+  end
+
+  def make_reservation(start_date, end_date)
+    # Validate the date range.
+    validate_date_range(start_date, end_date)
+    # Find room that's available.
+    available_rooms = find_available_rooms(start_date, end_date)
+    raise RuntimeError.new("there are no rooms available") if available_rooms.empty?
+    room_number = available_rooms.to_a.sample
+    # Create reservation.
+    reservation = Reservation.new(room_number, start_date, end_date)
+    # Store the reservation.
+    @reservations << reservation
+    # Return the reservation.
+    return reservation
+  end
+
+  def find_available_rooms(start_date, end_date)
+    available_rooms = get_rooms
+    @reservations.each do |reservation|
+      if reservation.start_date < end_date && reservation.end_date > start_date
+        available_rooms.delete(reservation.room_number)
+      end
+    end
+    return available_rooms
+  end
+
+  def validate_date_range(start_date, end_date)
+    raise ArgumentError.new("the end date is before start date") unless end_date >= start_date
+  end
+
+  def get_reservations_inclusive(date)
+    @reservations.find_all {|reservation| reservation.end_date >= date && reservation.start_date <= date}
+  end
+end

lib/reservation.rb

@@ -0,0 +1,18 @@
+class Reservation
+  attr_reader :room_number, :start_date, :end_date
+
+  ROOM_PRICE = 200
+
+  #room association, date range
+  def initialize(room_number, start_date, end_date)
+    @room_number = room_number #integer
+    @start_date = start_date #time
+    @end_date = end_date #time
+  end
+
+  def calculate_total_cost()
+    total_nights = @end_date - @start_date #probably convert to integer after test
+    total_nights = total_nights.to_i / (24 * 60 * 60)
+    ROOM_PRICE * total_nights
+  end
+end

spec/hotel_spec.rb

@@ -0,0 +1,80 @@
+require_relative 'spec_helper'
+
+describe "hotel" do
+  NUM_BOOKED_ROOMS = Hotel::NUM_ROOMS - 5
+
+  before do
+    @seventh = Time.parse("2018-07-07")
+    @eleventh = Time.parse("2018-07-11")
+  end
+
+  let(:empty_hotel) do
+    empty_hotel = Hotel.new
+  end
+
+  let(:booked_hotel) do
+    booked_hotel = Hotel.new
+    (NUM_BOOKED_ROOMS).times do
+      booked_hotel.make_reservation(@seventh, @eleventh)
+    end
+    booked_hotel
+  end
+
+  let(:fully_booked_hotel) do
+    fully_booked_hotel = Hotel.new
+    Hotel::NUM_ROOMS.times do
+      fully_booked_hotel.make_reservation(@seventh, @eleventh)
+    end
+    fully_booked_hotel
+  end
+
+  it "get rooms" do
+    rooms = empty_hotel.get_rooms
+    expect(rooms.length).must_equal Hotel::NUM_ROOMS
+  end
+
+  describe "makes reservation" do
+    it "make reservation for hotel with no reservations" do
+      reservation = empty_hotel.make_reservation(@seventh, @eleventh)
+    end
+
+    it "make reservation for hotel with several reservations" do
+      reservation = booked_hotel.make_reservation(@seventh, @eleventh)
+    end
+
+    it "make reservation for fully booked hotel" do
+      expect{fully_booked_hotel.make_reservation(@seventh, @eleventh)}.must_raise RuntimeError
+    end
+
+    it "make reservation with invalid dates" do
+      expect{fully_booked_hotel.make_reservation(@eleventh, @seventh)}.must_raise ArgumentError
+    end
+  end
+
+  describe "finds available rooms" do
+    it "find available rooms when some are available" do
+      available_rooms = booked_hotel.find_available_rooms(@seventh, @eleventh)
+      expect(available_rooms.length).must_equal Hotel::NUM_ROOMS - NUM_BOOKED_ROOMS
+    end
+
+    it "find available rooms when none are available" do
+      available_rooms = fully_booked_hotel.find_available_rooms(@seventh, @eleventh)
+      expect(available_rooms.empty?).must_equal true
+    end
+  end
+
+  it "get reservations inclusive" do
+    inclusive_reservations = booked_hotel.get_reservations_inclusive(@eleventh)
+    expect(inclusive_reservations.length).must_equal NUM_BOOKED_ROOMS
+  end
+
+  describe 'validate date range' do
+    it 'invalid date range throws exception' do
+      expect{empty_hotel.validate_date_range(@eleventh, @seventh)}.must_raise ArgumentError
+    end
+
+    it "valid date range throws no exception" do
+      empty_hotel.validate_date_range(@seventh, @eleventh)
+    end
+  end
+end

spec/reservation_spec.rb

@@ -0,0 +1,8 @@
+require_relative 'spec_helper'
+
+describe "reservation" do
+  it 'calculate_total_cost' do
+    reservation = Reservation.new(2, Time.parse("2018-07-23"), Time.parse("2018-07-28"))
+    expect(reservation.calculate_total_cost).must_equal 1000
+  end
+end

spec/spec_helper.rb

@@ -6,3 +6,7 @@
 Minitest::Reporters.use! Minitest::Reporters::SpecReporter.new
 
 # Require_relative your lib files here!
+require 'time'
+
+require_relative '../lib/hotel'
+require_relative '../lib/reservation'