From fa825185f57630ca0dcf1bd420cb306e559c5e90 Mon Sep 17 00:00:00 2001
From: rgarcia <voodoorai2000@gmail.com>
Date: Sat, 6 May 2017 03:37:11 +0200
Subject: [PATCH] adds model specs for investment filters

---
 spec/models/budget/investment_spec.rb | 86 +++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/spec/models/budget/investment_spec.rb b/spec/models/budget/investment_spec.rb
index e9d5ccd13..2c5f03738 100644
--- a/spec/models/budget/investment_spec.rb
+++ b/spec/models/budget/investment_spec.rb
@@ -337,6 +337,92 @@ describe Budget::Investment do
     end
   end
 
+  describe "apply_filters_and_search" do
+
+    let(:budget) { create(:budget) }
+
+    it "returns feasible investments" do
+      investment1 = create(:budget_investment, :feasible,   budget: budget)
+      investment2 = create(:budget_investment, :feasible,   budget: budget)
+      investment3 = create(:budget_investment, :unfeasible, budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, {}, :feasible)
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+
+    it "returns unfeasible investments" do
+      investment1 = create(:budget_investment, :unfeasible, budget: budget)
+      investment2 = create(:budget_investment, :unfeasible, budget: budget)
+      investment3 = create(:budget_investment, :feasible,   budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, {}, :unfeasible)
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+
+    it "returns selected investments" do
+      budget.update(phase: "balloting")
+
+      investment1 = create(:budget_investment, :feasible, :selected,   budget: budget)
+      investment2 = create(:budget_investment, :feasible, :selected,   budget: budget)
+      investment3 = create(:budget_investment, :feasible, :unselected, budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, {}, :selected)
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+
+    it "returns unselected investments" do
+      budget.update(phase: "balloting")
+
+      investment1 = create(:budget_investment, :feasible, :unselected, budget: budget)
+      investment2 = create(:budget_investment, :feasible, :unselected, budget: budget)
+      investment3 = create(:budget_investment, :feasible, :selected,   budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, {}, :unselected)
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+
+    it "returns investmens by heading" do
+      group = create(:budget_group, budget: budget)
+
+      heading1 = create(:budget_heading, group: group)
+      heading2 = create(:budget_heading, group: group)
+
+      investment1 = create(:budget_investment, heading: heading1, budget: budget)
+      investment2 = create(:budget_investment, heading: heading1, budget: budget)
+      investment3 = create(:budget_investment, heading: heading2, budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, heading_id: heading1.id)
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+
+    it "returns investments by search string" do
+      investment1 = create(:budget_investment, title: "health for all",  budget: budget)
+      investment2 = create(:budget_investment, title: "improved health", budget: budget)
+      investment3 = create(:budget_investment, title: "finance",         budget: budget)
+
+      results = Budget::Investment::apply_filters_and_search(budget, search: "health")
+
+      expect(results).to     include investment1
+      expect(results).to     include investment2
+      expect(results).to_not include investment3
+    end
+  end
+
   describe "search" do
 
     context "tags" do