From 2e67c860fd43734223b3d3328c9477dc05ef48f0 Mon Sep 17 00:00:00 2001
From: Eduardo Martinez Echevarria <eduardomech@gmail.com>
Date: Sun, 11 Jun 2017 14:00:19 +0200
Subject: [PATCH] Setups in investments controller for Polymorphic concern

---
 app/controllers/budgets/investments_controller.rb | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/app/controllers/budgets/investments_controller.rb b/app/controllers/budgets/investments_controller.rb
index 7eb1b1291..49f57d608 100644
--- a/app/controllers/budgets/investments_controller.rb
+++ b/app/controllers/budgets/investments_controller.rb
@@ -24,6 +24,7 @@ module Budgets
 
     invisible_captcha only: [:create, :update], honeypot: :subtitle, scope: :budget_investment
 
+    helper_method :resource_model, :resource_name
     respond_to :html, :js
 
     def index
@@ -72,6 +73,14 @@ module Budgets
 
     private
 
+      def resource_model
+        Budget::Investment
+      end
+
+      def resource_name
+        "budget_investment"
+      end
+
       def load_investment_votes(investments)
         @investment_votes = current_user ? current_user.budget_investment_votes(investments) : {}
       end