reports shows real budget information for total spend

tests/domain/test_reports.py

@@ -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}

tests/models/test_task_order.py

@@ -7,3 +7,15 @@ def test_as_dictionary():
     data = TaskOrderFactory.dictionary()
     real_task_order = TaskOrderFactory.create(**data)
     assert real_task_order.to_dictionary() == data
+
+
+def test_budget():
+    task_order = TaskOrderFactory.create(
+        clin_0001=500,
+        clin_0003=200,
+        clin_1001=None,
+        clin_1003=None,
+        clin_2001=None,
+        clin_2003=None,
+    )
+    assert task_order.budget == 700