update(security_test.py): moving tests to new framework

We should be moving tests over to pytest

[none]
This commit is contained in:
Charlton Austin 2017-01-27 11:22:40 -05:00
parent 0b51ea34b1
commit 85bcb63439
7 changed files with 190 additions and 107 deletions

View file

@ -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)