55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from datetime import date
|
|
from flask import url_for
|
|
import pytest
|
|
from datetime import timedelta, date
|
|
|
|
from atst.domain.permission_sets import PermissionSets
|
|
from atst.domain.task_orders import TaskOrders
|
|
from atst.models import *
|
|
from atst.models.portfolio_role import Status as PortfolioStatus
|
|
from atst.models.task_order import Status as TaskOrderStatus
|
|
from atst.utils.localization import translate
|
|
|
|
from tests.factories import *
|
|
from tests.utils import captured_templates
|
|
|
|
|
|
@pytest.fixture
|
|
def portfolio():
|
|
return PortfolioFactory.create()
|
|
|
|
|
|
@pytest.fixture
|
|
def user():
|
|
return UserFactory.create()
|
|
|
|
|
|
@pytest.fixture
|
|
def task_order():
|
|
user = UserFactory.create()
|
|
portfolio = PortfolioFactory.create(owner=user)
|
|
attachment = Attachment(filename="sample_attachment", object_name="sample")
|
|
|
|
return TaskOrderFactory.create(creator=user, portfolio=portfolio)
|
|
|
|
|
|
def test_review_task_order_not_draft(client, user_session, task_order):
|
|
TaskOrders.sign(task_order=task_order, signer_dod_id=random_dod_id())
|
|
user_session(task_order.portfolio.owner)
|
|
response = client.get(
|
|
url_for("task_orders.review_task_order", task_order_id=task_order.id)
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_review_task_order_draft(client, user_session, task_order):
|
|
TaskOrders.update(
|
|
task_order_id=task_order.id, number="1234567890", clins=[], pdf=None
|
|
)
|
|
user_session(task_order.portfolio.owner)
|
|
response = client.get(
|
|
url_for("task_orders.review_task_order", task_order_id=task_order.id)
|
|
)
|
|
assert response.status_code == 302
|
|
assert url_for("task_orders.edit", task_order_id=task_order.id) in response.location
|