Move display status logic to be a property of an ApplicationRole

This commit is contained in:
leigh-mil
2019-11-19 10:43:11 -05:00
parent 9f90f5abbd
commit 58a0b2dd9d
4 changed files with 48 additions and 15 deletions

View File

@@ -58,3 +58,24 @@ def test_environment_roles():
)
assert not EnvironmentRoles.get_by_user_and_environment(user.id, environment2.id)
def test_display_status():
app_role_expired_invite = ApplicationRoleFactory.create()
yesterday = datetime.date.today() - datetime.timedelta(days=1)
ApplicationInvitationFactory.create(
role=app_role_expired_invite, expiration_time=yesterday
)
assert app_role_expired_invite.display_status == "expired"
app_role_pending = ApplicationRoleFactory.create()
invite = ApplicationInvitationFactory.create(
role=app_role_pending, user=app_role_pending.user
)
assert app_role_pending.display_status == "pending"
app_role_active = ApplicationRoleFactory.create(status=ApplicationRoleStatus.ACTIVE)
assert app_role_active.display_status == "active"
env_role_pending = EnvironmentRoleFactory.create(application_role=app_role_active)
assert env_role_pending.application_role.display_status == "env_changes_pending"

View File

@@ -23,8 +23,11 @@ from atst.routes.applications.settings import (
filter_env_roles_form_data,
filter_env_roles_data,
get_environments_obj_for_app,
<<<<<<< HEAD
handle_create_member,
handle_update_member,
=======
>>>>>>> Move display status logic to be a property of an ApplicationRole
)
from tests.utils import captured_templates