113 lines
4.9 KiB
Python
113 lines
4.9 KiB
Python
from tests.base_test import BaseTest
|
|
from crc import session
|
|
from crc.models.file import FileDataModel, FileModel, LookupFileModel, LookupDataModel
|
|
from crc.services.lookup_service import LookupService
|
|
from crc.services.workflow_processor import WorkflowProcessor
|
|
from crc.services.workflow_service import WorkflowService
|
|
|
|
|
|
class TestWorkflowService(BaseTest):
|
|
|
|
def test_documentation_processing_handles_replacements(self):
|
|
self.load_example_data()
|
|
workflow = self.create_workflow('random_fact')
|
|
processor = WorkflowProcessor(workflow)
|
|
processor.do_engine_steps()
|
|
|
|
task = processor.next_task()
|
|
task.task_spec.documentation = "Some simple docs"
|
|
docs = WorkflowService._process_documentation(task)
|
|
self.assertEqual("Some simple docs", docs)
|
|
|
|
task.data = {"replace_me": "new_thing"}
|
|
task.task_spec.documentation = "{{replace_me}}"
|
|
docs = WorkflowService._process_documentation(task)
|
|
self.assertEqual("new_thing", docs)
|
|
|
|
documentation = """
|
|
# Bigger Test
|
|
|
|
* bullet one
|
|
* bullet two has {{replace_me}}
|
|
|
|
# other stuff.
|
|
"""
|
|
expected = """
|
|
# Bigger Test
|
|
|
|
* bullet one
|
|
* bullet two has new_thing
|
|
|
|
# other stuff.
|
|
"""
|
|
task.task_spec.documentation = documentation
|
|
result = WorkflowService._process_documentation(task)
|
|
self.assertEqual(expected, result)
|
|
|
|
def test_documentation_processing_handles_conditionals(self):
|
|
|
|
self.load_example_data()
|
|
workflow = self.create_workflow('random_fact')
|
|
processor = WorkflowProcessor(workflow)
|
|
processor.do_engine_steps()
|
|
|
|
task = processor.next_task()
|
|
task.task_spec.documentation = "This test {% if works == 'yes' %}works{% endif %}"
|
|
docs = WorkflowService._process_documentation(task)
|
|
self.assertEqual("This test ", docs)
|
|
|
|
task.data = {"works": 'yes'}
|
|
docs = WorkflowService._process_documentation(task)
|
|
self.assertEqual("This test works", docs)
|
|
|
|
def test_enum_options_from_file(self):
|
|
self.load_example_data()
|
|
workflow = self.create_workflow('enum_options_from_file')
|
|
processor = WorkflowProcessor(workflow)
|
|
processor.do_engine_steps()
|
|
task = processor.next_task()
|
|
WorkflowService.process_options(task, task.task_spec.form.fields[0])
|
|
options = task.task_spec.form.fields[0].options
|
|
self.assertEquals(19, len(options))
|
|
self.assertEquals('1000', options[0]['id'])
|
|
self.assertEquals("UVA - INTERNAL - GM USE ONLY", options[0]['name'])
|
|
|
|
def test_create_lookup_file(self):
|
|
spec = self.load_test_spec('enum_options_from_file')
|
|
file_model = session.query(FileModel).filter(FileModel.name == "customer_list.xls").first()
|
|
file_data_model = session.query(FileDataModel).filter(FileDataModel.file_model == file_model).first()
|
|
LookupService.get_lookup_table_from_data_model(file_data_model, "CUSTOMER_NUMBER", "CUSTOMER_NAME")
|
|
lookup_records = session.query(LookupFileModel).all()
|
|
self.assertIsNotNone(lookup_records)
|
|
self.assertEqual(1, len(lookup_records))
|
|
lookup_record = lookup_records[0]
|
|
self.assertIsNotNone(lookup_record)
|
|
self.assertEquals("CUSTOMER_NUMBER", lookup_record.value_column)
|
|
self.assertEquals("CUSTOMER_NAME", lookup_record.label_column)
|
|
self.assertEquals("CUSTOMER_NAME", lookup_record.label_column)
|
|
lookup_data = session.query(LookupDataModel).filter(LookupDataModel.lookup_file_model == lookup_record).all()
|
|
self.assertEquals(19, len(lookup_data))
|
|
|
|
self.assertEquals("1000", lookup_data[0].value)
|
|
self.assertEquals("UVA - INTERNAL - GM USE ONLY", lookup_data[0].label)
|
|
# search_results = session.query(LookupDataModel).\
|
|
# filter(LookupDataModel.lookup_file_model_id == lookup_record.id).\
|
|
# filter(LookupDataModel.__ts_vector__.op('@@')(func.plainto_tsquery('INTERNAL'))).all()
|
|
search_results = LookupDataModel.query.filter(LookupDataModel.label.match("INTERNAL")).all()
|
|
self.assertEquals(1, len(search_results))
|
|
search_results = LookupDataModel.query.filter(LookupDataModel.label.match("internal")).all()
|
|
self.assertEquals(1, len(search_results))
|
|
# This query finds results where a word starts with "bio"
|
|
search_results = LookupDataModel.query.filter(LookupDataModel.label.match("bio:*")).all()
|
|
self.assertEquals(2, len(search_results))
|
|
|
|
|
|
def test_random_data_populate_form_on_auto_complete(self):
|
|
self.load_example_data()
|
|
workflow = self.create_workflow('enum_options_with_search')
|
|
processor = WorkflowProcessor(workflow)
|
|
processor.do_engine_steps()
|
|
task = processor.next_task()
|
|
task_api = WorkflowService.spiff_task_to_api_task(task, add_docs_and_forms=True)
|
|
WorkflowService.populate_form_with_random_data(task, task_api)
|
|
self.assertTrue(isinstance(task.data["sponsor"], dict)) |