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)