reports shows real budget information for total spend

This commit is contained in:
dandds
2018-09-18 09:54:28 -04:00
parent d4e8dcd0a6
commit ca8b9028bc
5 changed files with 56 additions and 13 deletions

View File

@@ -0,0 +1,18 @@
from atst.domain.reports import Reports
from tests.factories import RequestFactory, TaskOrderFactory, WorkspaceFactory
CLIN_NUMS = ["0001", "0003", "1001", "1003", "2001", "2003"]
def test_workspace_totals():
task_order = TaskOrderFactory.create()
for num in CLIN_NUMS:
setattr(task_order, "clin_{}".format(num), 200)
request = RequestFactory.create(task_order=task_order)
workspace = WorkspaceFactory.create(request=request)
report = Reports.workspace_totals(workspace)
total = 200 * len(CLIN_NUMS)
assert report == {"budget": total, "spent": 0}