Skip to content

Instantly share code, notes, and snippets.

@andreyvit
Created January 25, 2010 07:09
Show Gist options
  • Save andreyvit/285696 to your computer and use it in GitHub Desktop.
Save andreyvit/285696 to your computer and use it in GitHub Desktop.
An extension to gae-testbed for testing user authentication scenarios
import os
def set_logged_in_user(email, admin=False):
if email is None or email == '':
email, domain = '', 'example.com'
admin = False
else:
domain = email[email.index('@')+1:]
os.environ['USER_EMAIL'] = email
os.environ['AUTH_DOMAIN'] = domain
os.environ['REMOTE_ADDR'] = '127.0.0.1'
os.environ['USER_IS_ADMIN'] = ('1' if admin else '0')
os.environ['SERVER_NAME'] = 'localhost'
os.environ['SERVER_PORT'] = ''
class UserTestCase(object):
def setUp(self):
super(UserTestCase, self).setUp()
if hasattr(self, 'ADMIN'):
set_logged_in_user(self.ADMIN, admin=True)
elif hasattr(self, 'USER'):
set_logged_in_user(self.USER)
else:
set_logged_in_user(None)
def tearDown(self):
set_logged_in_user(None)
super(UserTestCase, self).tearDown()
@adewale
Copy link

adewale commented Dec 12, 2010

Was this ever merged into the main project?
If not, please send me a pull request: https://github.com/adewale/gaetestbed and I'll merge it into my fork

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment