Water - Mackenzie and Jessica

lib/driver.rb

@@ -0,0 +1,68 @@
+require 'csv'
+require 'time'
+
+require_relative 'csv_record'
+
+module RideShare
+  class Driver < CsvRecord
+    attr_reader :id, :name, :vin, :status, :trips
+
+    def initialize(id:, name:, vin:, status: :AVAILABLE, trips: [])
+      raise ArgumentError.new("Invalid VIN length") unless vin.length == 17
+      raise ArgumentError.new("Invalid status") unless [:AVAILABLE, :UNAVAILABLE].include?(status.to_sym)
+
+      super(id)
+
+      @name = name
+      @vin = vin
+      @status = status.to_sym
+      @trips = trips
+    end
+
+    def add_trip(trip)
+      @trips << trip
+    end
+
+    def average_rating
+      all_ratings = @trips.map { |trip| trip.rating }
+
+      if all_ratings.length == 0
+        return 0
+      else
+        average = all_ratings.compact.sum.to_f / all_ratings.compact.length
+        return average
+      end
+
+    end
+
+    def total_revenue
+      all_cost = @trips.map do |trip|
+        case trip.cost
+        when nil
+          nil
+        when (0...1.65)
+          0
+        else
+          (trip.cost - 1.65) * 0.8
+        end
+      end
+
+      return all_cost.compact.sum
+    end
+
+    def make_unavailable
+      @status = :UNAVAILABLE
+    end
+
+    private
+
+    def self.from_csv(record)
+      return new(
+          id: record[:id],
+          name: record[:name],
+          vin: record[:vin],
+          status: record[:status]
+      )
+    end
+  end
+end

lib/passenger.rb

@@ -1,4 +1,5 @@
 require_relative 'csv_record'
+#require_relative 'trip'
 
 module RideShare
   class Passenger < CsvRecord
@@ -16,6 +17,17 @@ def add_trip(trip)
       @trips << trip
     end
 
+    def net_expenditures
+      all_cost = @trips.map { |trip| trip.cost}
+      return all_cost.compact.sum
+    end
+
+    def total_time_spent
+      all_times = @trips.map {|trip| trip.duration}
+      return all_times.compact.sum
+    end
+
+
     private
 
     def self.from_csv(record)

lib/trip.rb

@@ -1,22 +1,37 @@
 require 'csv'
+require 'time'
 
 require_relative 'csv_record'
+require_relative 'driver'
 
 module RideShare
   class Trip < CsvRecord
-    attr_reader :id, :passenger, :passenger_id, :start_time, :end_time, :cost, :rating
+    attr_reader :id, :driver, :driver_id, :passenger, :passenger_id, :start_time, :end_time, :cost, :rating
 
     def initialize(
           id:,
+          driver: nil,
+          driver_id: nil,
           passenger: nil,
           passenger_id: nil,
           start_time:,
-          end_time:,
+          end_time: nil,
           cost: nil,
-          rating:
+          rating: nil
         )
       super(id)
 
+      if driver
+        @driver = driver
+        @driver_id = driver.id
+
+      elsif driver_id
+        @driver_id = driver_id
+
+      else
+        raise ArgumentError, 'Driver or driver_id is required'
+      end
+
       if passenger
         @passenger = passenger
         @passenger_id = passenger.id
