883947b75f
Debug mode allows route integration tests to raise explicit exceptions on errors, instead of returning error pages. Some portions of the test suite need to be able to ignore exceptions (the response is not under test) so they use a separate pytest fixture version of the app and client that are configured with debug disabled, as it would be in production.
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from atst.domain.permission_sets import PermissionSets
|
|
from atst.models import Permissions
|
|
from atst.utils.context_processors import (
|
|
get_resources_from_context,
|
|
user_can_view,
|
|
portfolio as portfolio_context,
|
|
)
|
|
|
|
from tests.factories import *
|
|
|
|
|
|
def test_get_resources_from_context():
|
|
portfolio = PortfolioFactory.create()
|
|
task_order = TaskOrderFactory.create(portfolio=portfolio)
|
|
application = ApplicationFactory.create(portfolio=portfolio)
|
|
environment = EnvironmentFactory.create(application=application)
|
|
|
|
assert get_resources_from_context({"portfolio_id": portfolio.id}) == (portfolio,)
|
|
assert get_resources_from_context({"application_id": application.id}) == (
|
|
portfolio,
|
|
application,
|
|
)
|
|
assert get_resources_from_context({"environment_id": environment.id}) == (
|
|
portfolio,
|
|
application,
|
|
)
|
|
assert get_resources_from_context({"task_order_id": task_order.id}) == (
|
|
portfolio,
|
|
task_order,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def set_g(monkeypatch):
|
|
_g = Mock()
|
|
monkeypatch.setattr("atst.utils.context_processors.g", _g)
|
|
|
|
def _set_g(attr, val):
|
|
setattr(_g, attr, val)
|
|
|
|
yield _set_g
|
|
|
|
|
|
def test_user_can_view(set_g):
|
|
owner = UserFactory.create()
|
|
app_user = UserFactory.create()
|
|
rando = UserFactory.create()
|
|
|
|
portfolio = PortfolioFactory.create(owner=owner)
|
|
application = ApplicationFactory.create(portfolio=portfolio)
|
|
ApplicationRoleFactory.create(
|
|
user=app_user,
|
|
application=application,
|
|
permission_sets=PermissionSets.get_many([PermissionSets.VIEW_APPLICATION]),
|
|
)
|
|
|
|
set_g("portfolio", portfolio)
|
|
set_g("application", application)
|
|
set_g("current_user", owner)
|
|
assert user_can_view(Permissions.VIEW_APPLICATION)
|
|
|
|
set_g("current_user", app_user)
|
|
assert user_can_view(Permissions.VIEW_APPLICATION)
|
|
|
|
set_g("current_user", rando)
|
|
assert not user_can_view(Permissions.VIEW_APPLICATION)
|
|
|
|
|
|
def test_portfolio_no_user(set_g):
|
|
set_g("current_user", None)
|
|
assert portfolio_context() == {}
|
|
|
|
|
|
def test_portfolio_with_user(set_g):
|
|
user = UserFactory.create()
|
|
set_g("current_user", user)
|
|
set_g("portfolio", None)
|
|
assert portfolio_context() != {}
|