tests for new task order endpoints, remove new workspace routes

tests/routes/task_orders/test_new_task_order.py

@@ -1,10 +1,9 @@
 import pytest
 from flask import url_for
 
-from atst.database import db
-from atst.models.workspace import Workspace
+from atst.domain.task_orders import TaskOrders
 
-from tests.factories import UserFactory, WorkspaceFactory, TaskOrderFactory
+from tests.factories import UserFactory, TaskOrderFactory
 
 
 def test_new_task_order(client, user_session):
@@ -14,18 +13,43 @@ def test_new_task_order(client, user_session):
     assert response.status_code == 200
 
 
-def test_create_new_task_order(client, user_session):
-    creator = UserFactory.create()
-    task_order = TaskOrderFactory.create(
-        creator=creator, workspace=WorkspaceFactory.create()
-    )
-    user_session()
-
-    response = client.post(
-        url_for("task_orders.update", task_order_id=task_order.id),
-        data={**TaskOrderFactory.dictionary(), "clin_01": 12345, "clin_03": 12345},
+def post_to_task_order_step(client, data, screen, task_order_id=None):
+    return client.post(
+        url_for("task_orders.update", screen=screen, task_order_id=task_order_id),
+        data=data,
         follow_redirects=False,
     )
 
-    assert response.status_code == 200
-    assert task_order.clin_01 == 12345
+
+def slice_data_for_section(task_order_data, section):
+    attrs = TaskOrders.SECTIONS[section]
+    return {k: v for k, v in task_order_data.items() if k in attrs}
+
+
+# TODO: this test will need to be more complicated when we add validation to
+# the forms
+def test_create_new_task_order(client, user_session):
+    creator = UserFactory.create()
+    user_session(creator)
+
+    task_order_data = TaskOrderFactory.dictionary()
+    app_info_data = slice_data_for_section(task_order_data, "app_info")
+
+    response = client.post(
+        url_for("task_orders.update", screen=1),
+        data=app_info_data,
+        follow_redirects=False,
+    )
+    assert url_for("task_orders.new", screen=2) in response.headers["Location"]
+
+    funding_data = slice_data_for_section(task_order_data, "funding")
+    response = client.post(
+        response.headers["Location"], data=funding_data, follow_redirects=False
+    )
+    assert url_for("task_orders.new", screen=3) in response.headers["Location"]
+
+    oversight_data = slice_data_for_section(task_order_data, "oversight")
+    response = client.post(
+        response.headers["Location"], data=oversight_data, follow_redirects=False
+    )
+    assert url_for("task_orders.new", screen=4) in response.headers["Location"]