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