Create helper method to check ENV variables

Resolves #1422

Author: Sydney Young

app/controllers/application_controller.rb

@@ -321,7 +321,7 @@ def markdown_help
   # modify redirect after signing in
   def after_sign_in_path_for(user)
     # CODE FOR CAS LOGIN --> NEW USER
-    if ENV['CAS_AUTH'] && current_user && current_user.id.nil? &&
+    if env?('CAS_AUTH') && current_user && current_user.id.nil? &&
        current_user.username
       # store username in session since there's a request in between
       session[:new_username] = current_user.username

app/controllers/import_users_controller.rb

@@ -58,7 +58,7 @@ def valid_input_file?(imported_users, file) # rubocop:disable all
 
     # make sure we have username data (otherwise all will always fail)
     unless imported_users.first.keys.include?(:username) ||
-           ENV['CAS_AUTH'].nil?
+           env?('CAS_AUTH')
       flash[:error] = 'Unable to import CSV file. None of the users will be '\
         'able to log in without specifying \'username\' data.'
       redirect_to(:back) && return

app/controllers/users_controller.rb

@@ -24,7 +24,7 @@ def set_user
   end
 
   def check_cas_auth
-    @cas_auth = ENV['CAS_AUTH']
+    @cas_auth = env?('CAS_AUTH')
   end
 
   # ------------ end before filter methods ------------ #

app/models/user.rb

@@ -9,7 +9,7 @@ class User < ActiveRecord::Base
   # :cas_authenticatable module. If not, we implement password authentcation
   # using the :database_authenticatable module and also allow for password
   # resets.
-  if ENV['CAS_AUTH']
+  if env?('CAS_AUTH')
     devise :cas_authenticatable
   else
     devise :database_authenticatable, :recoverable, :rememberable
@@ -37,7 +37,7 @@ class User < ActiveRecord::Base
             presence: true, uniqueness: true,
             format: { with: /\A([\w\.%\+\-]+)@([\w\-]+\.)+([\w]{2,})\z/i }
   # validations for CAS authentication
-  if ENV['CAS_AUTH']
+  if env?('CAS_AUTH')
     validates :cas_login, presence: true, uniqueness: true
   # validations for password authentication
   else
@@ -91,9 +91,9 @@ def equipment_items
   # rubocop:disable CyclomaticComplexity
   def self.search_ldap(login)
     return nil if login.blank?
-    return nil unless ENV['USE_LDAP']
+    return nil unless env?('USE_LDAP')
 
-    filter_param = if ENV['CAS_AUTH']
+    filter_param = if env?('CAS_AUTH')
                      Rails.application.secrets.ldap_login
                    else
                      Rails.application.secrets.ldap_email
@@ -138,7 +138,7 @@ def self.search_ldap(login)
                                     .select { |s| s && !s.empty? }.join(' ')
 
       # define username based on authentication method
-      out[:username] = if ENV['CAS_AUTH']
+      out[:username] = if env?('CAS_AUTH')
                          result[Rails.application.secrets.ldap_login.to_sym][0]
                        else
                          out[:email]
@@ -157,7 +157,7 @@ def self.select_options
   end
 
   def render_name
-    ENV['CAS_AUTH'] ? "#{name} #{username}" : name.to_s
+    env?('CAS_AUTH') ? "#{name} #{username}" : name.to_s
   end
 
   # ---- Reservation methods ---- #

app/views/users/_find_user.html.erb

@@ -3,7 +3,7 @@
     <%= label_tag :searched_id, 'Find User' %>
     <div class="input-group">
       <%= autocomplete_field_tag 'fake_searched_id', '', autocomplete_user_last_name_users_path,
-        size: 30, update_elements: {id: '#searched_id'}, placeholder: "Enter a name or #{ENV['CAS_AUTH'] ? 'username' : 'email'}", class: 'submittable form-control' %>
+        size: 30, update_elements: {id: '#searched_id'}, placeholder: "Enter a name or #{env?('CAS_AUTH') ? 'username' : 'email'}", class: 'submittable form-control' %>
         <%= hidden_field_tag 'searched_id' %>
         <span class="input-group-btn">
           <%= button_tag sanitize("#{content_tag :i, nil, class: "fa fa-search"}"), class: "btn btn-default", rel: 'tooltip', title: 'Search Users' %>

config/environments/production.rb

@@ -23,7 +23,7 @@
 
   # Disable serving static files from the `/public` folder by default since
   # Apache or NGINX already handles this.
-  config.serve_static_files = ENV['RAILS_SERVE_STATIC_FILES'].present?
+  config.serve_static_files = ENV['RAILS_SERVE_STATIC_FILES']
 
   # Compress JavaScripts and CSS.
   config.assets.js_compressor = :uglifier
@@ -76,7 +76,7 @@
   # config.action_controller.asset_host = 'http://assets.example.com'
 
   # Disable e-mails if environment variable is set
