示例#1
0
def user(app, session, request):
    """Add a test user in the test database."""
    user = User(username='******', email='*****@*****.**',
                password='******', active=True)
    session.add(user)
    session.commit()
    with app.app_context():
        user.auth_token = user.generate_auth_token()
    session.add(user)
    session.commit()

    @request.addfinalizer
    def fin():
        """Delete user from database after test finishes."""
        session.delete(user)
        session.commit()

    return user