commit
241ac3611f
@ -96,7 +96,9 @@ def user_can_view(permission):
|
|||||||
|
|
||||||
|
|
||||||
def portfolio():
|
def portfolio():
|
||||||
if g.portfolio is not None:
|
if g.current_user is None:
|
||||||
|
return {}
|
||||||
|
elif g.portfolio is not None:
|
||||||
active_task_orders = [
|
active_task_orders = [
|
||||||
task_order for task_order in g.portfolio.task_orders if task_order.is_active
|
task_order for task_order in g.portfolio.task_orders if task_order.is_active
|
||||||
]
|
]
|
||||||
|
@ -2,7 +2,11 @@ import pytest
|
|||||||
|
|
||||||
from atst.domain.permission_sets import PermissionSets
|
from atst.domain.permission_sets import PermissionSets
|
||||||
from atst.models import Permissions
|
from atst.models import Permissions
|
||||||
from atst.utils.context_processors import get_resources_from_context, user_can_view
|
from atst.utils.context_processors import (
|
||||||
|
get_resources_from_context,
|
||||||
|
user_can_view,
|
||||||
|
portfolio as portfolio_context,
|
||||||
|
)
|
||||||
|
|
||||||
from tests.factories import *
|
from tests.factories import *
|
||||||
|
|
||||||
@ -63,3 +67,14 @@ def test_user_can_view(set_g):
|
|||||||
|
|
||||||
set_g("current_user", rando)
|
set_g("current_user", rando)
|
||||||
assert not user_can_view(Permissions.VIEW_APPLICATION)
|
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)
|
||||||
|
assert portfolio_context() != {}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user