update(security_test.py): moving tests to new framework
We should be moving tests over to pytest [none]
This commit is contained in:
parent
0b51ea34b1
commit
85bcb63439
7 changed files with 190 additions and 107 deletions
|
@ -2,6 +2,7 @@ import bcrypt
|
|||
import logging
|
||||
import json
|
||||
import uuid
|
||||
from flask.ext.login import UserMixin
|
||||
|
||||
from peewee import JOIN_LEFT_OUTER, IntegrityError, fn
|
||||
from uuid import uuid4
|
||||
|
@ -841,3 +842,25 @@ def get_federated_logins(user_ids, service_name):
|
|||
.join(LoginService)
|
||||
.where(FederatedLogin.user << user_ids,
|
||||
LoginService.name == service_name))
|
||||
|
||||
|
||||
class LoginWrappedDBUser(UserMixin):
|
||||
def __init__(self, user_uuid, db_user=None):
|
||||
self._uuid = user_uuid
|
||||
self._db_user = db_user
|
||||
|
||||
def db_user(self):
|
||||
if not self._db_user:
|
||||
self._db_user = get_user_by_uuid(self._uuid)
|
||||
return self._db_user
|
||||
|
||||
@property
|
||||
def is_authenticated(self):
|
||||
return self.db_user() is not None
|
||||
|
||||
@property
|
||||
def is_active(self):
|
||||
return self.db_user().verified
|
||||
|
||||
def get_id(self):
|
||||
return unicode(self._uuid)
|
||||
|
|
Reference in a new issue