9f64129be5
These variables can be considered a block, and so removing them doesn't make the test much harder to undestand. Sometimes these variables formed the setup, sometimes they formed an isolated part of the setup, and sometimes they were the part of the test that made the test different from other tests.
128 lines
4.1 KiB
Ruby
128 lines
4.1 KiB
Ruby
require "rails_helper"
|
|
|
|
describe Budget::Ballot do
|
|
|
|
describe "validations" do
|
|
|
|
it "is valid" do
|
|
budget = create(:budget)
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
|
|
expect(ballot).to be_valid
|
|
end
|
|
|
|
it "is not valid with the same investment twice" do
|
|
budget = create(:budget)
|
|
investment = create(:budget_investment, :selected, budget: budget)
|
|
ballot = create(:budget_ballot, budget: budget.reload, investments: [investment])
|
|
|
|
expect { ballot.investments << investment }.to raise_error(ActiveRecord::RecordNotUnique)
|
|
end
|
|
|
|
end
|
|
|
|
describe "#amount_spent" do
|
|
it "returns the total amount spent in investments" do
|
|
budget = create(:budget)
|
|
|
|
group1 = create(:budget_group, budget: budget)
|
|
group2 = create(:budget_group, budget: budget)
|
|
|
|
heading1 = create(:budget_heading, group: group1, price: 100000)
|
|
heading2 = create(:budget_heading, group: group2, price: 200000)
|
|
|
|
inv1 = create(:budget_investment, :selected, price: 10000, heading: heading1)
|
|
inv2 = create(:budget_investment, :selected, price: 20000, heading: heading2)
|
|
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
ballot.investments << inv1
|
|
|
|
expect(ballot.total_amount_spent).to eq 10000
|
|
|
|
ballot.investments << inv2
|
|
|
|
expect(ballot.total_amount_spent).to eq 30000
|
|
end
|
|
|
|
it "returns the amount spent on all investments assigned to a specific heading" do
|
|
budget = create(:budget)
|
|
|
|
group1 = create(:budget_group, budget: budget)
|
|
group2 = create(:budget_group, budget: budget)
|
|
|
|
heading1 = create(:budget_heading, group: group1, price: 100000)
|
|
heading2 = create(:budget_heading, group: group2, price: 200000)
|
|
|
|
inv1 = create(:budget_investment, :selected, price: 10000, heading: heading1)
|
|
inv2 = create(:budget_investment, :selected, price: 20000, heading: heading2)
|
|
inv3 = create(:budget_investment, :selected, price: 40000, heading: heading1)
|
|
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
ballot.investments << inv1 << inv2
|
|
|
|
expect(ballot.amount_spent(heading1)).to eq 10000
|
|
expect(ballot.amount_spent(heading2)).to eq 20000
|
|
|
|
ballot.investments << inv3
|
|
|
|
expect(ballot.amount_spent(heading1)).to eq 50000
|
|
expect(ballot.amount_spent(heading2)).to eq 20000
|
|
end
|
|
end
|
|
|
|
describe "#amount_available" do
|
|
it "returns how much is left after taking some investments" do
|
|
budget = create(:budget)
|
|
|
|
group1 = create(:budget_group, budget: budget)
|
|
group2 = create(:budget_group, budget: budget)
|
|
|
|
heading1 = create(:budget_heading, group: group1, price: 1000)
|
|
heading2 = create(:budget_heading, group: group2, price: 300)
|
|
|
|
inv1 = create(:budget_investment, :selected, price: 100, heading: heading1)
|
|
inv2 = create(:budget_investment, :selected, price: 200, heading: heading2)
|
|
inv3 = create(:budget_investment, :selected, price: 400, heading: heading1)
|
|
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
ballot.investments << inv1 << inv2
|
|
|
|
expect(ballot.amount_available(heading1)).to eq 900
|
|
expect(ballot.amount_available(heading2)).to eq 100
|
|
|
|
ballot.investments << inv3
|
|
|
|
expect(ballot.amount_available(heading1)).to eq 500
|
|
end
|
|
end
|
|
|
|
describe "#heading_for_group" do
|
|
|
|
it "returns the heading with balloted investments for a group" do
|
|
budget = create(:budget)
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
group = create(:budget_group, budget: budget)
|
|
|
|
heading1 = create(:budget_heading, group: group)
|
|
heading2 = create(:budget_heading, group: group)
|
|
|
|
create(:budget_investment, :selected, heading: heading1)
|
|
create(:budget_investment, :selected, heading: heading2, ballots: [ballot])
|
|
|
|
expect(ballot.heading_for_group(group)).to eq heading2
|
|
end
|
|
|
|
it "returns nil if there are no headings with balloted investments in a group" do
|
|
budget = create(:budget)
|
|
group = create(:budget_group, budget: budget)
|
|
ballot = create(:budget_ballot, budget: budget)
|
|
|
|
2.times { create(:budget_heading, group: group) }
|
|
|
|
expect(ballot.heading_for_group(group)).to eq nil
|
|
end
|
|
|
|
end
|
|
|
|
end
|