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 listusers(current=None):
checkprivilege()
if current==None:
current=AuthUser.load_current_user()
q=server.db.query(Company)
q.raw_output()
return json.dumps(q.filter({'users': {'$elemMatch': {'username': current.username}}}).first()['users'])
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 getuser(username=None):
if username==None:
username=AuthUser.load_current_user()
if username==None:
flask.abort(403)
username=username.username
company=server.db.query(Company).filter({'users': {'$elemMatch': {'username': username}}}).first()
if company==None:
return None,None
for user in company.users:
if user.username==username:
return user,company
raise Exception('User found on DB but not on user array')
def test_authenticate(self):
with self.app.test_request_context():
assert self.user.authenticate(self.PASSWORD) is True
assert self.user.is_logged_in() is True
assert AuthUser.load_current_user() == self.user
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 checkprivilege():
if AuthUser.load_current_user().role!='admin':
flask.abort(403)
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_authenticate(self):
with self.app.test_request_context():
assert self.user.authenticate(self.PASSWORD) is True
assert self.user.is_logged_in() is True
assert AuthUser.load_current_user() == self.user
