Brenna's RideShare

.gitignore

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

Rakefile

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

lib/driver.rb

@@ -0,0 +1,41 @@
+require_relative 'rideshare'
+
+module RideShare
+
+  class Driver
+    attr_reader :id, :name, :vin
+
+    def initialize(id, name, vin)
+      raise ArgumentError.new("The ID is invalid.") if id.class != Integer
+      raise ArgumentError.new("The name is invalid.") if name.class != String
+      raise InvalidVinError.new("The VIN number is invalid.") if vin.length != 17 || !vin.upcase.match(/^[0-9A-Z]+$/)
+
+      @id = id.to_i
+      @name = name
+      @vin = vin.upcase
+    end
+
+
+    def self.all
+      CSV.read("support/drivers.csv", headers: true).map do | line |
+        RideShare::Driver.new(line[0].to_i, line[1], line[2])
+      end
+    end
+
+    def self.find(driver_id)
+      raise ArgumentError.new("The driver ID is invalid.") if driver_id.class != Integer
+      self.all.find { |driver| driver.id == driver_id }
+    end
+
+    def trips
+      RideShare::Trip.find_by_driver(@id)
+    end
+
+    def average_rating
+      all_ratings = trips.map { |trip| trip.rating }
+      all_ratings.inject(:+).to_f / all_ratings.length
+    end
+
+  end
+
+end

lib/invalidrating.rb

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

lib/invalidvin.rb

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

lib/rider.rb

@@ -0,0 +1,44 @@
+require_relative 'rideshare'
+
+module RideShare
+
+  class Rider
+
+    attr_reader :id, :name, :phone
+
+    def initialize(id, name, phone)
+      raise ArgumentError.new("The ID is invalid.") if id.class != Integer
+      raise ArgumentError.new("The name is invalid.") if name.class != String
+      raise ArgumentError.new("The phone number is invalid.") if phone.class != String
+
+      @id = id
+      @name = name
+      @phone = phone
+    end
+
+    def self.all
+      CSV.read("support/riders.csv", headers: true).map do | line |
+        RideShare::Rider.new(line[0].to_i, line[1], line[2])
+      end
+    end
+
+    def self.find(rider_id)
+      raise ArgumentError.new("The driver ID is invalid.") if rider_id.class != Integer
+      self.all.find { |rider| rider.id == rider_id }
+    end
+
+    def trips
+      RideShare::Trip.find_by_rider(@id)
+    end
+
+    def drivers
+      rider_driver_ids = trips.map { |trip| trip.driver_id }
+      rider_drivers = rider_driver_ids.map do |driver_id|
+        RideShare::Trip.find_by_driver(driver_id)
+      end
+      rider_drivers.uniq
+    end
+
+  end
+
+end

lib/rideshare.rb

@@ -0,0 +1,9 @@
+require 'csv'
+
+module RideShare; end
+
+require_relative 'invalidvin'
+require_relative 'invalidrating'
+require_relative 'trip'
+require_relative 'driver'
+require_relative 'rider'

lib/trip.rb

@@ -0,0 +1,51 @@
+require_relative 'rideshare'
+
+module RideShare
+
+  class Trip
+    attr_reader :id, :driver_id, :rider_id, :date, :rating
+
+    # could try to implement hash argument instead
+    def initialize(id, driver_id, rider_id, date, rating)
+      raise ArgumentError.new("The ID is invalid.") if id.class != Integer
+      raise ArgumentError.new("The driver ID is invalid.") if driver_id.class != Integer
+      raise ArgumentError.new("The rider ID is invalid.") if rider_id.class != Integer
+      raise ArgumentError.new("The date is invalid.") if date.class != String
+      raise InvalidRatingError.new("The rating is invalid.") if !(1..5).include?(rating)
+
+
+      @id = id.to_i
+      @driver_id = driver_id
+      @rider_id = rider_id
+      @date = date
+      @rating = rating.to_i
+    end
+
+    # could implement with hash
+    def self.all
+      CSV.read("support/trips.csv", headers: true).map do | line |
+        RideShare::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_by_driver(driver_id)
+      raise ArgumentError.new("The driver ID is invalid.") if driver_id.class != Integer
+      self.all.find_all { |trip| trip.driver_id == driver_id }
+    end
+
+    def self.find_by_rider(rider_id)
+      raise ArgumentError.new("The rider ID is invalid.") if rider_id.class != Integer
+      self.all.find_all { |trip| trip.rider_id == rider_id }
+    end
+
+    def find_driver
+      RideShare::Driver.find(@driver_id)
+    end
+
+    def find_rider
+      RideShare::Rider.find(@rider_id)
+    end
+
+  end
+
+end

