common.py 文件源码

def createTestSession(username=None, password=None, admin=True):
    """Creates a test session and, if provided, sets up authentication"""
    # Create session sid
    sid = 'test_session'

    # Create user object, if it doesn't already exist
    try:
        User.objects.get(uid=username)
    except Exception:
        User(uid=username, admin=admin).save()

    # Create cookie for setting header
    cookie = Cookie.SimpleCookie()
    cookie['sid'] = sid
    cookie['sid']['expires'] = 24 * 60 * 60
    RedisConnection.hset('session_' + sid, 'username', username)
    RedisConnection.hset('session_' + sid, 'password', password)
    return sid, cookie.output()