Sebastian IC week 4

lib/candidate.rb

@@ -1,2 +1,14 @@
 class Candidate
+    attr_reader :name, :party, :votes
+
+    def initialize(info_hash)
+        @name = info_hash[:name]
+        @party = info_hash[:party]
+
+        @votes = 0
+    end
+
+    def vote_for
+        @votes += 1
+    end
 end

lib/election.rb

@@ -1,2 +1,26 @@
 class Election
+    attr_reader :year, :races
+
+    def initialize(year)
+        @year = year
+        @races = []
+    end
+
+    def add_race(race)
+        @races << race
+    end
+
+    def candidates
+        @races.map { |race| race.candidates }.flatten
+    end
+
+    def vote_counts
+        votes = {}
+        candidates.each { |candidate| votes[candidate.name] = candidate.votes }
+        votes
+    end
+
+    def winners
+        @races.find_all { |race| !race.tie? && !race.open? }.map(&:winner)
+    end
 end

lib/race.rb

@@ -1,2 +1,35 @@
 class Race
+    attr_reader :office, :candidates
+
+    def initialize(office)
+        @office = office
+        @candidates = []
+        @open = true
+    end
+
+    def register_candidate(info)
+        candidate = Candidate.new(info)
+        @candidates << candidate
+        candidate
+    end
+
+    def open?
+        @open
+    end
+
+    def close!
+        @open = false
+    end
+
+    def winner
+        return false if @open
+
+        winners = @candidates.find_all { |candidate| candidate.votes == @candidates.max_by(&:votes).votes }
+
+        return winners[0]
+    end
+
+    def tie? 
+        @candidates.find_all { |candidate| candidate.votes == @candidates.max_by(&:votes).votes }.size > 1
+    end
 end

spec/candidate_spec.rb

@@ -0,0 +1,38 @@
+require 'rspec'
+require './lib/candidate'
+
+describe Candidate do
+    before do
+        @diana = Candidate.new({name: "Diana D", party: :democrat})
+    end
+
+    describe '#initialize' do
+        it 'exists' do
+            expect(@diana).to be_a(Candidate)
+        end
+
+        it 'has a name' do
+            expect(@diana.name).to eq("Diana D")
+        end
+
+        it 'has a party' do
+            expect(@diana.party).to eq(:democrat)
+        end
+
+        it 'starts with 0 votes' do
+            expect(@diana.votes).to eq(0)
+        end
+    end
+
+    describe '#vote_for' do
+        it 'can be voted for' do
+            3.times { @diana.vote_for }
+
+            expect(@diana.votes).to eq(3)
+
+            @diana.vote_for
+
+            expect(@diana.votes).to eq(4)
+        end
+    end
+end

spec/election_spec.rb

@@ -0,0 +1,80 @@
+require 'rspec'
+require './lib/candidate'
+require './lib/race'
+require './lib/election'
+
+describe Election do
+    before do
+        @election = Election.new(2024)
+
+        @race1 = Race.new("Texas Governor")
+        @race2 = Race.new("United States President")
+
+        @prez_candidate1 = @race2.register_candidate({name: "Joe B", party: :democrat})
+        @prez_candidate2 = @race2.register_candidate({name: "Donald T", party: :republican})
+
+        @candidate1 = @race1.register_candidate({name: "Diana D", party: :democrat})
+        @candidate2 = @race1.register_candidate({name: "Roberto R", party: :republican})
+    end
+
+    describe '#initialze' do
+        it 'exists' do
+            expect(@election).to be_a(Election)
+        end
+
+        it 'has a year' do
+            expect(@election.year).to eq(2024)
+        end
+
+        it 'begins with no races' do
+            expect(@election.races).to eq([])
+        end
+    end
+
+    describe '#add_race' do
+        it 'can add a race' do
+            expect(@election.add_race(@race1)).to eq([@race1])
+            expect(@election.add_race(@race2)).to eq([@race1, @race2])
+        end
+    end
+
+    describe '#candidates' do
+        it 'can list runnung candidates' do
+            @election.add_race(@race2)
+            @election.add_race(@race1)
+
+            expect(@election.candidates).to eq([@prez_candidate1, @prez_candidate2, @candidate1, @candidate2])
+        end
+    end
+
+    describe '#vote_counts' do
+        it 'can count votes' do
+            @election.add_race(@race2)
+            @election.add_race(@race1)
+
+            4.times { @election.candidates.each(&:vote_for) }
+
+            votes = @election.vote_counts
+
+            expect(votes["Donald T"]).to eq(4)
+            expect(votes["Joe B"]).to eq(4)
+            expect(votes["Diana D"]).to eq(4)
+            expect(votes["Roberto R"]).to eq(4)
+        end
+    end
+
+    describe '#winners' do
+        it 'can returns the winners of all races' do
+            @election.add_race(@race2)
+            @election.add_race(@race1)
+            @election.add_race(Race.new("race"))
+
+            @prez_candidate1.vote_for
+
+            @race1.close!
+            @race2.close!
+
+            expect(@election.winners).to eq([@prez_candidate1])
+        end
+    end
+end

spec/race_spec.rb

@@ -0,0 +1,91 @@
+require 'rspec'
+require './lib/race'
+require './lib/candidate'
+
+describe Race do
+    before do
+        @race = Race.new("Texas Governor")
+    end
+
+    describe '#initialze' do
+        it 'exists' do
+            expect(@race).to be_a(Race)
+        end
+
+        it 'has an office' do
+            expect(@race.office).to eq("Texas Governor")
+        end
+
+        it 'begins with no candidates' do
+            expect(@race.candidates).to eq([])
+        end
+
+        it 'begins open' do
+            expect(@race.open?).to eq(true)
+        end
+    end
+
+    describe '#register_candidate' do
+        it 'can register a candidate' do
+            candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+
+            expect(candidate1).to be_a(Candidate)
+            expect(candidate1.name).to eq("Diana D")
+            expect(candidate1.party).to eq(:democrat)
+        end
+
+        it 'can store several candidates' do
+            candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+            candidate2 = @race.register_candidate({name: "Roberto R", party: :republican})
+
+            expect(@race.candidates).to eq([candidate1, candidate2])
+        end
+    end
+
+    describe '#close!' do
+        it 'can close' do
+            @race.close!
+
+            expect(@race.open?).to eq(false)
+        end
+    end
+
+    describe '#winner' do
+        it 'returns false if race is open' do
+            expect(@race.winner).to eq(false)
+        end
+
+        it 'returns the winner of a closed race' do
+            candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+            candidate2 = @race.register_candidate({name: "Roberto R", party: :republican})
+
+            candidate1.vote_for
+            @race.close!
+
+            expect(@race.winner).to eq(candidate1)
+        end
+    end
+
+    describe '#tie?' do
+        it 'can determine if there is a tie' do
+            candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+            candidate2 = @race.register_candidate({name: "Roberto R", party: :republican})
+
+            candidate1.vote_for
+            candidate2.vote_for
+            @race.close!
+
+            expect(@race.tie?).to eq(true)
+        end
+
+        it 'can determine if there is not a tie' do
+            candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+            candidate2 = @race.register_candidate({name: "Roberto R", party: :republican})
+
+            candidate1.vote_for
+            @race.close!
+
+            expect(@race.tie?).to eq(false)
+        end
+    end
+end