download link for task order summary

tests/factories.py

@@ -377,8 +377,8 @@ class TaskOrderFactory(Base):
     defense_component = factory.LazyFunction(random_service_branch)
     app_migration = random_choice(data.APP_MIGRATION)
     native_apps = random.choices(["yes", "no", "not_sure"])
-    complexity = random_choice(data.PROJECT_COMPLEXITY)
-    dev_team = random_choice(data.DEV_TEAM)
+    complexity = [random_choice(data.PROJECT_COMPLEXITY)]
+    dev_team = [random_choice(data.DEV_TEAM)]
     team_experience = random_choice(data.TEAM_EXPERIENCE)
 
     scope = factory.Faker("sentence")

tests/routes/task_orders/test_index.py

@@ -0,0 +1,28 @@
+from flask import url_for
+from io import BytesIO
+import re
+from zipfile import ZipFile
+
+from atst.utils.docx import Docx
+
+from tests.factories import TaskOrderFactory
+
+
+def xml_translated(val):
+    return re.sub("'", "'", str(val))
+
+
+def test_download_summary(client, user_session):
+    user_session()
+    task_order = TaskOrderFactory.create()
+    response = client.get(
+        url_for("task_orders.download_summary", task_order_id=task_order.id)
+    )
+    bytes_str = BytesIO(response.data)
+    zip_ = ZipFile(bytes_str, mode="r")
+    doc = zip_.read(Docx.DOCUMENT_FILE).decode()
+    for attr, val in task_order.to_dictionary().items():
+        assert attr in doc
+        if not xml_translated(val) in doc:
+            __import__("ipdb").set_trace()
+        assert xml_translated(val) in doc

tests/utils/test_docx.py

@@ -7,6 +7,6 @@ from atst.utils.docx import Docx
 def test_render_docx():
     data = {"droid_class": "R2"}
     docx_file = Docx.render(data=data)
-    zip_ = ZipFile(BytesIO(docx_file), mode="r")
+    zip_ = ZipFile(docx_file, mode="r")
     document = zip_.read(Docx.DOCUMENT_FILE)
     assert b"droid_class: R2" in document