Preserve cookies, so that we don't lose the session between requests. Make sure you are using the memory session store.

This commit is contained in:
Gwyn Morfey 2008-04-16 17:12:31 +01:00
parent 04261e66d8
commit f74037e4da
1 changed files with 29 additions and 13 deletions

View File

@ -33,29 +33,38 @@ class Merb::Test::RspecStory
#Session defines the following (used by webrat), but RspecStory doesn't. Merb's get/put/delete return a controller,
#which is where we get our status and response from.
#
#We have to preserve cookies like this, or the session is lost.
def request_via_redirect(method,path,parameters={},headers={})
mycookies = @controller.cookies rescue nil #will be nil if no requests yet
@controller=self.send(method, path, parameters, headers) do |new_controller|
new_controller.cookies = mycookies
end
follow_redirect! while redirect?
status
end
def get_via_redirect(path, parameters = {}, headers = {})
@controller=get path, parameters, headers
follow_redirect! while redirect?
status
request_via_redirect(:get,path,parameters,headers)
end
def put_via_redirect(path, parameters = {}, headers = {})
@controller=put path, parameters, headers
follow_redirect! while redirect?
status
request_via_redirect(:put,path,parameters,headers)
end
def post_via_redirect(path, parameters = {}, headers = {})
@controller=post path, parameters, headers
follow_redirect! while redirect?
status
request_via_redirect(:post,path,parameters,headers)
end
def delete_via_redirect(path, parameters = {}, headers = {})
@controller=delete path, parameters, headers
follow_redirect! while redirect?
status
request_via_redirect(:delete,path,parameters,headers)
end
def follow_redirect!
@controller=get @controller.headers["Location"]
mycookies = @controller.cookies rescue nil
@controller=get @controller.headers["Location"] do |new_controller|
new_controller.cookies=mycookies
end
end
def redirect?
@ -79,6 +88,13 @@ class Merb::Test::RspecStory
end
end
class Application < Merb::Controller
def cookies=(newcookies)
@_cookies = newcookies
end
end
#Other utilities used by Webrat that are present in Rails but not Merb. We can require heavy dependencies
#here because we're only loaded in Test mode.
require 'strscan'