@@ -33,8 +48,16 @@ def initialize(
       @cost = cost
       @rating = rating
 
-      if @rating > 5 || @rating < 1
-        raise ArgumentError.new("Invalid rating #{@rating}")
+      unless @rating == nil
+        if @rating > 5 || @rating < 1
+          raise ArgumentError.new("Invalid rating #{@rating}")
+        end
+      end
+
+      unless end_time == nil
+        if end_time < start_time
+          raise ArgumentError.new("End time cannot be before start time.")
+        end
       end
     end
 
@@ -43,26 +66,34 @@ def inspect
       # trip contains a passenger contains a trip contains a passenger...
       "#<#{self.class.name}:0x#{self.object_id.to_s(16)} " +
         "id=#{id.inspect} " +
+          "driver_id=#{driver&.id.inspect} " +
         "passenger_id=#{passenger&.id.inspect} " +
         "start_time=#{start_time} " +
         "end_time=#{end_time} " +
         "cost=#{cost} " +
         "rating=#{rating}>"
     end
 
-    def connect(passenger)
+    def connect(passenger, driver)
       @passenger = passenger
+      @driver = driver
       passenger.add_trip(self)
+      driver.add_trip(self)
+    end
+
+    def duration
+      @end_time == nil ? nil : (@end_time - @start_time)
     end
 
     private
 
     def self.from_csv(record)
       return self.new(
                id: record[:id],
+               driver_id: record[:driver_id],
                passenger_id: record[:passenger_id],
-               start_time: record[:start_time],
-               end_time: record[:end_time],
+               start_time: Time.parse(record[:start_time]),
+               end_time: Time.parse(record[:end_time]),
                cost: record[:cost],
                rating: record[:rating]
              )

lib/trip_dispatcher.rb

@@ -3,6 +3,7 @@
 
 require_relative 'passenger'
 require_relative 'trip'
+require_relative 'driver'
 
 module RideShare
   class TripDispatcher
@@ -11,6 +12,7 @@ class TripDispatcher
     def initialize(directory: './support')
       @passengers = Passenger.load_all(directory: directory)
       @trips = Trip.load_all(directory: directory)
+      @drivers = Driver.load_all(directory: directory)
       connect_trips
     end
 
@@ -19,6 +21,11 @@ def find_passenger(id)
       return @passengers.find { |passenger| passenger.id == id }
     end
 
+    def find_driver(id)
+      Driver.validate_id(id)
+      return @drivers.find { |driver| driver.id == id}
+    end
+
     def inspect
       # Make puts output more useful
       return "#<#{self.class.name}:0x#{object_id.to_s(16)} \
@@ -27,12 +34,50 @@ def inspect
               #{passengers.count} passengers>"
     end
 
+    def available_driver
+      available_drivers = []
+      @drivers.each do |driver|
+        if driver.status == :AVAILABLE
+          available_drivers << driver
+        end
+      end
+      return available_drivers
+    end
+
+    def trip_id_generator
+      return @trips.last.id.to_i + 1
+    end
+
+    def request_trip(passenger_id)
+      driver = available_driver[0]
+      passenger = find_passenger(passenger_id)
+
+      if driver == nil
+        raise ArgumentError.new("No available drivers")
+      end
+
+      new_trip = Trip.new(
+          id: trip_id_generator,
+          driver_id: driver.id,
+          passenger_id: passenger_id,
+          start_time: Time.now,
+          )
+      driver.make_unavailable
+
+      new_trip.connect(passenger, driver)
+
+      @trips << new_trip
+
+      return new_trip
+    end
+
     private
 
     def connect_trips
       @trips.each do |trip|
         passenger = find_passenger(trip.passenger_id)
-        trip.connect(passenger)
+        driver = find_driver(trip.driver_id)
+        trip.connect(passenger, driver)
       end
 
       return trips

test/driver_test.rb

@@ -1,6 +1,6 @@
 require_relative 'test_helper'
 
-xdescribe "Driver class" do
+describe "Driver class" do
   describe "Driver instantiation" do
     before do
       @driver = RideShare::Driver.new(
@@ -128,9 +128,70 @@
 
       expect(@driver.average_rating).must_be_close_to (5.0 + 1.0) / 2.0, 0.01
     end
+
+    it"ignores trips in progress" do
+      trip = RideShare::Trip.new(
+          id: 8,
+          driver: @driver,
+          passenger_id: 3,
+          start_time: Time.now,
+          end_time: nil,
+          rating: nil
+      )
+      @driver.add_trip(trip)
+      expect(@driver.average_rating).must_equal 5
+    end
   end
 
   describe "total_revenue" do
     # You add tests for the total_revenue method
+    before do
+      @driver = RideShare::Driver.new(
+          id: 54,
+          name: "Rogers Bartell IV",
+          vin: "1C9EVBRM0YBC564DZ"
+      )
+      trip = RideShare::Trip.new(
+          id: 8,
+          driver: @driver,
+          passenger_id: 3,
+          start_time: Time.new(2016, 8, 8),
+          end_time: Time.new(2016, 8, 8),
+          cost: 11.65,
+          rating: 5
+      )
+      @driver.add_trip(trip)
+    end
+
+    it "calculates accurate revenue" do
+      expect(@driver.total_revenue).must_be_close_to 8.0
+    end
+
+    it "returns 0 for trips less than 1.65" do
+      trip = RideShare::Trip.new(
+          id:29,
+          driver: @driver,
+          passenger_id: 94,
+          start_time: Time.new(2016, 8, 8),
+          end_time: Time.new(2016, 8, 8),
+          cost: 1.29,
+          rating: 4
+      )
+      expect(@driver.total_revenue).must_be_close_to 8
+    end
+
+    it "ignores trips in progress" do
+      trip = RideShare::Trip.new(
+          id: 9,
+          driver: @driver,
+          passenger_id: 2,
+          start_time: Time.new(2016, 8, 8),
+          end_time: nil,
+          cost: nil,
+          rating: nil
+      )
+      @driver.add_trip(trip)
+      expect(@driver.total_revenue).must_equal 8.0
+    end
   end
 end