-  config.action_mailer.perform_deliveries = ENV['DISABLE_EMAILS'].nil?
+  config.action_mailer.perform_deliveries = ENV['DISABLE_EMAILS']
 
   # Disable delivery errors, bad email addresses will be ignored
   # config.action_mailer.raise_delivery_errors = false
@@ -104,5 +104,5 @@
   config.assets.precompile += %w(print.css)
 
   # set up PartyFoul
-  config.middleware.use('PartyFoul::Middleware') if ENV['PARTY_FOUL_TOKEN']
+  config.middleware.use('PartyFoul::Middleware') if env?('PARTY_FOUL_TOKEN')
 end

config/initializers/000_env_helper.rb

@@ -0,0 +1 @@
+include EnvironmentHandler

config/initializers/00_devise.rb

@@ -30,7 +30,7 @@
   config.skip_session_storage = [:http_auth]
 
   # ==> Configuration for :database_authenticatable
-  unless ENV['CAS_AUTH']
+  unless env?('CAS_AUTH')
     # For bcrypt, this is the cost for hashing the password and defaults to 10.
     # If using other encryptors, it sets how many times you want the password
     # re-encrypted.
@@ -71,7 +71,7 @@
   end
 
   # ==> devise_cas_authenticatable configuration
-  if ENV['CAS_AUTH']
+  if env?('CAS_AUTH')
     # configure the base URL of your CAS server
     config.cas_base_url = Rails.application.secrets.cas_base_url
 

config/initializers/authentication.rb

@@ -1,13 +1,13 @@
 # Check for authentication method and copy data over if necessary (ENV variable
 # to skip if necessary, skip if migrating from a pre-v4.1.0 DB or no table)
-unless ENV['SKIP_AUTH_INIT'] || !User.table_exists? ||
+unless env?('SKIP_AUTH_INIT') || !User.table_exists? ||
        !User.column_names.include?('username')
 
   user = User.first
 
   # if we want to use CAS authentication and the username parameter doesn't
   # match the cas_login parameter, we need to copy that over
-  if ENV['CAS_AUTH'] && user && (user.username != user.cas_login)
+  if env?('CAS_AUTH') && user && (user.username != user.cas_login)
     # if there are any users that don't have cas_logins, we can't use CAS
     if User.where(cas_login: ['', nil]).count > 0
       raise 'There are users missing their CAS logins, you cannot use CAS '\
@@ -17,7 +17,7 @@
     end
   # if we want to use password authentication all users can reset their
   # passwords so it doesn't matter if they already have them or not
-  elsif ENV['CAS_AUTH'].nil? && user && (user.username != user.email)
+  elsif env?('CAS_AUTH')
     User.update_all 'username = email'
   end
 end

config/initializers/setup_mail.rb

@@ -8,7 +8,7 @@
 }
 
 # optional server authentication
-if ENV['RES_SMTP_AUTH']
+if env?('RES_SMTP_AUTH')
   ActionMailer::Base.smtp_settings[:authentication] = :login
   ActionMailer::Base.smtp_settings[:user_name] =
     Rails.application.secrets.smtp_username
@@ -19,7 +19,7 @@
 # logging of automatically sent emails
 class MailObserver
   def self.delivered_email(message)
-    if ENV['LOG_EMAILS']
+    if env?('LOG_EMAILS')
       Rails.logger.info "Sent #{message.subject} to #{message.to}"
     end
   end

db/migrate/20150323013431_add_cas_login_to_users.rb

@@ -3,7 +3,7 @@ def change
     add_column :users, :cas_login, :string
 
     # copy username to cas_login if you use CAS already
-    if ENV['CAS_AUTH']
+    if env?('CAS_AUTH')
       ActiveRecord::Base.connection
         .execute('update users set cas_login=username')
     end

db/seeds.rb

@@ -153,11 +153,11 @@ def generate_user
     u.nickname = FFaker::Name.first_name
     u.phone = FFaker::PhoneNumber.short_phone_number
     u.email = FFaker::Internet.email
-    u.cas_login = FFaker::Internet.user_name if ENV['CAS_AUTH']
+    u.cas_login = FFaker::Internet.user_name if env?('CAS_AUTH')
     u.affiliation = 'YC ' + %w(BK BR CC DC ES JE MC PC SM SY TC TD).sample +
                     ' ' + rand(2012..2015).to_s
     u.role = %w(normal checkout).sample
-    u.username = ENV['CAS_AUTH'] ? u.cas_login : u.email
+    u.username = env?('CAS_AUTH') ? u.cas_login : u.email
   end
 end
 
@@ -355,7 +355,7 @@ def generate_objs(method, obj, n)
   u.save
 
   if MINIMAL || SEMI
-    if ENV['CAS_AUTH']
+    if env?('CAS_AUTH')
       prompt_field(u, :cas_login)
       u.username = u.cas_login
     else
@@ -372,7 +372,7 @@ def generate_objs(method, obj, n)
     prompt_field(u, :phone)
     prompt_field(u, :email)
     prompt_field(u, :affiliation)
