redirect back to original path to clear openid query string

lib/rack/openid/simple_auth.rb

@@ -1,4 +1,5 @@
 require 'rack/openid'
+require 'rack/request'
 
 module Rack #:nodoc:
   class OpenID
@@ -26,7 +27,7 @@ def call(env)
           app.call(env)
         elsif successful_response?(env)
           authenticate_session(env)
-          app.call(env)
+          redirect_to requested_url(env)
         else
           authentication_request
         end
@@ -55,6 +56,15 @@ def successful_response?(env)
           end
         end
 
+        def requested_url(env)
+          req = Rack::Request.new(env)
+          req.url
+        end
+
+        def redirect_to(url)
+          [303, {'Content-Type' => 'text/html', 'Location' => url}, []]
+        end
+
         def authentication_request
           [401, { OpenID::AUTHENTICATE_HEADER => www_authenticate_header }, []]
         end

test/test_openid.rb

@@ -270,12 +270,13 @@ def test_successful_login
     process '/dashboard'
     follow_redirect!
 
-    assert_equal 200, @response.status
-    assert_equal 'Hello', @response.body
+    assert_equal 303, @response.status
+    assert_equal 'http://example.org/dashboard', @response.headers['Location']
 
     cookie = @response.headers['Set-Cookie'].split(';').first
     process '/dashboard', 'HTTP_COOKIE' => cookie
     assert_equal 200, @response.status
+    assert_equal 'Hello', @response.body
   end
 
   def test_failed_login