def build_wsgi_environ(self, *a, **kw):
"""Extend base class to support authenticating as a certain user.
"""
# csrf - for both anon and authenticated
csrf_token = kw.get('csrf_token', 'ThisIsATokenThatIsThirtyTwoBytes')
if csrf_token:
cookies = kw.setdefault('cookies', {})
cookies[CSRF_TOKEN] = csrf_token
kw['HTTP_X-CSRF-TOKEN'] = csrf_token
# user authentication
auth_as = kw.pop('auth_as', None)
if auth_as:
assert auth_as.session_token
cookies = kw.setdefault('cookies', {})
cookies[SESSION] = '%s:%s' % (auth_as.id, auth_as.session_token)
return Client.build_wsgi_environ(self, *a, **kw)
def __init__(self, *a, **kw):
Client.__init__(self, *a, **kw)
Client.website = website
def __init__(self, *a, **kw):
Client.__init__(self, *a, **kw)
Client.website = website
