Merge pull request #4544 from consul/single-budget-views

Simplify investment form in single heading budgets

app/controllers/budgets/investments_controller.rb

@@ -67,6 +67,7 @@ module Budgets
 
     def create
       @investment.author = current_user
+      @investment.heading = @budget.headings.first if @budget.single_heading?
 
       if @investment.save
         Mailer.budget_investment_created(@investment).deliver_later

app/controllers/management/budgets/investments_controller.rb

@@ -25,6 +25,7 @@ class Management::Budgets::InvestmentsController < Management::BaseController
   def create
     @investment.terms_of_service = "1"
     @investment.author = managed_user
+    @investment.heading = @budget.headings.first if @budget.single_heading?
 
     if @investment.save
       notice = t("flash.actions.create.notice", resource_name: Budget::Investment.model_name.human, count: 1)