Merge pull request #833 from dod-ccpo/failing-tests

Failing tests
This commit is contained in:
George Drummond 2019-05-17 14:02:40 -04:00 committed by GitHub
commit 241ac3611f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 2 deletions

View File

@ -96,7 +96,9 @@ def user_can_view(permission):
def portfolio():
if g.portfolio is not None:
if g.current_user is None:
return {}
elif g.portfolio is not None:
active_task_orders = [
task_order for task_order in g.portfolio.task_orders if task_order.is_active
]

View File

@ -2,7 +2,11 @@ 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
from atst.utils.context_processors import (
get_resources_from_context,
user_can_view,
portfolio as portfolio_context,
)
from tests.factories import *
@ -63,3 +67,14 @@ def test_user_can_view(set_g):
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)
assert portfolio_context() != {}