Create query to get all CCPO users
This commit is contained in:
parent
41820813ed
commit
90ae235cd0
@ -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:
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user