91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
import os
|
|
from flask import url_for
|
|
|
|
from atst.models.attachment import Attachment
|
|
from atst.models.request_status_event import RequestStatus
|
|
from atst.domain.roles import Roles
|
|
|
|
from tests.factories import (
|
|
RequestFactory, TaskOrderFactory, UserFactory, RequestReviewFactory
|
|
)
|
|
|
|
|
|
def test_ccpo_can_view_approval(user_session, client):
|
|
ccpo = Roles.get("ccpo")
|
|
user = UserFactory.create(atat_role=ccpo)
|
|
user_session(user)
|
|
|
|
request = RequestFactory.create()
|
|
response = client.get(url_for("requests.approval", request_id=request.id))
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_non_ccpo_cannot_view_approval(user_session, client):
|
|
user = UserFactory.create()
|
|
user_session(user)
|
|
|
|
request = RequestFactory.create(creator=user)
|
|
response = client.get(url_for("requests.approval", request_id=request.id))
|
|
assert response.status_code == 404
|
|
|
|
|
|
def test_task_order_download(app, client, user_session, pdf_upload):
|
|
user = UserFactory.create()
|
|
user_session(user)
|
|
|
|
attachment = Attachment.attach(pdf_upload)
|
|
task_order = TaskOrderFactory.create(number="abc123", pdf=attachment)
|
|
request = RequestFactory.create(task_order=task_order, creator=user)
|
|
|
|
# ensure that real data for pdf upload has been flushed to disk
|
|
pdf_upload.seek(0)
|
|
pdf_content = pdf_upload.read()
|
|
pdf_upload.close()
|
|
full_path = os.path.join(
|
|
app.config.get("STORAGE_CONTAINER"), attachment.object_name
|
|
)
|
|
with open(full_path, "wb") as output_file:
|
|
output_file.write(pdf_content)
|
|
output_file.flush()
|
|
|
|
response = client.get(
|
|
url_for("requests.task_order_pdf_download", request_id=request.id)
|
|
)
|
|
assert response.data == pdf_content
|
|
|
|
|
|
def test_task_order_download_does_not_exist(client, user_session):
|
|
user = UserFactory.create()
|
|
user_session(user)
|
|
request = RequestFactory.create(creator=user)
|
|
response = client.get(
|
|
url_for("requests.task_order_pdf_download", request_id=request.id)
|
|
)
|
|
assert response.status_code == 404
|
|
|
|
|
|
def test_can_submit_request_approval(client, user_session):
|
|
user = UserFactory.from_atat_role("ccpo")
|
|
user_session(user)
|
|
request = RequestFactory.create()
|
|
review_data = RequestReviewFactory.dictionary()
|
|
review_data["approved"] = True
|
|
response = client.post(
|
|
url_for("requests.submit_approval", request_id=request.id), data=review_data
|
|
)
|
|
assert response.status_code == 302
|
|
assert request.status == RequestStatus.PENDING_FINANCIAL_VERIFICATION
|
|
|
|
|
|
def test_can_submit_request_denial(client, user_session):
|
|
user = UserFactory.from_atat_role("ccpo")
|
|
user_session(user)
|
|
request = RequestFactory.create()
|
|
review_data = RequestReviewFactory.dictionary()
|
|
review_data["denied"] = True
|
|
response = client.post(
|
|
url_for("requests.submit_approval", request_id=request.id), data=review_data
|
|
)
|
|
assert response.status_code == 302
|
|
assert request.status == RequestStatus.CHANGES_REQUESTED
|