specs/driver_spec.rb

@@ -0,0 +1,117 @@
+require_relative 'spec_helper'
+
+
+describe "Driver" do
+  let(:drivers) { RideShare::Driver.all }
+
+  describe "Driver#initialize" do
+    let(:vindiesel) { RideShare::Driver.new(777, "Vin Diesel", "FASTNFURIOUS00007") }
+
+    it "Driver has a name, ID, and VIN" do
+      vindiesel.id.must_equal 777
+      vindiesel.name.must_equal "Vin Diesel"
+      vindiesel.vin.must_equal "FASTNFURIOUS00007"
+    end
+
+    it "raises error if Driver's ID or name are invalid input" do
+      proc {
+        RideShare::Driver.new("VD", "Vin Diesel", "fastnfurious00007")
+      }.must_raise ArgumentError
+      proc {
+        RideShare::Driver.new(777, ["Vin Diesel"], "FASTNFURIOUS00007")
+      }.must_raise ArgumentError
+    end
+
+    it "raises error if VIN has invalid characters or wrong length" do
+      proc {
+        RideShare::Driver.new(777, "Vin Diesel", "FASTNFURIOUS@@!!7")
+      }.must_raise InvalidVinError
+      proc {
+        RideShare::Driver.new(777, "Vin Diesel", "FASTNFURIOUS007")
+      }.must_raise InvalidVinError
+    end
+
+    it "accepts lowercase letters in VIN" do
+      vindiesel = RideShare::Driver.new(777, "Vin Diesel", "fastnfurious00007")
+      vindiesel.vin.must_equal "FASTNFURIOUS00007"
+    end
+
+  end
+
+
+  describe "Driver#all" do
+
+    it "returns an array of Driver instances" do
+      drivers.must_be_kind_of Array
+      drivers.all? do | driver |
+        driver.must_be_instance_of RideShare::Driver
+      end
+    end
+
+    it "returns array with the correct number of Drivers from csv" do
+      drivers.length.must_equal 100
+    end
+
+    it "Driver instances match what's in the csv file" do
+      index = 0
+      CSV.read("support/drivers.csv", headers: true) do |line|
+        drivers[index].id.must_equal line[0].to_i
+        drivers[index].name.must_equal line[1]
+        drivers[index].vin.must_equal line[2]
+        index += 1
+      end
+    end
+
+  end
+
+
+  describe "Driver#find" do
+
+    it "raises an error if a non-integer is provided" do
+      proc {
+        RideShare::Driver.find("first driver")
+      }.must_raise ArgumentError
+    end
+
+    it "returns an instance of Driver" do
+      RideShare::Driver.find(1).must_be_instance_of RideShare::Driver
+    end
+
+    it "can find any driver based on a randomly generated number" do
+      10.times do
+        random_id = rand(1..100)
+        driver_check = drivers[random_id - 1].id
+        driver = RideShare::Driver.find(random_id)
+        expect(driver.id).must_equal driver_check
+      end
+    end
+
+    it "returns nil if account doesn't exist" do
+      RideShare::Driver.find(383).must_be_nil
+    end
+
+  end
+
+
+  describe "Driver#trips and #ratings" do
+    let(:shakira) { RideShare::Driver.new(16, "Shakira Stamm", "SALUVSAL3WA67SBPZ") }
+
+    it "returns an array whose length matches the number of Driver's trips" do
+      shakira.trips.must_be_kind_of Array
+      shakira.trips.length.must_equal RideShare::Trip.find_by_driver(16).length
+    end
+
+    it "returns an array of all Trip instances" do
+      shakira.trips.all? do | trip |
+        trip.must_be_instance_of RideShare::Trip
+      end
+    end
+
+    it "returns correct average rating for a Driver" do
+      # (2 + 5 + 1 + 2 + 4 + 1) / 6
+      shakira.average_rating.must_equal 2.5
+    end
+
+  end
+
+end