Test: Extract with_app_env helper method

Author: tangrufus

test/features/app_env_test.rb

@@ -5,30 +5,26 @@ module Rails::AppEnv::FeaturesTest
   class AppEnvTest < ActiveSupport::TestCase
     include EnvHelpers
 
-    def setup
-      Rails.instance_variable_set :@_app_env, nil
-    end
-
     test "Rails.app_env is a kind of ActiveSupport::EnvironmentInquirer when APP_ENV is present" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         assert_kind_of ActiveSupport::EnvironmentInquirer, Rails.app_env
       end
     end
 
     test "Rails.app_env is a kind of ActiveSupport::EnvironmentInquirer when APP_ENV is blank" do
-      switch_env "APP_ENV", nil do
+      with_app_env(nil) do
         assert_kind_of ActiveSupport::EnvironmentInquirer, Rails.app_env
       end
     end
 
     test "Rails.app_env is an instance of Rails::AppEnv::EnvironmentInquirer when APP_ENV is present" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         assert_instance_of Rails::AppEnv::EnvironmentInquirer, Rails.app_env
       end
     end
 
     test "Rails.app_env is set from APP_ENV when both APP_ENV are RAILS_ENV present" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         with_rails_env("bar") do
           assert_equal "foo", Rails.app_env
           assert_equal "bar", Rails.env
@@ -37,7 +33,7 @@ def setup
     end
 
     test "Rails.app_env is set from APP_ENV when APP_ENV is present but RAILS_ENV is blank" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         with_rails_env(nil) do
           assert_equal "foo", Rails.app_env
           assert_equal DEFAULT_RAILS_ENV, Rails.env
@@ -46,7 +42,7 @@ def setup
     end
 
     test "Rails.app_env falls back to Rails.env when APP_ENV is blank but RAILS_ENV is present" do
-      switch_env "APP_ENV", nil do
+      with_app_env(nil) do
         with_rails_env("foo") do
           assert_equal "foo", Rails.app_env
           assert_equal "foo", Rails.env
@@ -55,7 +51,7 @@ def setup
     end
 
     test "Rails.app_env falls back to default Rails.env when both APP_ENV and RAILS_ENV are blank" do
-      switch_env "APP_ENV", nil do
+      with_app_env(nil) do
         with_rails_env(nil) do
           assert_equal DEFAULT_RAILS_ENV, Rails.app_env
           assert_equal DEFAULT_RAILS_ENV, Rails.env
@@ -64,7 +60,7 @@ def setup
     end
 
     test "Rails.app_env does not follow Rails.env changes when both APP_ENV and RAILS_ENV are present" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         with_rails_env("bar") do
           assert_equal "foo", Rails.app_env
           assert_equal "bar", Rails.env
@@ -78,7 +74,7 @@ def setup
     end
 
     test "Rails.app_env does not follow Rails.env changes when APP_ENV is present and RAILS_ENV is blank" do
-      switch_env "APP_ENV", "foo" do
+      with_app_env("foo") do
         with_rails_env(nil) do
           assert_equal "foo", Rails.app_env
           assert_equal DEFAULT_RAILS_ENV, Rails.env
@@ -92,7 +88,7 @@ def setup
     end
 
     test "Rails.app_env does not follow Rails.env changes when APP_ENV is blank but RAILS_ENV is present" do
-      switch_env "APP_ENV", nil do
+      with_app_env(nil) do
         with_rails_env("foo") do
           assert_equal "foo", Rails.app_env
           assert_equal "foo", Rails.env
@@ -106,7 +102,7 @@ def setup
     end
 
     test "Rails.app_env does not follow Rails.env changes when both APP_ENV and RAILS_ENV are blank" do
-      switch_env "APP_ENV", nil do
+      with_app_env(nil) do
         with_rails_env(nil) do
           assert_equal DEFAULT_RAILS_ENV, Rails.app_env
           assert_equal DEFAULT_RAILS_ENV, Rails.env

test/features/env_helpers.rb

@@ -1,6 +1,11 @@
 module EnvHelpers
   private
 
+  def with_app_env(app_env, &block)
+    Rails.instance_variable_set :@_app_env, nil
+    switch_env "APP_ENV", app_env, &block
+  end
+
   def with_rails_env(env, &block)
     Rails.instance_variable_set :@_env, nil
     switch_env "RAILS_ENV", env do