Alena's Ride Share

.gitignore

@@ -0,0 +1,2 @@
+/coverage/
+/specs/coverage/

Rakefile

@@ -1 +1,10 @@
 # Fill me in!
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs = ["lib"]
+  t.warning = true
+  t.test_files = FileList['specs/*_spec.rb']
+end
+
+task default: :test

lib/InvalidIDError.rb

@@ -0,0 +1,2 @@
+class InvalidIDError < StandardError
+end

lib/InvalidRatingError.rb

@@ -0,0 +1,2 @@
+class InvalidRatingError < StandardError
+end

lib/InvalidVinError.rb

@@ -0,0 +1,2 @@
+class InvalidVinError < StandardError
+end

lib/driver.rb

@@ -0,0 +1,48 @@
+require 'csv'
+require_relative 'trip'
+require_relative 'InvalidVinError'
+require_relative 'InvalidIDError'
+
+class Driver
+  attr_reader :id, :name, :vin
+
+  @@all_drivers = nil
+
+  def initialize(id, name, vin)
+    @id = id
+    @name = name
+
+    if vin.length == 17
+      @vin = vin
+    else
+      raise InvalidVinError.new("vin must be 17 characters long")
+    end
+  end
+
+  def self.all
+    return @@all_drivers if @@all_drivers
+    drivers = CSV.read("support/drivers.csv", { :headers => true })
+
+    @@all_drivers = drivers.map { |line| Driver.new(line[0].to_i, line[1], line[2]) }
+  end
+
+  def self.find(driver_id)
+    unless (driver_id.is_a? Integer) && driver_id >= 0
+      raise InvalidIDError.new("driver ID:#{driver_id} is not valid")
+    end
+
+    Driver.all.find { |driver| driver.id == driver_id }
+  end
+
+  def list_trips
+    Trip.find_trips_driver(@id)
+  end
+
+  def avg_rating
+    ratings = list_trips.map { |trip| trip.rating.to_f }
+    return 0 if ratings == []
+    rating = (ratings.reduce(:+)) / ratings.length
+    rating.round(2)
+  end
+
+end

lib/rider.rb

@@ -0,0 +1,42 @@
+require 'csv'
+require_relative 'trip'
+require_relative 'driver'
+require_relative 'InvalidIDError'
+
+class Rider
+  attr_reader :id, :name, :phone_num
+
+  @@all_riders = nil
+
+  def initialize(id, name, phone_num)
+    @id = id
+    @name = name
+    @phone_num = phone_num
+  end
+
+  def self.all
+    return @@all_riders if @@all_riders
+    riders = CSV.read("support/riders.csv", { :headers => true })
+
+    @@all_riders = riders.map { |line| Rider.new(line[0].to_i, line[1], line[2]) }
+  end
+
+  def self.find(rider_id)
+    unless (rider_id.is_a? Integer) && rider_id >= 0
+      raise InvalidIDError.new("rider ID:#{rider_id} is not valid")
+    end
+
+    Rider.all.find { |rider| rider.id == rider_id }
+  end
+
+  def list_trips
+    Trip.find_trips_rider(@id)
+  end
+
+  def list_drivers
+    drivers = list_trips.map { |trip| Driver.find(trip.driver_id) }
+
+    drivers.uniq { |driver| driver.id }
+  end
+
+end

lib/trip.rb

@@ -0,0 +1,47 @@
+require 'csv'
+require_relative 'InvalidRatingError'
+
+class Trip
+  attr_reader :id, :driver_id, :rider_id, :date, :rating
+
+  @@all_trips = nil
+
+  def initialize(id, driver_id, rider_id, date, rating)
+    @id = id
+    @driver_id = driver_id
+    @rider_id = rider_id
+    @date = date
+
+    unless (rating.is_a? Integer) && (1..5).include?(rating)
+      raise InvalidRatingError.new("rating must be an integer within range 1-5")
+    else
+      @rating = rating
+    end
+  end
+
+  def self.all
+    return @@all_trips if @@all_trips
+    trips = CSV.read("support/trips.csv", { :headers => true })
+
+    @@all_trips = trips.map do |line|
+      Trip.new(line[0].to_i, line[1].to_i, line[2].to_i, line[3], line[4].to_i)
+    end
+  end
+
+  def self.find_trips_driver(driver_id)
+    Trip.all.select { |trip| trip.driver_id == driver_id }
+  end
+
+  def self.find_trips_rider(rider_id)
+    Trip.all.select { |trip| trip.rider_id == rider_id }
+  end
+
+  def driver
+    Driver.find(@driver_id)
+  end
+
+  def rider
+    Rider.find(@rider_id)
+  end
+
+end

