Update test to check that env role was deleted and that the status changed on the app role

This commit is contained in:
leigh-mil 2019-09-26 10:18:42 -04:00
parent bea9439330
commit c9f654664c
2 changed files with 12 additions and 11 deletions

View File

@ -3,7 +3,7 @@
"files": "^.secrets.baseline$", "files": "^.secrets.baseline$",
"lines": null "lines": null
}, },
"generated_at": "2019-09-25T09:43:11Z", "generated_at": "2019-09-26T13:53:31Z",
"plugins_used": [ "plugins_used": [
{ {
"base64_limit": 4.5, "base64_limit": 4.5,
@ -199,5 +199,5 @@
} }
] ]
}, },
"version": "0.12.5" "version": "0.12.6"
} }

View File

@ -1,6 +1,7 @@
import pytest import pytest
from atst.domain.application_roles import ApplicationRoles from atst.domain.application_roles import ApplicationRoles
from atst.domain.environment_roles import EnvironmentRoles
from atst.domain.exceptions import NotFoundError from atst.domain.exceptions import NotFoundError
from atst.domain.permission_sets import PermissionSets from atst.domain.permission_sets import PermissionSets
from atst.models import ApplicationRoleStatus from atst.models import ApplicationRoleStatus
@ -68,19 +69,19 @@ def test_get_by_id():
ApplicationRoles.get_by_id(app_role.id) ApplicationRoles.get_by_id(app_role.id)
def test_disable(): def test_disable(session):
application = ApplicationFactory.create() application = ApplicationFactory.create()
user = UserFactory.create() user = UserFactory.create()
member_role = ApplicationRoleFactory.create(application=application, user=user) member_role = ApplicationRoleFactory.create(
application=application, user=user, status=ApplicationRoleStatus.ACTIVE
)
environment = EnvironmentFactory.create(application=application) environment = EnvironmentFactory.create(application=application)
environment_role = EnvironmentRoleFactory.create( environment_role = EnvironmentRoleFactory.create(
application_role=member_role, environment=environment application_role=member_role, environment=environment
) )
assert member_role == ApplicationRoles.get( assert member_role.status == ApplicationRoleStatus.ACTIVE
user_id=user.id, application_id=application.id
)
ApplicationRoles.disable(member_role) ApplicationRoles.disable(member_role)
assert ( session.refresh(member_role)
ApplicationRoles.get(user_id=user.id, application_id=application.id).status assert member_role.status == ApplicationRoleStatus.DISABLED
== ApplicationRoleStatus.DISABLED assert not EnvironmentRoles.get_by_user_and_environment(user.id, environment.id)
)