Stub environment in tests

test/controllers/users_controller_test.rb

@@ -17,15 +17,17 @@ class UsersControllerTest < ActionDispatch::IntegrationTest
   end
 
   test "should create user" do
-    assert_difference("User.count") do
+    stub_environment(REGISTRATION_ALLOWED: "true") do
-      params = {
+      assert_difference("User.count") do
-        user: {
+        params = {
-          email: "userthree@example.local",
+          user: {
-          password: "secret",
+            email: "userthree@example.local",
-          password_confirmation: "secret"
+            password: "secret",
+            password_confirmation: "secret"
+          }
         }
-      }
+        post users_url, params: params
-      post users_url, params: params
+      end
     end
 
     assert_redirected_to user_url(User.last)

test/system/users_test.rb

@@ -15,12 +15,14 @@ class UsersTest < ApplicationSystemTestCase
     visit users_url
     click_on "Sign up"
 
-    fill_in "Email", with: "userthree@example.local"
+    stub_environment(REGISTRATION_ALLOWED: "true") do
-    fill_in "Password", with: "secret"
+      fill_in "Email", with: "userthree@example.local"
-    fill_in "Password confirmation", with: "secret"
+      fill_in "Password", with: "secret"
-    click_on "Create User"
+      fill_in "Password confirmation", with: "secret"
+      click_on "Create User"
 
-    assert_text "User was successfully created"
+      assert_text "User was successfully created"
+    end
     click_on "Back"
   end
 

test/test_helper.rb

@@ -14,4 +14,15 @@ class ActiveSupport::TestCase
     post sessions_url, params: { session: { email: email, password: password } }
     assert session[:current_user_id].present?
   end
+
+  def stub_environment(env)
+    old_env = ENV.to_hash
+    ENV.update(env.stringify_keys)
+
+    begin
+      yield
+    ensure
+      ENV.replace(old_env)
+    end
+  end
 end