specs/driver_spec.rb

@@ -0,0 +1,137 @@
+require_relative 'spec_helper'
+
+describe "Driver" do
+
+  describe "#initialize" do
+    it "Takes an ID, name, and vin" do
+      name = "Bob Belcher"
+      id = 12345
+      vin = "9e87e65r98302e984"
+      driver = Driver.new(id, name, vin)
+
+      driver.must_respond_to :id
+      driver.id.must_equal id
+
+      driver.must_respond_to :name
+      driver.name.must_equal name
+
+      driver.must_respond_to :vin
+      driver.vin.must_equal vin
+    end
+
+    it "Is a kind of Driver" do
+      name = "Bob Belcher"
+      id = 12345
+      vin = "9e87e65r98302e984"
+      driver = Driver.new(id, name, vin)
+
+      driver.must_be_kind_of Driver
+    end
+
+    it "Doesn't create a driver if vin is not 17 characters long" do
+      proc {
+        Driver.new(123, "Tina Belcher", "w3475rs78t")
+      }.must_raise InvalidVinError
+    end
+  end
+
+  describe "Driver.all" do
+
+    before do
+      @drivers = Driver.all
+    end
+
+    it "Returns an array of all drivers" do
+      @drivers.class.must_equal Array
+      @drivers.each { |driver| driver.must_be_instance_of Driver }
+      @drivers.length.must_equal 100
+
+      @drivers.first.id.must_equal 1
+      @drivers[0].name.must_equal "Bernardo Prosacco"
+      @drivers.first.vin.must_equal "WBWSS52P9NEYLVDE9"
+
+      @drivers.last.id.must_equal 100
+      @drivers[-1].name.must_equal "Minnie Dach"
+      @drivers.last.vin.must_equal "XF9Z0ST7X18WD41HT"
+
+      # index = 0
+      # CSV.read("support/drivers.csv", { :headers => true }).each do |line|
+      #
+      #   @drivers[index].id.must_equal line[0].to_i
+      #   @drivers[index].name.must_equal line[1].to_s
+      #   @drivers[index].vin.must_equal line[2]
+      #   index += 1
+      # end
+    end
+
+  end
+
+  describe "Driver.find" do
+
+    before do
+      @drivers = Driver.all
+    end
+
+    it "Returns a driver that exists" do
+      driver = Driver.find(54)
+      driver.must_be_instance_of Driver
+    end
+
+    it "Can find the first driver from the CSV" do
+      driver = Driver.find(1)
+      driver.name.must_equal @drivers.first.name
+    end
+
+    it "Can find the last driver from the CSV" do
+      driver = Driver.find(100)
+      driver.name.must_equal @drivers.last.name
+    end
+
+    it "Returns nil for a driver that doesn't exist" do
+      driver = Driver.find(101)
+      driver.must_be_nil
+    end
+
+    it "Raises an error for an invalid driver ID" do
+      proc {
+        Driver.find("a")
+      }.must_raise InvalidIDError
+    end
+  end
+
+  describe "Driver#list_trips" do
+    it "Returns a list of trips for a specific driver" do
+      driver = Driver.new(36, "Mr. Marcelina Jenkins", "WD3VLLK2X04HF50PL")
+      driver.list_trips.must_be_kind_of Array
+      driver.list_trips.each { |trip| trip.must_be_kind_of Trip }
+    end
+
+    it "Returns a correct number of trips for a specific driver" do
+      driver = Driver.new(1, "Bernardo Prosacco",	"WBWSS52P9NEYLVDE9")
+      driver.list_trips.length.must_equal 9
+    end
+
+    it "Returns an empty array if there are no trips for that driver" do
+      driver = Driver.new(100, "Minnie Dach",	"XF9Z0ST7X18WD41HT")
+      driver.list_trips.must_equal []
+    end
+  end
+
+  describe "Driver#avg_rating" do
+    it "Returns a number" do
+      driver = Driver.new(60, "Oma Swift DDS", "TAMCBRPM7EN5GD88L")
+      driver.avg_rating.must_be_kind_of Float
+    end
+
+    it "Returns correct average for a specific driver" do
+      driver = Driver.new(1, "Bernardo Prosacco",	"WBWSS52P9NEYLVDE9")
+      driver.avg_rating.must_equal 2.33
+    end
+
+    it "Returns 0 if there are no trips for this driver" do
+      driver = Driver.new(100, "Minnie Dach",	"XF9Z0ST7X18WD41HT")
+      driver.avg_rating.must_equal 0
+    end
+  end
+
+end