mirror of
https://github.com/sartography/cr-connect-workflow.git
synced 2025-02-23 21:28:32 +00:00
8ec58fd6d7
I also have a pull request into the folks that run the python docx library to improve their error messages that we depend on here.
83 lines
3.5 KiB
Python
83 lines
3.5 KiB
Python
import os
|
|
from io import BytesIO
|
|
|
|
from lxml import etree
|
|
|
|
from tests.base_test import BaseTest
|
|
from crc.services.workflow_processor import WorkflowProcessor
|
|
from crc.services.workflow_service import WorkflowService
|
|
from crc.services.jinja_service import JinjaService
|
|
from crc.api.common import ApiError
|
|
|
|
from crc import mail, app
|
|
|
|
import json
|
|
|
|
|
|
class TestJinjaService(BaseTest):
|
|
|
|
def test_jinja_service_element_documentation(self):
|
|
self.load_example_data()
|
|
workflow = self.create_workflow('random_fact')
|
|
processor = WorkflowProcessor(workflow)
|
|
processor.do_engine_steps()
|
|
|
|
task = processor.next_task()
|
|
task.data = {"my_template": "Hi {{name}}, This is a jinja template too!",
|
|
"name": "Dan"}
|
|
task.task_spec.documentation = """{% include 'my_template' %} Cool Right?"""
|
|
docs = WorkflowService._process_documentation(task)
|
|
self.assertEqual("Hi Dan, This is a jinja template too! Cool Right?", docs)
|
|
|
|
def test_jinja_service_email(self):
|
|
workflow = self.create_workflow('jinja_email')
|
|
workflow_api = self.get_workflow_api(workflow)
|
|
task = workflow_api.next_task
|
|
data = {'subject': "My Email Subject",
|
|
'recipients': 'user@example.com',
|
|
'include_me': "Hello {{name}}, This is a jinja template too!",
|
|
'name': 'World'}
|
|
|
|
with mail.record_messages() as outbox:
|
|
workflow_api = self.complete_form(workflow, task, data)
|
|
|
|
self.assertIn('Hello World, This is a jinja template too!', outbox[0].body)
|
|
self.assertIn("# Email Model", workflow_api.next_task.documentation)
|
|
self.assertIn("My Email Subject", workflow_api.next_task.documentation)
|
|
self.assertIn("user@example.com", workflow_api.next_task.documentation)
|
|
|
|
print(f'test_jinja_service_email: {workflow_api.next_task.data}')
|
|
|
|
def test_jinja_service_tools_markdown(self):
|
|
template = "This is my template. {% include 'include_me' %} Was something included?"
|
|
data = {"name": "World",
|
|
"include_me": "Hello {{name}}!"}
|
|
rv = self.app.get('/v1.0/render_markdown?template=%s&data=%s' %
|
|
(template, json.dumps(data)))
|
|
self.assert_success(rv)
|
|
self.assertIn("Hello World", rv.get_data(as_text=True))
|
|
|
|
def test_jinja_service_word_documents(self):
|
|
filepath = os.path.join(app.root_path, '..', 'tests', 'data', 'template.docx')
|
|
with open(filepath, 'rb') as myfile:
|
|
file_data = BytesIO(myfile.read())
|
|
context = {'title': 'My Title', 'my_list': ["a", "b", "c"], 'show_table': True}
|
|
result = JinjaService().make_template(file_data, context)
|
|
self.assertIsNotNone(result) # Not a lot we can do here, just assure there is not an error.
|
|
|
|
def test_jinja_service_word_document_errors_are_sensible(self):
|
|
filepath = os.path.join(app.root_path, '..', 'tests', 'data', 'template_error.docx')
|
|
with open(filepath, 'rb') as myfile:
|
|
file_data = BytesIO(myfile.read())
|
|
context = {'title': 'My Title', 'my_list': ["a", "b", "c"], 'show_table': True}
|
|
with self.assertRaises(ApiError) as ae:
|
|
result = JinjaService().make_template(file_data, context)
|
|
self.assertIn('{{% no_such_variable_error ! @ __ %}}', ae.exception.error_line)
|
|
self.assertEquals("Word Document creation error : unexpected '%'", ae.exception.message)
|
|
self.assertEquals(14, ae.exception.line_number)
|
|
|
|
|
|
|
|
def test_jinja_service_properties(self):
|
|
pass
|