diff --git a/app/controllers/admin/budget_headings_controller.rb b/app/controllers/admin/budget_headings_controller.rb index 3c8ccafa0..56903b744 100644 --- a/app/controllers/admin/budget_headings_controller.rb +++ b/app/controllers/admin/budget_headings_controller.rb @@ -1,4 +1,6 @@ class Admin::BudgetHeadingsController < Admin::BaseController + include FeatureFlags + feature_flag :budgets def create @budget = Budget.find params[:budget_id] diff --git a/app/controllers/admin/budget_investments_controller.rb b/app/controllers/admin/budget_investments_controller.rb index 17cc76923..399ec0ec8 100644 --- a/app/controllers/admin/budget_investments_controller.rb +++ b/app/controllers/admin/budget_investments_controller.rb @@ -1,4 +1,6 @@ class Admin::BudgetInvestmentsController < Admin::BaseController + include FeatureFlags + feature_flag :budgets before_action :load_budget before_action :load_investment, only: [:show, :edit, :update] diff --git a/app/controllers/admin/budgets_controller.rb b/app/controllers/admin/budgets_controller.rb index 144b43a7f..2dd210947 100644 --- a/app/controllers/admin/budgets_controller.rb +++ b/app/controllers/admin/budgets_controller.rb @@ -1,4 +1,6 @@ class Admin::BudgetsController < Admin::BaseController + include FeatureFlags + feature_flag :budgets has_filters %w{open finished}, only: :index diff --git a/app/controllers/budgets_controller.rb b/app/controllers/budgets_controller.rb index db05aafea..9c8a2036b 100644 --- a/app/controllers/budgets_controller.rb +++ b/app/controllers/budgets_controller.rb @@ -1,4 +1,7 @@ class BudgetsController < ApplicationController + include FeatureFlags + feature_flag :budgets + load_and_authorize_resource respond_to :html, :js diff --git a/spec/features/admin/budget_investments_spec.rb b/spec/features/admin/budget_investments_spec.rb index c34354ede..a9ec45dba 100644 --- a/spec/features/admin/budget_investments_spec.rb +++ b/spec/features/admin/budget_investments_spec.rb @@ -9,6 +9,15 @@ feature 'Admin budget investments' do @budget = create(:budget) end + context "Feature flag" do + + scenario 'Disabled with a feature flag' do + Setting['feature.budgets'] = nil + expect{ visit admin_budgets_path }.to raise_exception(FeatureFlags::FeatureDisabled) + end + + end + context "Index" do scenario 'Displaying investmentss' do