Fix tests

This commit is contained in:
richard-dds
2019-10-08 16:14:16 -04:00
parent ed7bc33e44
commit 0c480ccc41
4 changed files with 19 additions and 22 deletions

View File

@@ -56,7 +56,17 @@ class EnvironmentRoles(object):
@classmethod
def delete(cls, application_role_id, environment_id):
existing_env_role = EnvironmentRoles.get(application_role_id, environment_id)
existing_env_role = (
db.session.query(EnvironmentRole)
.join(ApplicationRole)
.filter(
ApplicationRole.id == application_role_id,
EnvironmentRole.environment_id == environment_id,
EnvironmentRole.status != EnvironmentRole.Status.PENDING_DELETE,
)
.one_or_none()
)
if existing_env_role:
existing_env_role.status = EnvironmentRole.Status.PENDING_DELETE
existing_env_role.role = "deleted"