From 90ae235cd006e1aed287591704b0fb989588bef9 Mon Sep 17 00:00:00 2001
From: leigh-mil <leigh-ctr@friends.dds.mil>
Date: Mon, 5 Aug 2019 16:04:31 -0400
Subject: [PATCH] Create query to get all CCPO users

---
 atst/domain/users.py       |  4 ++++
 tests/domain/test_users.py | 11 +++++++++++
 2 files changed, 15 insertions(+)

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