Zion Mekonnen

lib/candidate.rb

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

lib/election.rb

@@ -1,2 +1,16 @@
 class Election
+    attr_reader :year, :races
+    def initialize(year)
+        @year = year
+        @races = []
+    end
+    def add_race(race)
+        @races << race
+    end
+
+    def candidates 
+        @races.flat_map do |race|
+            race.candidates
+        end
+    end
 end

lib/race.rb

@@ -1,2 +1,14 @@
 class Race
+    attr_reader :position_in_office, :candidates
+
+    def initialize(position_in_office)
+        @position_in_office = position_in_office
+        @candidates = []
+    end
+
+    def register_candidate(candidate_info)
+        candidate = Candidate.new(candidate_info)
+        @candidates << candidate
+        candidate
+    end
 end

spec/candidate_spec.rb

@@ -0,0 +1,25 @@
+require './lib/candidate'
+require 'rspec'
+
+RSpec.describe Candidate do
+    before (:each) do
+        @diana = Candidate.new({name: "Diana D", party: :democrat})
+    end
+
+    it 'returns candidates name and party' do
+        expect(@diana.name).to eq("Diana D")
+        expect(@diana.party).to eq(:democrat)
+    end
+
+    it 'can calculate the number of votes' do
+        expect(@diana.votes).to eq(0)
+
+        @diana.vote_for
+        @diana.vote_for
+        @diana.vote_for
+        expect(@diana.votes).to eq(3)
+
+        @diana.vote_for
+        expect(@diana.votes).to eq(4)
+    end
+end

spec/election_spec.rb

@@ -0,0 +1,35 @@
+require './lib/candidate'
+require './lib/race'
+require './lib/election' 
+require 'rspec'
+
+RSpec.describe Election do
+    before (:each) do
+        @diana = Candidate.new({name: "Diana D", party: :democrat})
+        @roberto = Candidate.new({ name: "Roberto R", party: :republican })
+        @race = Race.new("Texas Governor")
+        @race.register_candidate({ name: "Diana D", party: :democrat })
+        @race.register_candidate({ name: "Roberto R", party: :republican })
+        @election = Election.new("2025")
+    end
+
+    it 'initializes' do
+        expect(@election.year).to eq("2025")
+        expect(@election.races).to eq([])
+    end
+
+    it 'adds races to the election' do
+        @election.add_race(@race)
+        expect(@election.races).to include(@race)
+    end
+
+    it "can list the candidates in the election" do
+        @election.add_race(@race)
+
+        candidate_names = @election.candidates.map do |candidate| 
+            candidate.name
+        end
+        expect(candidate_names).to include("Diana D", "Roberto R")
+      end
+
+end

spec/race_spec.rb

@@ -0,0 +1,32 @@
+require './lib/candidate'
+require './lib/race'
+require 'rspec'
+
+RSpec.describe Race do
+    before (:each) do
+        @diana = Candidate.new({name: "Diana D", party: :democrat})
+        @race = Race.new("Texas Governor")
+    end
+
+    it 'returns candidate position' do
+        expect(@race.position_in_office).to eq("Texas Governor")
+
+    end
+
+    it 'returns [] for candidates' do
+        expect(@race.candidates).to eq([])
+    end
+
+    it 'returns the candidates info and that they are in the race, once they have been registered' do 
+        candidate1 = @race.register_candidate({name: "Diana D", party: :democrat})
+        expect(candidate1.class).to be(Candidate)
+        expect(candidate1.name).to eq("Diana D")
+        expect(candidate1.party).to eq(:democrat)
+
+        candidate2 = @race.register_candidate({name: "Roberto R", party: :republican})
+        expect(@race.candidates).to include(candidate1, candidate2)
+
+    end
+
+
+end