Add is_app_member() to user model

2019-04-23 13:58:38 -04:00
parent 30018a061b
commit c9deaeec72
2 changed files with 16 additions and 0 deletions
--- a/tests/models/test_user.py
+++ b/tests/models/test_user.py
@@ -26,6 +26,19 @@ def test_cannot_update_dod_id(session):
        session.commit()


+def test_is_app_member():
+    user = UserFactory.create()
+    app = ApplicationFactory.create()
+    ApplicationRoleFactory.create(user=user, application=app)
+    assert user.is_app_member(app)
+
+
+def test_is_not_app_member():
+    user = UserFactory.create()
+    app = ApplicationFactory.create()
+    assert not user.is_app_member(app)
+
+
 def test_deleted_application_roles_are_ignored(session):
    user = UserFactory.create()
    app = ApplicationFactory.create()