-    if ENV['CAS_AUTH']
+    if env?('CAS_AUTH')
       prompt_field(u, :cas_login)
       u.username = u.cas_login
       u.save
@@ -476,6 +476,6 @@ def generate_objs(method, obj, n)
 end
 
 puts "\n***Successfully seeded all records! (#{Time.zone.now - t1}s)***\n\n"
-if !ENV['CAS_AUTH'] && MINIMAL && display_login_msg
+if !env?('CAS_AUTH') && MINIMAL && display_login_msg
   puts "You can log in using e-mail '[email protected]' and password 'passw0rd'\n"
 end

lib/csv_import.rb

@@ -48,7 +48,7 @@ def import_users(array_of_user_data, overwrite = false, user_type = 'normal')
       user_data[:role] = user_type
       user_data[:csv_import] = true
       next if attempt_save_with_csv_data?(user_data)
-      if ENV['USE_LDAP']
+      if env?('USE_LDAP')
         attempt_save_with_ldap(user_data)
       else
         @array_of_fail << [user_data, 'Invalid user parameters.']
@@ -64,7 +64,7 @@ def import_users(array_of_user_data, overwrite = false, user_type = 'normal')
   # otherwise it replaces the keys in the data hash with the ldap data.
   def import_with_ldap(user_data)
     # use username if using cas, email otherwise
-    ldap_param = user_data[ENV['CAS_AUTH'] ? :username : :email]
+    ldap_param = user_data[env?('CAS_AUTH') ? :username : :email]
 
     # check LDAP for missing data
     ldap_user_hash = User.search_ldap(ldap_param)
@@ -84,7 +84,7 @@ def import_with_ldap(user_data)
   # tries to save using only the csv data. This method will return
   # false if the data specified in the csv is invalid on the user model.
   def attempt_save_with_csv_data?(user_data)
-    if ENV['CAS_AUTH']
+    if env?('CAS_AUTH')
       # set the cas login
       user_data[:cas_login] = user_data[:username]
     else

lib/environment_handler.rb

@@ -0,0 +1,7 @@
+module EnvironmentHandler
+  FALSE = [0, '0', false, 'false', nil].freeze
+
+  def env?(var)
+    !FALSE.include? ENV[var]
+  end
+end

lib/tasks/setup_application.rake

@@ -34,7 +34,7 @@ namespace :app do
         phone = STDIN.gets.chomp
         puts 'Email Address:'
         email = STDIN.gets.chomp
-        if ENV['CAS_AUTH']
+        if env?('CAS_AUTH')
           puts 'Username (i.e. NetID, ensure this is correct):'
           cas_login = STDIN.gets.chomp
           username = cas_login
@@ -55,12 +55,12 @@ namespace :app do
               u.last_name = last_name
               u.phone = phone
               u.email = email
-              u.cas_login = cas_login if ENV['CAS_AUTH']
+              u.cas_login = cas_login if env?('CAS_AUTH')
               u.username = username
               u.affiliation = affiliation
               u.role = 'superuser'
               u.view_mode = 'superuser'
-              unless ENV['CAS_AUTH']
+              unless env?('CAS_AUTH')
                 u.password = password
                 u.password_confirmation = password_confirmation
               end

spec/factories/users.rb

@@ -17,7 +17,7 @@
     role 'normal'
     view_mode 'normal'
 
-    if ENV['CAS_AUTH']
+    if env?('CAS_AUTH')
       username { cas_login }
     else
       username { email }

spec/lib/environment_handler_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+include EnvironmentHandler
+
+describe EnvironmentHandler do
+  describe '.env?' do
+    it 'returns false when variable not set' do
+      allow(ENV).to receive(:[]).with('CAS_AUTH').and_return(nil)
+      expect(env?('CAS_AUTH')).to be_falsey
+    end
+    it 'returns false when variable set to 0' do
+      allow(ENV).to receive(:[]).with('CAS_AUTH').and_return('0')
+      expect(env?('CAS_AUTH')).to be_falsey
+    end
+    it 'returns false when variable set to false' do
+      allow(ENV).to receive(:[]).with('CAS_AUTH').and_return('false')
+      expect(env?('CAS_AUTH')).to be_falsey
+    end
+    it 'returns true when variable set to anything except 0 or false' do
+      allow(ENV).to receive(:[]).with('CAS_AUTH').and_return('true')
+      expect(env?('CAS_AUTH')).to be_truthy
+    end
+  end
+end

spec/mailers/user_mailer_spec.rb

@@ -63,7 +63,7 @@
     end
 
     it "doesn't log if the env is not set" do
-      expect(ENV['LOG_EMAILS']).to be_nil
+      expect(env?('LOG_EMAILS')).to be_falsey
       expect(Rails.logger).to receive(:info).with(/Sent/).exactly(0).times
       # force a request email; there is not an email for a basic reservation
       @mail = UserMailer.reservation_status_update(@res,