102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
from werkzeug.datastructures import FileStorage
|
|
import pytest, datetime
|
|
|
|
from atst.models import *
|
|
from atst.models.clin import JEDICLINType
|
|
from atst.models.task_order import TaskOrder, Status
|
|
|
|
from tests.factories import (
|
|
CLINFactory,
|
|
random_future_date,
|
|
random_past_date,
|
|
TaskOrderFactory,
|
|
)
|
|
from tests.mocks import PDF_FILENAME
|
|
|
|
|
|
class TestTaskOrderStatus:
|
|
@pytest.mark.skip(reason="Reimplement after adding CLINs")
|
|
def test_started_status(self):
|
|
to = TaskOrder()
|
|
assert to.status == Status.STARTED
|
|
|
|
@pytest.mark.skip(reason="See if still needed after implementing CLINs")
|
|
def test_pending_status(self):
|
|
to = TaskOrder(number="42")
|
|
assert to.status == Status.PENDING
|
|
|
|
@pytest.mark.skip(reason="See if still needed after implementing CLINs")
|
|
def test_active_status(self):
|
|
to = TaskOrder(number="42")
|
|
assert to.status == Status.ACTIVE
|
|
|
|
@pytest.mark.skip(reason="See if still needed after implementing CLINs")
|
|
def test_expired_status(self):
|
|
to = TaskOrder(number="42")
|
|
assert to.status == Status.EXPIRED
|
|
|
|
|
|
class TestBudget:
|
|
def test_total_contract_amount(self):
|
|
to = TaskOrder()
|
|
assert to.total_contract_amount == 0
|
|
|
|
clin1 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_1)
|
|
clin2 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_2)
|
|
clin3 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_3)
|
|
|
|
assert (
|
|
to.total_contract_amount
|
|
== clin1.obligated_amount + clin2.obligated_amount + clin3.obligated_amount
|
|
)
|
|
|
|
def test_total_obligated_funds(self):
|
|
to = TaskOrder()
|
|
clin4 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_4)
|
|
assert to.total_obligated_funds == 0
|
|
|
|
clin1 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_1)
|
|
clin2 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_2)
|
|
clin3 = CLINFactory(task_order=to, jedi_clin_type=JEDICLINType.JEDI_CLIN_3)
|
|
assert (
|
|
to.total_obligated_funds == clin1.obligated_amount + clin3.obligated_amount
|
|
)
|
|
|
|
|
|
class TestPDF:
|
|
def test_setting_pdf_with_attachment(self):
|
|
to = TaskOrder()
|
|
attachment = Attachment(filename="sample.pdf", object_name="sample")
|
|
to.pdf = attachment
|
|
|
|
assert to.pdf_attachment_id == attachment.id
|
|
|
|
def test_setting_pdf_with_file_storage(self):
|
|
to = TaskOrder()
|
|
with open(PDF_FILENAME, "rb") as fp:
|
|
fs = FileStorage(fp, content_type="application/pdf")
|
|
to.pdf = fs
|
|
|
|
assert to.pdf is not None
|
|
assert to.pdf.filename == PDF_FILENAME
|
|
|
|
def test_setting_pdf_with_invalid_object(self):
|
|
to = TaskOrder()
|
|
with pytest.raises(TypeError):
|
|
to.pdf = "invalid"
|
|
|
|
def test_setting_pdf_with_empty_value(self):
|
|
to = TaskOrder()
|
|
assert to.pdf is None
|
|
|
|
to.pdf = ""
|
|
assert to.pdf is None
|
|
|
|
def test_removing_pdf(self):
|
|
attachment = Attachment(filename="sample.pdf", object_name="sample")
|
|
to = TaskOrder(pdf=attachment)
|
|
assert to.pdf is not None
|
|
|
|
to.pdf = ""
|
|
assert to.pdf is None
|