Underground-Digital/Workflow-Engine

1"""2This test file is used to verify the compatibility of Workflow before and after supporting multiple file types.3"""4 5import json6 7from models import Workflow8 9OLD_VERSION_WORKFLOW_FEATURES = {10    "file_upload": {11        "image": {12            "enabled": True,13            "number_limits": 6,14            "transfer_methods": ["remote_url", "local_file"],15        }16    },17    "opening_statement": "",18    "retriever_resource": {"enabled": True},19    "sensitive_word_avoidance": {"enabled": False},20    "speech_to_text": {"enabled": False},21    "suggested_questions": [],22    "suggested_questions_after_answer": {"enabled": False},23    "text_to_speech": {"enabled": False, "language": "", "voice": ""},24}25 26NEW_VERSION_WORKFLOW_FEATURES = {27    "file_upload": {28        "enabled": True,29        "allowed_file_types": ["image"],30        "allowed_extensions": [],31        "allowed_upload_methods": ["remote_url", "local_file"],32        "number_limits": 6,33    },34    "opening_statement": "",35    "retriever_resource": {"enabled": True},36    "sensitive_word_avoidance": {"enabled": False},37    "speech_to_text": {"enabled": False},38    "suggested_questions": [],39    "suggested_questions_after_answer": {"enabled": False},40    "text_to_speech": {"enabled": False, "language": "", "voice": ""},41}42 43 44def test_workflow_features():45    workflow = Workflow(46        tenant_id="",47        app_id="",48        type="",49        version="",50        graph="",51        features=json.dumps(OLD_VERSION_WORKFLOW_FEATURES),52        created_by="",53        environment_variables=[],54        conversation_variables=[],55    )56 57    assert workflow.features_dict == NEW_VERSION_WORKFLOW_FEATURES58