From 5b008e044a2c17364f1eab6cce36ceea1019551e Mon Sep 17 00:00:00 2001
From: leigh-mil <leigh-ctr@friends.dds.mil>
Date: Thu, 8 Aug 2019 10:33:27 -0400
Subject: [PATCH] Add tests

---
 tests/domain/test_users.py | 12 ++++++++++++
 tests/test_access.py       | 31 +++++++++++++++++++++++++++++++
 2 files changed, 43 insertions(+)

diff --git a/tests/domain/test_users.py b/tests/domain/test_users.py
index fda69a12..08f75412 100644
--- a/tests/domain/test_users.py
+++ b/tests/domain/test_users.py
@@ -85,3 +85,15 @@ def test_get_ccpo_users():
     assert ccpo_1 in ccpo_users
     assert ccpo_2 in ccpo_users
     assert rando not in ccpo_users
+
+
+def test_update_ccpo_permissions():
+    rando = UserFactory.create()
+    Users.update_ccpo_permissions(rando, add_perms=True)
+
+    ccpo = UserFactory.create_ccpo()
+    Users.update_ccpo_permissions(ccpo)
+
+    ccpo_users = Users.get_ccpo_users()
+    assert rando in ccpo_users
+    assert ccpo not in ccpo_users
diff --git a/tests/test_access.py b/tests/test_access.py
index 141fed18..29e3a796 100644
--- a/tests/test_access.py
+++ b/tests/test_access.py
@@ -130,6 +130,37 @@ def test_atst_ccpo_users_access(get_url_assert_status):
     get_url_assert_status(rando, url, 404)
 
 
+# atst.add_new_ccpo_user
+def test_atst_add_new_ccpo_user_access(get_url_assert_status):
+    ccpo = user_with(PermissionSets.MANAGE_CCPO_USERS)
+    rando = user_with()
+
+    url = url_for("atst.add_new_ccpo_user")
+    get_url_assert_status(ccpo, url, 200)
+    get_url_assert_status(rando, url, 404)
+
+
+# atst.submit_add_new_ccpo_user
+def test_atst_submit_add_new_ccpo_user_access(post_url_assert_status):
+    ccpo = user_with(PermissionSets.MANAGE_CCPO_USERS)
+    rando = user_with()
+
+    url = url_for("atst.submit_add_new_ccpo_user")
+    post_url_assert_status(ccpo, url, 200, data={"dod_id": "1234567890"})
+    post_url_assert_status(rando, url, 404, data={"dod_id": "1234567890"})
+
+
+# atst.confirm_new_ccpo_user
+def test_atst_confirm_new_ccpo_user_access(post_url_assert_status):
+    ccpo = user_with(PermissionSets.MANAGE_CCPO_USERS)
+    rando = user_with()
+    user = UserFactory.create()
+
+    url = url_for("atst.confirm_new_ccpo_user")
+    post_url_assert_status(ccpo, url, 302, data={"dod_id": user.dod_id})
+    post_url_assert_status(rando, url, 404, data={"dod_id": user.dod_id})
+
+
 # applications.access_environment
 def test_applications_access_environment_access(get_url_assert_status):
     dev = UserFactory.create()