def test_logout(self):
with self.app.test_request_context():
login(self.user)
user_data = logout()
assert user_data["username"] == self.user.username
assert session.get(SESSION_USER_KEY) is None
assert session.get(SESSION_LOGIN_KEY) is None
def test_user_expiration_override(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert AuthUser.load_current_user(apply_timeout=False) == self.user
def test_logout(self):
with self.app.test_request_context():
login(self.user)
user_data = logout()
assert user_data['username'] == self.user.username
assert session.get(SESSION_USER_KEY) is None
assert session.get(SESSION_LOGIN_KEY) is None
def test_user_expiration(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert get_current_user_data() is None
assert AuthUser.load_current_user() is None
def test_user_expiration_override(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert AuthUser.load_current_user(apply_timeout=False) == self.user
def test_user_expiration(self):
import time
with self.app.test_request_context():
self.app.auth.user_timeout = 0.01
login(self.user)
time.sleep(0.02)
assert get_current_user_data() is None
assert AuthUser.load_current_user() is None
def create_user():
form = RegistrationForm(request.form)
if form.validate():
user = User(username = form.email.data)
user.set_and_encrypt_password(form.password.data)
user.put()
login(user)
flash(u'Thanks for registering')
return redirect(url_for('subscribe'))
return redirect(url_for('new_user'))
def logon():
username = request.form['username']
password = request.form['password']
if musik.auth.check_auth(username, password):
user = AuthUser(username)
login(user)
return '', 200
pass
else:
return 'Not authorized', 401
pass
def facebook_authorized(resp):
if resp is None:
flash(u'Access denied: reason=%s error=%s' % (
request.args['error_reason'],
request.args['error_description']
))
return redirect(url_for('home'))
session['oauth_token'] = (resp['access_token'], '')
me = facebook.get('/me')
user = User.get_by_username(me.data['email'])
if user is not None:
login(user)
else:
user = User(username=me.data['email'], name=me.data['name'])
user.put()
login(user)
flash(u"You've logged in with Facebook")
return redirect(url_for('subscribe'))
def test_current_user(self):
with self.app.test_request_context():
login(self.user)
assert get_current_user_data() == self.user.__dict__
assert AuthUser.load_current_user() == self.user
def test_login(self):
with self.app.test_request_context():
login(self.user)
assert session[SESSION_USER_KEY] is not None
assert session.get(SESSION_LOGIN_KEY)
def test_current_user(self):
with self.app.test_request_context():
login(self.user)
assert get_current_user_data() == self.user.__dict__
assert AuthUser.load_current_user() == self.user
def test_login(self):
with self.app.test_request_context():
login(self.user)
assert session[SESSION_USER_KEY] is not None
assert session.get(SESSION_LOGIN_KEY)
