Reimplement delete environment tests (puma#2710)

JuanitoFatas · Sep 9, 2022 · 7fb3b42 · 7fb3b42
1 parent 9b94ed0
commit 7fb3b42
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/test/test_cli.rb b/test/test_cli.rb
@@ -439,4 +439,25 @@ def test_environment_app_env
     ENV.delete 'APP_ENV'
     ENV.delete 'RAILS_ENV'
   end
-end
+
+  def test_environment_rack_env
+    ENV['RACK_ENV'] = @environment
+
+    cli = Puma::CLI.new []
+    cli.send(:setup_options)
+
+    assert_equal @environment, cli.instance_variable_get(:@conf).environment.call
+  end
+
+  def test_environment_rails_env
+    ENV.delete 'RACK_ENV'
+    ENV['RAILS_ENV'] = @environment
+
+    cli = Puma::CLI.new []
+    cli.send(:setup_options)
+
+    assert_equal @environment, cli.instance_variable_get(:@conf).environment.call
+  ensure
+    ENV.delete 'RAILS_ENV'
+  end
+end