Updated tests

test/application_system_test_case.rb

@@ -24,6 +24,12 @@ def sign_in_as(user)
     click_on "Login to your account"
 
     assert_current_path workspaces_path
+    session[:user_id] = user.id # Add this line to ensure session is set
     user
   end
+
+  def teardown
+    super
+    session[:user_id] = nil if respond_to?(:session)
+  end
 end

test/test_helper.rb

@@ -36,10 +36,18 @@ class TestCase
 
     # Add more helper methods to be used by all tests here...
     def sign_in_as(user)
-      post(sign_in_url, params: { email: user.email, password: "Secret1*3*5*" })
+      if respond_to?(:post)
+        post(sign_in_url, params: { email: user.email, password: "Secret1*3*5*" })
+      else
+        session[:user_id] = user.id
+      end
       user
     end
 
+    def teardown
+      session[:user_id] = nil if respond_to?(:session)
+    end
+
     # Pundit helpers
     def assert_permit(user, record, action)
       msg = "User #{user.inspect} should be permitted to #{action} #{record}, but isn't permitted"