291 lines
11 KiB
Python
291 lines
11 KiB
Python
"""OA 发起预览：条件字段门闩与单路 walk"""
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from app.services.oa_engine import (
|
|
INITIATOR_CONDITION_FIELD,
|
|
_FLOW_PREVIEW_PENDING_MSG,
|
|
_collect_condition_fields,
|
|
_condition_fields_ready,
|
|
_form_value_present,
|
|
_iter_resolved_path_nodes,
|
|
_walk_preview_nodes,
|
|
)
|
|
|
|
|
|
def _cond_flow():
|
|
return {
|
|
"id": "s",
|
|
"type": "start",
|
|
"name": "发起人",
|
|
"child": {
|
|
"id": "a0",
|
|
"type": "approval",
|
|
"name": "一级审批",
|
|
"approverConfig": {"assigneeType": "users", "userIds": ["u1"], "signMode": "or"},
|
|
"child": {
|
|
"id": "c1",
|
|
"type": "condition",
|
|
"conditionGroups": [
|
|
{
|
|
"name": "条件1",
|
|
"rules": [{"field": "days", "op": ">", "value": 3}],
|
|
"child": {
|
|
"id": "a1",
|
|
"type": "approval",
|
|
"name": "董事长审批",
|
|
"approverConfig": {
|
|
"assigneeType": "users",
|
|
"userIds": ["u2"],
|
|
"signMode": "or",
|
|
},
|
|
"child": {"id": "e1", "type": "end", "name": "结束"},
|
|
},
|
|
},
|
|
{
|
|
"name": "默认",
|
|
"isDefault": True,
|
|
"rules": [],
|
|
"child": {"id": "e2", "type": "end", "name": "结束"},
|
|
},
|
|
],
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
class TestFormValuePresent(unittest.TestCase):
|
|
def test_empty(self):
|
|
self.assertFalse(_form_value_present({}, "days"))
|
|
self.assertFalse(_form_value_present({"days": None}, "days"))
|
|
self.assertFalse(_form_value_present({"days": ""}, "days"))
|
|
self.assertFalse(_form_value_present({"days": " "}, "days"))
|
|
self.assertFalse(_form_value_present({"days": []}, "days"))
|
|
|
|
def test_present(self):
|
|
self.assertTrue(_form_value_present({"days": 0}, "days"))
|
|
self.assertTrue(_form_value_present({"ok": False}, "ok"))
|
|
self.assertTrue(_form_value_present({"days": 5}, "days"))
|
|
self.assertTrue(_form_value_present({"t": "事假"}, "t"))
|
|
|
|
|
|
class TestCollectConditionFields(unittest.TestCase):
|
|
def test_linear_no_fields(self):
|
|
root = {
|
|
"id": "s",
|
|
"type": "start",
|
|
"child": {"id": "e", "type": "end"},
|
|
}
|
|
self.assertEqual(_collect_condition_fields(root), [])
|
|
self.assertTrue(_condition_fields_ready(root, {}))
|
|
|
|
def test_rules_field(self):
|
|
root = _cond_flow()
|
|
self.assertEqual(_collect_condition_fields(root), ["days"])
|
|
self.assertFalse(_condition_fields_ready(root, {}))
|
|
self.assertTrue(_condition_fields_ready(root, {"days": 5}))
|
|
|
|
def test_when_formula_fields(self):
|
|
root = {
|
|
"type": "condition",
|
|
"conditionGroups": [
|
|
{"name": "x", "when": "{days} > 3 and {leaveType} == '事假'", "child": {}},
|
|
{"name": "d", "isDefault": True, "rules": []},
|
|
],
|
|
}
|
|
self.assertEqual(_collect_condition_fields(root), ["days", "leaveType"])
|
|
|
|
def test_initiator_system_field_not_gated_by_form_data(self):
|
|
"""系统伪字段 __initiator__ 不依赖 formData，空表单也应 resolved 门闩通过。"""
|
|
root = {
|
|
"id": "s",
|
|
"type": "start",
|
|
"child": {
|
|
"id": "c1",
|
|
"type": "condition",
|
|
"conditionGroups": [
|
|
{
|
|
"name": "条件1",
|
|
"rules": [
|
|
{
|
|
"field": INITIATOR_CONDITION_FIELD,
|
|
"op": "contains",
|
|
"value": ["u1"],
|
|
}
|
|
],
|
|
"child": {"id": "e1", "type": "end"},
|
|
},
|
|
{
|
|
"name": "默认",
|
|
"isDefault": True,
|
|
"rules": [],
|
|
"child": {"id": "e2", "type": "end"},
|
|
},
|
|
],
|
|
},
|
|
}
|
|
self.assertEqual(_collect_condition_fields(root), [])
|
|
self.assertTrue(_condition_fields_ready(root, {}))
|
|
|
|
def test_initiator_plus_form_field_still_gates_form(self):
|
|
root = {
|
|
"type": "condition",
|
|
"conditionGroups": [
|
|
{
|
|
"name": "条件1",
|
|
"rules": [
|
|
{
|
|
"field": INITIATOR_CONDITION_FIELD,
|
|
"op": "contains",
|
|
"value": ["u1"],
|
|
},
|
|
{"field": "days", "op": ">", "value": 3},
|
|
],
|
|
"child": {},
|
|
},
|
|
{"name": "默认", "isDefault": True, "rules": []},
|
|
],
|
|
}
|
|
self.assertEqual(_collect_condition_fields(root), ["days"])
|
|
self.assertFalse(_condition_fields_ready(root, {}))
|
|
self.assertTrue(_condition_fields_ready(root, {"days": 5}))
|
|
|
|
|
|
class TestWalkPreviewResolvedPath(unittest.TestCase):
|
|
def test_hit_condition1(self):
|
|
db = MagicMock()
|
|
steps = []
|
|
with patch("app.services.oa_engine.resolve_assignees", return_value=(["u1"], {})), patch(
|
|
"app.services.oa_engine._preview_people",
|
|
side_effect=lambda _db, ids: [{"id": i, "nickName": i} for i in ids],
|
|
), patch("app.services.oa_engine._assignee_meta", return_value="指定成员"):
|
|
ok = _walk_preview_nodes(
|
|
db,
|
|
node=_cond_flow(),
|
|
tenant_no="t1",
|
|
initiator_user_id="init",
|
|
initiator_dept_id=None,
|
|
advanced={},
|
|
initiator_select_by_node=None,
|
|
form_data={"days": 5},
|
|
steps=steps,
|
|
)
|
|
self.assertTrue(ok)
|
|
types = [s["type"] for s in steps]
|
|
names = [s["name"] for s in steps]
|
|
self.assertNotIn("branch", types)
|
|
self.assertEqual(names, ["发起人", "一级审批", "董事长审批", "结束"])
|
|
|
|
def test_hit_default(self):
|
|
db = MagicMock()
|
|
steps = []
|
|
with patch("app.services.oa_engine.resolve_assignees", return_value=(["u1"], {})), patch(
|
|
"app.services.oa_engine._preview_people",
|
|
side_effect=lambda _db, ids: [{"id": i, "nickName": i} for i in ids],
|
|
), patch("app.services.oa_engine._assignee_meta", return_value="指定成员"):
|
|
ok = _walk_preview_nodes(
|
|
db,
|
|
node=_cond_flow(),
|
|
tenant_no="t1",
|
|
initiator_user_id="init",
|
|
initiator_dept_id=None,
|
|
advanced={},
|
|
initiator_select_by_node=None,
|
|
form_data={"days": 1},
|
|
steps=steps,
|
|
)
|
|
self.assertTrue(ok)
|
|
names = [s["name"] for s in steps]
|
|
self.assertEqual(names, ["发起人", "一级审批", "结束"])
|
|
self.assertNotIn("董事长审批", names)
|
|
|
|
def test_pending_message_constant(self):
|
|
self.assertEqual(_FLOW_PREVIEW_PENDING_MSG, "必填信息填写后，流程将自动显示")
|
|
|
|
def test_merge_after_condition_skips_branch_end(self):
|
|
"""条件分支叶子 end 后若有汇合主链，预览只保留真正的结束节点。"""
|
|
root = {
|
|
"id": "s",
|
|
"type": "start",
|
|
"name": "发起人",
|
|
"child": {
|
|
"id": "c1",
|
|
"type": "condition",
|
|
"conditionGroups": [
|
|
{
|
|
"name": "条件1",
|
|
"rules": [{"field": "days", "op": ">", "value": 3}],
|
|
"child": {
|
|
"id": "a1",
|
|
"type": "approval",
|
|
"name": "领导会签",
|
|
"approverConfig": {
|
|
"assigneeType": "users",
|
|
"userIds": ["u2"],
|
|
"signMode": "and",
|
|
},
|
|
"child": {"id": "e_branch", "type": "end", "name": "结束"},
|
|
},
|
|
},
|
|
{
|
|
"name": "默认",
|
|
"isDefault": True,
|
|
"rules": [],
|
|
"child": {"id": "e_def", "type": "end", "name": "结束"},
|
|
},
|
|
],
|
|
"child": {
|
|
"id": "cc1",
|
|
"type": "cc",
|
|
"name": "抄送人",
|
|
"approverConfig": {
|
|
"assigneeTypes": ["initiator_select"],
|
|
"assigneeType": "initiator_select",
|
|
},
|
|
"child": {"id": "e_final", "type": "end", "name": "结束"},
|
|
},
|
|
},
|
|
}
|
|
db = MagicMock()
|
|
steps = []
|
|
with patch(
|
|
"app.services.oa_engine.resolve_assignees", return_value=(["u2"], {})
|
|
), patch(
|
|
"app.services.oa_engine.resolve_cc_assignees", return_value=([], {})
|
|
), patch(
|
|
"app.services.oa_engine._preview_people",
|
|
side_effect=lambda _db, ids: [{"id": i, "nickName": i} for i in ids],
|
|
), patch(
|
|
"app.services.oa_engine._assignee_meta", return_value="指定成员"
|
|
), patch(
|
|
"app.services.oa_engine._assignee_meta_for_cc", return_value="发起人自选"
|
|
):
|
|
ok = _walk_preview_nodes(
|
|
db,
|
|
node=root,
|
|
tenant_no="t1",
|
|
initiator_user_id="init",
|
|
initiator_dept_id=None,
|
|
advanced={},
|
|
initiator_select_by_node=None,
|
|
form_data={"days": 5},
|
|
steps=steps,
|
|
)
|
|
self.assertTrue(ok)
|
|
names = [s["name"] for s in steps]
|
|
types = [s["type"] for s in steps]
|
|
self.assertEqual(names, ["发起人", "领导会签", "抄送人", "结束"])
|
|
self.assertEqual(types.count("end"), 1)
|
|
self.assertEqual(steps[-1]["id"], "e_final")
|
|
|
|
path = _iter_resolved_path_nodes(root, {"days": 5}, initiator_user_id="init")
|
|
self.assertIsNotNone(path)
|
|
path_ids = [n.get("id") for n in path]
|
|
self.assertNotIn("e_branch", path_ids)
|
|
self.assertIn("e_final", path_ids)
|
|
self.assertEqual(sum(1 for n in path if n.get("type") == "end"), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|