diff --git a/atst/domain/users.py b/atst/domain/users.py index daf69c68..a75a1f28 100644 --- a/atst/domain/users.py +++ b/atst/domain/users.py @@ -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: diff --git a/tests/domain/test_users.py b/tests/domain/test_users.py index 116e5294..fda69a12 100644 --- a/tests/domain/test_users.py +++ b/tests/domain/test_users.py @@ -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