Make portfolio_id an init kwarg on ShowTaskOrderWorkflow class

This commit is contained in:
Montana
2019-03-08 09:31:30 -05:00
parent acf70d1144
commit 970eb8aba3
2 changed files with 19 additions and 16 deletions

View File

@@ -44,8 +44,8 @@ def serialize_dates(data):
def test_new_to_can_edit_pf_attributes_screen_1():
portfolio = PortfolioFactory.create()
workflow = ShowTaskOrderWorkflow(user=portfolio.owner)
assert not workflow.pf_attributes_read_only(portfolio.id)
workflow = ShowTaskOrderWorkflow(user=portfolio.owner, portfolio_id=portfolio.id)
assert not workflow.pf_attributes_read_only()
def test_new_pf_can_edit_pf_attributes_on_back_navigation():
@@ -61,9 +61,9 @@ def test_to_on_pf_cannot_edit_pf_attributes():
portfolio = PortfolioFactory.create()
pf_task_order = TaskOrderFactory(portfolio=portfolio)
workflow = ShowTaskOrderWorkflow(user=portfolio.owner)
workflow = ShowTaskOrderWorkflow(user=portfolio.owner, portfolio_id=portfolio.id)
assert portfolio.num_task_orders == 1
assert workflow.pf_attributes_read_only(portfolio.id)
assert workflow.pf_attributes_read_only()
second_task_order = TaskOrderFactory(portfolio=portfolio)
second_workflow = ShowTaskOrderWorkflow(
@@ -87,8 +87,8 @@ def test_get_portfolio_when_task_order_exists():
def test_get_portfolio_with_portfolio_id():
user = UserFactory.create()
portfolio = PortfolioFactory.create(owner=user)
workflow = ShowTaskOrderWorkflow(user=portfolio.owner)
assert portfolio == workflow.get_portfolio(portfolio_id=portfolio.id)
workflow = ShowTaskOrderWorkflow(user=portfolio.owner, portfolio_id=portfolio.id)
assert portfolio == workflow.get_portfolio()
# TODO: this test will need to be more complicated when we add validation to