1 files changed, 26 insertions, 0 deletions
diff --git a/railties/test/env_helpers.rb b/railties/test/env_helpers.rb
new file mode 100644
index 0000000000..6223c85bbf
--- /dev/null
+++ b/railties/test/env_helpers.rb
@@ -0,0 +1,26 @@
+module EnvHelpers
+  private
+
+  def with_rails_env(env)
+    switch_env 'RAILS_ENV', env do
+      switch_env 'RACK_ENV', nil do
+        yield
+      end
+    end
+  end
+
+  def with_rack_env(env)
+    switch_env 'RACK_ENV', env do
+      switch_env 'RAILS_ENV', nil do
+        yield
+      end
+    end
+  end
+
+  def switch_env(key, value)
+    old, ENV[key] = ENV[key], value
+    yield
+  ensure
+    ENV[key] = old
+  end
+end