From f1909b3bf26d15fb7919b5d37a269d7822cbf1fb Mon Sep 17 00:00:00 2001 From: Rob Zolkos Date: Fri, 2 Feb 2024 17:01:47 +0000 Subject: [PATCH] Add initial list of valid account types --- app/controllers/accounts_controller.rb | 4 +--- app/models/account.rb | 2 ++ 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app/controllers/accounts_controller.rb b/app/controllers/accounts_controller.rb index 18d6106e..44b1aff9 100644 --- a/app/controllers/accounts_controller.rb +++ b/app/controllers/accounts_controller.rb @@ -33,9 +33,7 @@ class AccountsController < ApplicationController end def account_type_class - valid_account_types = %w[Checking CreditCard] - - if params[:type].present? && valid_account_types.include?(params[:type]) + if params[:type].present? && Account::VALID_ACCOUNT_TYPES.include?(params[:type]) params[:type].constantizes else Account # Default to Account if type is not provided or invalid diff --git a/app/models/account.rb b/app/models/account.rb index 9f130896..73a8b482 100644 --- a/app/models/account.rb +++ b/app/models/account.rb @@ -1,3 +1,5 @@ class Account < ApplicationRecord belongs_to :family + + VALID_ACCOUNT_TYPES = %w[Investment Depository Credit Loan Property Vehicle OtherAsset OtherLiability].freeze end