Create query to get all CCPO users

This commit is contained in:
leigh-mil 2019-08-05 16:04:31 -04:00
parent 41820813ed
commit 90ae235cd0
2 changed files with 15 additions and 0 deletions

View File

@ -28,6 +28,10 @@ class Users(object):
return user
@classmethod
def get_ccpo_users(cls):
return db.session.query(User).filter(User.permission_sets != None).all()
@classmethod
def create(cls, dod_id, permission_sets=None, **kwargs):
if permission_sets:

View File

@ -74,3 +74,14 @@ def test_update_user_with_last_login():
last_login = new_user.last_login
Users.update_last_login(new_user)
assert new_user.last_login > last_login
def test_get_ccpo_users():
ccpo_1 = UserFactory.create_ccpo()
ccpo_2 = UserFactory.create_ccpo()
rando = UserFactory.create()
ccpo_users = Users.get_ccpo_users()
assert ccpo_1 in ccpo_users
assert ccpo_2 in ccpo_users
assert rando not in ccpo_users