diff --git a/spec/models/budget/investment_spec.rb b/spec/models/budget/investment_spec.rb index e2db88221..a4e1d246d 100644 --- a/spec/models/budget/investment_spec.rb +++ b/spec/models/budget/investment_spec.rb @@ -340,6 +340,24 @@ describe Budget::Investment do end end + describe "#by_valuator_group" do + + it "returns investments assigned to a valuator's group" do + valuator = create(:valuator) + valuator_group = create(:valuator_group, valuators: [valuator]) + assigned_investment = create(:budget_investment, valuators: [valuator], + valuator_groups: [valuator_group]) + another_assigned_investment = create(:budget_investment, valuator_groups: [valuator_group]) + unassigned_investment = create(:budget_investment, valuators: [valuator], valuator_groups: []) + create(:budget_investment, valuators: [valuator], valuator_groups: [create(:valuator_group)]) + + by_valuator_group = described_class.by_valuator_group(valuator.valuator_group_id) + + expect(by_valuator_group.size).to eq(2) + expect(by_valuator_group).to contain_exactly(assigned_investment, another_assigned_investment) + end + end + describe "scopes" do describe "valuation_open" do it "returns all investments with false valuation_finished" do