From 9a75067d7306530a466daa935cc40db5b2e6eeaf Mon Sep 17 00:00:00 2001 From: rgarcia Date: Sun, 14 May 2017 18:15:04 +0200 Subject: [PATCH] Do not display voting button for unselected investments --- app/models/budget/investment.rb | 2 +- spec/features/budgets/investments_spec.rb | 18 ++++++++++++++++++ 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/app/models/budget/investment.rb b/app/models/budget/investment.rb index b86cc2518..4f7659aa2 100644 --- a/app/models/budget/investment.rb +++ b/app/models/budget/investment.rb @@ -216,7 +216,7 @@ class Budget end def should_show_ballots? - budget.balloting? + budget.balloting? && selected? end def should_show_price? diff --git a/spec/features/budgets/investments_spec.rb b/spec/features/budgets/investments_spec.rb index ca5427820..a6f988e56 100644 --- a/spec/features/budgets/investments_spec.rb +++ b/spec/features/budgets/investments_spec.rb @@ -652,6 +652,24 @@ feature 'Budget Investments' do expect(page).to have_current_path(expected_path) end + scenario "Do not display vote button for unselected investments in index" do + investment = create(:budget_investment, :unselected, heading: heading) + + visit budget_investments_path(budget_id: budget.id, heading_id: heading.id, filter: "unselected") + + expect(page).to have_content investment.title + expect(page).to_not have_link("Vote") + end + + scenario "Do not display vote button for unselected investments in show" do + investment = create(:budget_investment, :unselected, heading: heading) + + visit budget_investment_path(budget, investment) + + expect(page).to have_content investment.title + expect(page).to_not have_link("Vote") + end + scenario "Reclassification" do user = create(:user, :level_two) investment = create(:budget_investment, :selected, heading: heading)