Added physical votes to stats

app/models/budget/stats.rb

@@ -95,7 +95,9 @@ class Budget
       end
 
       def participants
-        stats_cache("participants") { User.where(id: (authors + voters + balloters).uniq) }
+        stats_cache("participants") do
+          User.where(id: (authors + voters + balloters + poll_ballot_voters).uniq)
+        end
       end
 
       def authors
@@ -110,6 +112,12 @@ class Budget
         stats_cache("balloters") { @budget.ballots.where("ballot_lines_count > ?", 0).pluck(:user_id) }
       end
 
+      def poll_ballot_voters
+        stats_cache("poll_ballot_voters") do
+          @budget&.poll ? @budget.poll.voters.pluck(:user_id) : []
+        end
+      end
+
       def total_participants_with_gender
         stats_cache("total_participants_with_gender") { participants.where.not(gender: nil).distinct.count }
       end

spec/models/budget/stats_spec.rb

@@ -0,0 +1,195 @@
+require "rails_helper"
+
+describe Budget::Stats do
+
+  before(:all) do
+    @budget = create(:budget)
+    @group = create(:budget_group, budget: @budget)
+    @heading = create(:budget_heading, group: @group, price: 1000)
+
+    @investment1 = create(:budget_investment, :selected, author: create(:user, gender: "female"),
+                          heading: @heading, price: 200, ballot_lines_count: 900, winner: true)
+    @investment2 = create(:budget_investment, :selected, author: create(:user, gender: "female"),
+                          heading: @heading, price: 300, ballot_lines_count: 800, winner: true)
+    @investment3 = create(:budget_investment, :selected, author: create(:user, gender: "female",
+                          date_of_birth: 40.years.ago), heading: @heading, price: 400,
+                          ballot_lines_count: 880, winner: true)
+    @investment4 = create(:budget_investment, :selected, author: create(:user, gender: "male"),
+                          heading: @heading, price: 100, ballot_lines_count: 915, winner: true)
+    @investment5 = create(:budget_investment, :unfeasible, author: create(:user, gender: "male",
+                          date_of_birth: 25.years.ago), heading: @heading)
+
+    @support1 = create(:vote, votable: @investment1, voter: create(:user, gender: "male"))
+    @support2 = create(:vote, votable: @investment2, voter: create(:user))
+
+    @budget_ballot1 = create(:budget_ballot, budget: @budget, user: create(:user, gender: "female",
+                             date_of_birth: 54.years.ago))
+    @budget_ballot2 = create(:budget_ballot, budget: @budget, user: create(:user, gender: "female"))
+    @budget_ballot3 = create(:budget_ballot, budget: @budget, user: create(:user, gender: "male"))
+
+    @budget_ballot_line1 = create(:budget_ballot_line, ballot: @budget_ballot1,
+                                  investment: @investment1)
+    @budget_ballot_line2 = create(:budget_ballot_line, ballot: @budget_ballot2,
+                                  investment: @investment2)
+    @budget_ballot_line3 = create(:budget_ballot_line, ballot: @budget_ballot3,
+                                  investment: @investment3)
+
+    @poll = create(:poll, budget: @budget)
+    @poll_voter = create(:poll_voter, :from_booth, poll: @poll)
+
+    @budget_ballot4 = create(:budget_ballot, budget: @budget, physical: true, user: nil)
+    @budget_ballot_line4 = create(:budget_ballot_line, ballot: @budget_ballot4,
+                                  investment: @investment4)
+
+    @stats = Budget::Stats.new(@budget).generate
+  end
+
+  context "#total_participants" do
+
+    it "returns the number of total participants" do
+      expect(@stats[:total_participants]).to be 11
+    end
+
+  end
+
+  context "#total_participants_support_phase" do
+
+    it "returns the number of total participants in the support phase" do
+      expect(@stats[:total_participants_support_phase]).to be 2
+    end
+
+  end
+
+  context "#total_participants_vote_phase" do
+
+    it "returns the number of total participants in the votes phase" do
+      expect(@stats[:total_participants_vote_phase]).to be 4
+    end
+
+  end
+
+  context "#total_budget_investments" do
+
+    it "returns the number of total budget investments" do
+      expect(@stats[:total_budget_investments]).to be 5
+    end
+
+  end
+
+  context "#total_votes" do
+
+    it "returns the number of total votes" do
+      expect(@stats[:total_votes]).to be 4
+    end
+
+  end
+
+  context "#total_selected_investments" do
+
+    it "returns the number of total selected investments" do
+      expect(@stats[:total_selected_investments]).to be 4
+    end
+
+  end
+
+  context "#total_unfeasible_investments" do
+
+    it "returns the number of total unfeasible investments" do
+      expect(@stats[:total_unfeasible_investments]).to be 1
+    end
+
+  end
+
+  context "#total_male_participants" do
+
+    it "returns the number of total male participants" do
+      expect(@stats[:total_male_participants]).to be 4
+    end
+
+  end
+
+  context "#total_female_participants" do
+
+    it "returns the number of total female participants" do
+      expect(@stats[:total_female_participants]).to be 6
+    end
+
+  end
+
+  context "#total_supports" do
+
+    it "returns the number of total supports" do
+      expect(@stats[:total_supports]).to be 2
+    end
+
+  end
+
+  context "#total_unknown_gender_or_age" do
+
+    it "returns the number of total unknown participants' gender or age" do
+      expect(@stats[:total_unknown_gender_or_age]).to be 1
+    end
+
+  end
+
+  context "#age_groups" do
+
+    it "returns the age groups hash" do
+      expect(@stats[:age_groups]["16 - 19"]).to be 0
+      expect(@stats[:age_groups]["20 - 24"]).to be 7
+      expect(@stats[:age_groups]["25 - 29"]).to be 1
+      expect(@stats[:age_groups]["30 - 34"]).to be 0
+      expect(@stats[:age_groups]["35 - 39"]).to be 1
+      expect(@stats[:age_groups]["40 - 44"]).to be 1
+      expect(@stats[:age_groups]["45 - 49"]).to be 0
+      expect(@stats[:age_groups]["50 - 54"]).to be 1
+      expect(@stats[:age_groups]["55 - 59"]).to be 0
+      expect(@stats[:age_groups]["60 - 64"]).to be 0
+      expect(@stats[:age_groups]["65 - 69"]).to be 0
+      expect(@stats[:age_groups]["70 - 140"]).to be 0
+    end
+
+  end
+
+  context "#male_percentage" do
+
+    it "returns the percentage of male participants" do
+      expect(@stats[:male_percentage]).to be 40.0
+    end
+
+  end
+
+  context "#female_percentage" do
+
+    it "returns the percentage of female participants" do
+      expect(@stats[:female_percentage]).to be 60.0
+    end
+
+  end
+
+  context "#headings" do
+
+    it "returns headings data" do
+      expect(@stats[:headings][1][:total_investments_count]).to be 5
+      expect(@stats[:headings][1][:total_participants_support_phase]).to be 2
+      expect(@stats[:headings][1][:total_participants_vote_phase]).to be 4
+      expect(@stats[:headings][1][:total_participants_all_phase]).to be 6
+      expect(@stats[:headings][1][:percentage_participants_support_phase]).to be 100.0
+      expect(@stats[:headings][1][:percentage_district_population_support_phase]).to be 0.162
+      expect(@stats[:headings][1][:percentage_participants_vote_phase]).to be 100.0
+      expect(@stats[:headings][1][:percentage_district_population_vote_phase]).to be 0.324
+      expect(@stats[:headings][1][:percentage_participants_all_phase]).to be 100.0
+      expect(@stats[:headings][1][:percentage_district_population_all_phase]).to be 0.486
+
+      expect(@stats[:headings][:total][:total_investments_count]).to be 5
+      expect(@stats[:headings][:total][:total_participants_support_phase]).to be 2
+      expect(@stats[:headings][:total][:total_participants_vote_phase]).to be 4
+      expect(@stats[:headings][:total][:total_participants_all_phase]).to be 6
+      expect(@stats[:headings][:total][:percentage_participants_support_phase]).to be 100.0
+      expect(@stats[:headings][:total][:percentage_participants_vote_phase]).to be 100.0
+      expect(@stats[:headings][:total][:percentage_participants_all_phase]).to be 100.0
+    end
+
+  end
+
+end