download link for task order summary

This commit is contained in:
dandds
2018-12-20 10:41:54 -05:00
parent 59510819e7
commit 718f88d828
7 changed files with 55 additions and 5 deletions

View File

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