pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow removing a csp estimate from a task order

Browse Source
This commit is contained in:
Patrick Smith 2019-01-22 13:05:04 -05:00
parent c155de0e84
commit 672c562323
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
atst/models/task_order.py
View File

@ -86,6 +86,8 @@ class TaskOrder(Base, mixins.TimestampsMixin):
self._csp_estimate = Attachment.attach( self._csp_estimate = Attachment.attach(
new_csp_estimate, "task_order", self.id new_csp_estimate, "task_order", self.id
) )
elif not new_csp_estimate and self._csp_estimate:
self._csp_estimate = None
else: else:
raise TypeError("Could not set csp_estimate with invalid type") raise TypeError("Could not set csp_estimate with invalid type")

8
tests/models/test_task_order.py
View File

@ -58,3 +58,11 @@ class TestCSPEstimate:
to = TaskOrder() to = TaskOrder()
with pytest.raises(TypeError): with pytest.raises(TypeError):
to.csp_estimate = "invalid" to.csp_estimate = "invalid"
def test_removing_estimate(self):
attachment = Attachment(filename="sample.pdf", object_name="sample")
to = TaskOrder(csp_estimate=attachment)
assert to.csp_estimate is not None
to.csp_estimate = ""
assert to.csp_estimate is None