72 lines
2.7 KiB
Ruby
72 lines
2.7 KiB
Ruby
require 'rails_helper'
|
|
|
|
describe Budget::Ballot do
|
|
|
|
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
|
|
|
|
end
|