Move label display logic into the Label macro

This commit is contained in:
leigh-mil
2019-11-19 11:01:07 -05:00
parent 58a0b2dd9d
commit 8aa302357b
7 changed files with 29 additions and 37 deletions

View File

@@ -61,21 +61,18 @@ def test_environment_roles():
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"
expired_invite = ApplicationInvitationFactory.create(expiration_time=yesterday)
assert expired_invite.role.display_status == "invite_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"
assert app_role_pending.display_status == "invite_pending"
app_role_active = ApplicationRoleFactory.create(status=ApplicationRoleStatus.ACTIVE)
assert app_role_active.display_status == "active"
assert app_role_active.display_status == None
env_role_pending = EnvironmentRoleFactory.create(application_role=app_role_active)
assert env_role_pending.application_role.display_status == "env_changes_pending"
assert env_role_pending.application_role.display_status == "changes_pending"