df06d1b62f
In the future, an `application_invitation1 will not refer to a `user` until someone accepts the invitation; they'll only reference an `application_role`. When a user is invited to an application, the inviter can specify the environments the invitee should have access to. For this to be possible, an `environment_role` should reference an `application_role`, because no `user` entity will be known at that time. In addition to updating all the models and domain methods necessary for this change, this commit deletes unused code and tests that were dependent on `environment_roles` having a `user_id` foreign key.
91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock
|
|
|
|
from atst.domain.environment_roles import EnvironmentRoles
|
|
|
|
from tests.factories import *
|
|
|
|
|
|
@pytest.fixture
|
|
def application_role():
|
|
user = UserFactory.create()
|
|
application = ApplicationFactory.create()
|
|
return ApplicationRoleFactory.create(application=application, user=user)
|
|
|
|
|
|
@pytest.fixture
|
|
def environment(application_role):
|
|
return EnvironmentFactory.create(application=application_role.application)
|
|
|
|
|
|
def test_create(application_role, environment, monkeypatch):
|
|
mock_create_role = MagicMock()
|
|
monkeypatch.setattr(
|
|
"atst.domain.environment_roles.app.csp.cloud.create_role", mock_create_role
|
|
)
|
|
|
|
environment_role = EnvironmentRoles.create(
|
|
application_role, environment, "network admin"
|
|
)
|
|
assert environment_role.application_role == application_role
|
|
assert environment_role.environment == environment
|
|
assert environment_role.role == "network admin"
|
|
mock_create_role.assert_called_with(environment_role)
|
|
|
|
|
|
def test_get(application_role, environment):
|
|
EnvironmentRoleFactory.create(
|
|
application_role=application_role, environment=environment
|
|
)
|
|
|
|
environment_role = EnvironmentRoles.get(application_role.id, environment.id)
|
|
assert environment_role
|
|
assert environment_role.application_role == application_role
|
|
assert environment_role.environment == environment
|
|
|
|
|
|
def test_get_by_user_and_environment(application_role, environment):
|
|
expected_role = EnvironmentRoleFactory.create(
|
|
application_role=application_role, environment=environment
|
|
)
|
|
actual_role = EnvironmentRoles.get_by_user_and_environment(
|
|
application_role.user.id, environment.id
|
|
)
|
|
assert expected_role == actual_role
|
|
|
|
|
|
def test_delete(application_role, environment, monkeypatch):
|
|
mock_delete_role = MagicMock()
|
|
monkeypatch.setattr(
|
|
"atst.domain.environment_roles.app.csp.cloud.delete_role", mock_delete_role
|
|
)
|
|
|
|
environment_role = EnvironmentRoleFactory.create(
|
|
application_role=application_role, environment=environment
|
|
)
|
|
assert EnvironmentRoles.delete(application_role.id, environment.id)
|
|
mock_delete_role.assert_called_with(environment_role)
|
|
assert not EnvironmentRoles.delete(application_role.id, environment.id)
|
|
|
|
|
|
def test_get_for_application_member(application_role, environment):
|
|
EnvironmentRoleFactory.create(
|
|
application_role=application_role, environment=environment
|
|
)
|
|
|
|
roles = EnvironmentRoles.get_for_application_member(application_role.id)
|
|
assert len(roles) == 1
|
|
assert roles[0].environment == environment
|
|
assert roles[0].application_role == application_role
|
|
|
|
|
|
def test_get_for_application_member_does_not_return_deleted(
|
|
application_role, environment
|
|
):
|
|
EnvironmentRoleFactory.create(
|
|
application_role=application_role, environment=environment, deleted=True
|
|
)
|
|
|
|
roles = EnvironmentRoles.get_for_application_member(application_role.id)
|
|
assert len(roles) == 0
|