X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/8a774e75193fcb66eea9e76b60f10623e1e0e682..35a98e045e85455fa351cf98cc185922504b52d4:/test/system/user_logout_test.rb

diff --git a/test/system/user_logout_test.rb b/test/system/user_logout_test.rb
index a2e145fcc..2f5331711 100644
--- a/test/system/user_logout_test.rb
+++ b/test/system/user_logout_test.rb
@@ -4,19 +4,45 @@ class UserLogoutTest < ApplicationSystemTestCase
   test "Sign out via link" do
     user = create(:user)
     sign_in_as(user)
+    assert_no_content "Log In"
 
     click_on user.display_name
     click_on "Log Out"
-    assert page.has_content? "Log In"
+    assert_content "Log In"
+  end
+
+  test "Sign out via link with referer" do
+    user = create(:user)
+    sign_in_as(user)
+    visit traces_path
+    assert_no_content "Log In"
+
+    click_on user.display_name
+    click_on "Log Out"
+    assert_content "Log In"
+    assert_content "Public GPS Traces"
   end
 
   test "Sign out via fallback page" do
     sign_in_as(create(:user))
+    assert_no_content "Log In"
 
     visit logout_path
-    assert page.has_content? "Logout from OpenStreetMap"
+    assert_content "Logout from OpenStreetMap"
+
+    click_on "Logout"
+    assert_content "Log In"
+  end
+
+  test "Sign out via fallback page with referer" do
+    sign_in_as(create(:user))
+    assert_no_content "Log In"
+
+    visit logout_path(:referer => "/traces")
+    assert_content "Logout from OpenStreetMap"
 
-    click_button "Logout"
-    assert page.has_content? "Log In"
+    click_on "Logout"
+    assert_content "Log In"
+    assert_content "Public GPS Traces"
   end
 end