216 lines
9.1 KiB
Python
216 lines
9.1 KiB
Python
from rpa_yisaiqi.contract_flow_steps import (
|
|
ContractFlowContext,
|
|
ContractGoodsLine,
|
|
run_contract_create_flow,
|
|
)
|
|
|
|
|
|
class FakeBrowser:
|
|
def __init__(self):
|
|
self.calls = []
|
|
self.read_values = {
|
|
"contract_number": "HT-20260811-REAL-001",
|
|
}
|
|
|
|
def open_url(self, url):
|
|
self.calls.append(("open_url", url))
|
|
|
|
def fill_first(self, selectors, value):
|
|
self.calls.append(("fill_first", tuple(selectors), value))
|
|
|
|
def click_first(self, selectors):
|
|
self.calls.append(("click_first", tuple(selectors)))
|
|
|
|
def wait_for_text(self, text, timeout_seconds=15):
|
|
self.calls.append(("wait_for_text", text, timeout_seconds))
|
|
|
|
def select_first(self, selectors, value):
|
|
self.calls.append(("select_first", tuple(selectors), value))
|
|
|
|
def upload_file_first(self, selectors, file_path):
|
|
self.calls.append(("upload_file_first", tuple(selectors), file_path))
|
|
|
|
def read_first(self, selectors, value_name):
|
|
self.calls.append(("read_first", tuple(selectors), value_name))
|
|
return self.read_values[value_name]
|
|
|
|
def screenshot(self, name):
|
|
self.calls.append(("screenshot", name))
|
|
return f"/tmp/{name}.png"
|
|
|
|
|
|
def test_contract_flow_logs_in_with_real_contract_login_selectors():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="合同节点联调供应商",
|
|
order_date="2026-08-11",
|
|
source_type="XYY",
|
|
template_type="FORMAT",
|
|
goods_lines=[],
|
|
)
|
|
|
|
run_contract_create_flow(browser, ctx)
|
|
|
|
assert browser.calls[:5] == [
|
|
("open_url", "http://contract-test.gzmpc.com:8080/"),
|
|
("fill_first", ('input[name="j_username"]', 'input[placeholder="用户名"]'), "test002"),
|
|
("fill_first", ('input[name="j_password"]', 'input[placeholder="密码"]'), "1"),
|
|
("click_first", ('input[name="btn_submit"]', 'a[href*="btn_submit"]', 'text=登录')),
|
|
("wait_for_text", "合同", 20),
|
|
]
|
|
|
|
|
|
def test_contract_flow_uses_order_date_and_uploads_yyc_merged_pdf():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="合同节点联调供应商",
|
|
order_date="2026-08-11",
|
|
source_type="YYC",
|
|
template_type="FORMAT",
|
|
goods_lines=[
|
|
ContractGoodsLine(
|
|
operation_code="ADD",
|
|
goods_code="YP-001",
|
|
goods_name="阿卡波糖片",
|
|
approval_no="国药准字H00000001",
|
|
spec="50mg*30片",
|
|
unit="盒",
|
|
quantity=12,
|
|
)
|
|
],
|
|
merged_pdf_path="C:/rpa/jobs/JOB-1/contract-merged.pdf",
|
|
)
|
|
|
|
contract_no = run_contract_create_flow(browser, ctx)
|
|
|
|
assert contract_no == "HT-20260811-REAL-001"
|
|
assert ("fill_first", ('label=合同签订日期', 'input[name="fdSignDate"]', 'input[placeholder*="合同签订"]'), "2026-08-11") in browser.calls
|
|
assert ("fill_first", ('label=合同开始日期', 'input[name="fdStartDate"]', 'input[placeholder*="合同开始"]'), "2026-08-11") in browser.calls
|
|
assert ("upload_file_first", ('input[type="file"]', 'label=合同证文上传'), "C:/rpa/jobs/JOB-1/contract-merged.pdf") in browser.calls
|
|
|
|
|
|
def test_contract_flow_skips_pdf_upload_for_xyy():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="小药药供应商",
|
|
order_date="2026-08-11",
|
|
source_type="XYY",
|
|
template_type="FORMAT",
|
|
goods_lines=[],
|
|
merged_pdf_path="C:/should-not-upload.pdf",
|
|
)
|
|
|
|
run_contract_create_flow(browser, ctx)
|
|
|
|
assert not [call for call in browser.calls if call[0] == "upload_file_first"]
|
|
|
|
|
|
def test_contract_flow_enters_real_draft_menu_and_recent_template():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="重庆亿昊药业",
|
|
order_date="2026-08-11",
|
|
source_type="YYC",
|
|
template_type="NON_FORMAT",
|
|
template_name="非格式--采购协议",
|
|
goods_lines=[],
|
|
merged_pdf_path="C:/rpa/jobs/JOB-1/contract-merged.pdf",
|
|
)
|
|
|
|
run_contract_create_flow(browser, ctx)
|
|
|
|
assert ("click_first", ("text=合同管理", "aria=合同管理")) in browser.calls
|
|
assert ("click_first", ("text=合同起草", "aria=合同起草")) in browser.calls
|
|
assert ("wait_for_text", "最近使用模板", 20) in browser.calls
|
|
assert (
|
|
"click_first",
|
|
("text=非格式--采购协议", "text=非格式--采购协议...", "title=非格式--采购协议"),
|
|
) in browser.calls
|
|
|
|
|
|
def test_contract_flow_fills_real_contract_draft_form_fields_from_screenshot():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="重庆亿昊药业",
|
|
order_date="2026-08-11",
|
|
source_type="XYY",
|
|
template_type="NON_FORMAT",
|
|
template_name="非格式--采购协议",
|
|
customer_code="CUST-001",
|
|
customer_name="重庆亿昊药业",
|
|
contract_category="采购协议",
|
|
settlement_type="付款",
|
|
contract_background="按订单自动起草采购协议",
|
|
contract_nature="主合同",
|
|
amount_currency="人民币",
|
|
amount="12000.50",
|
|
contract_end_date="2026-12-31",
|
|
is_major_contract="否",
|
|
is_prepayment="否",
|
|
belong_year="2026",
|
|
payment_method="账期支付",
|
|
payment_days="30",
|
|
contract_status="正式合同",
|
|
contains_blood_products="否",
|
|
goods_lines=[],
|
|
)
|
|
|
|
run_contract_create_flow(browser, ctx)
|
|
|
|
assert ("fill_first", ('label=客户编码', 'input[name="fdCustomerCode"]', 'input[placeholder*="客户编码"]'), "CUST-001") in browser.calls
|
|
assert ("fill_first", ('label=客户名称', 'input[name="fdCustomerName"]', 'input[placeholder*="客户名称"]'), "重庆亿昊药业") in browser.calls
|
|
assert ("select_first", ('label=合同类别', 'select[name="fdContractCategory"]'), "采购协议") in browser.calls
|
|
assert ("select_first", ('label=结算类型', 'select[name="fdSettlementType"]'), "付款") in browser.calls
|
|
assert ("fill_first", ('label=合同背景概述', 'textarea[name="fdContractBackground"]'), "按订单自动起草采购协议") in browser.calls
|
|
assert ("click_first", ("label=主合同", "text=主合同")) in browser.calls
|
|
assert ("select_first", ('label=合同金额', 'select[name="fdCurrency"]'), "人民币") in browser.calls
|
|
assert ("fill_first", ('label=金额（元）', 'input[name="fdAmount"]'), "12000.50") in browser.calls
|
|
assert ("fill_first", ('label=合同签订日期', 'input[name="fdSignDate"]', 'input[placeholder*="合同签订"]'), "2026-08-11") in browser.calls
|
|
assert ("fill_first", ('label=合同开始日期', 'input[name="fdStartDate"]', 'input[placeholder*="合同开始"]'), "2026-08-11") in browser.calls
|
|
assert ("fill_first", ('label=合同结束日期', 'input[name="fdEndDate"]', 'input[placeholder*="合同结束"]'), "2026-12-31") in browser.calls
|
|
assert ("select_first", ('label=是否重大合同', 'select[name="fdIsMajorContract"]'), "否") in browser.calls
|
|
assert ("select_first", ('label=是否预付款', 'select[name="fdIsPrepayment"]'), "否") in browser.calls
|
|
assert ("fill_first", ('label=合同归属年份', 'input[name="fdBelongYear"]'), "2026") in browser.calls
|
|
assert ("select_first", ('label=付款方式', 'select[name="fdPaymentMethod"]', 'select[name="paymentMethod"]'), "账期支付") in browser.calls
|
|
assert ("fill_first", ('label=付款期限', 'input[name="fdPaymentDays"]', 'input[name="paymentDays"]'), "30") in browser.calls
|
|
assert ("select_first", ('label=合同状态', 'select[name="fdContractStatus"]'), "正式合同") in browser.calls
|
|
assert ("select_first", ('label=是否包含血液制品', 'select[name="fdContainsBloodProducts"]'), "否") in browser.calls
|
|
|
|
|
|
def test_contract_flow_dry_run_skips_submit_but_still_reads_page_state():
|
|
browser = FakeBrowser()
|
|
ctx = ContractFlowContext(
|
|
contract_url="http://contract-test.gzmpc.com:8080/",
|
|
username="test002",
|
|
password="1",
|
|
supplier_name="重庆亿昊药业",
|
|
order_date="2026-08-11",
|
|
source_type="XYY",
|
|
template_type="NON_FORMAT",
|
|
template_name="非格式--采购协议",
|
|
submit_mode="dry_run",
|
|
goods_lines=[],
|
|
)
|
|
|
|
contract_no = run_contract_create_flow(browser, ctx)
|
|
|
|
assert not [call for call in browser.calls if call[0] == "click_first" and "text=提交" in call[1]]
|
|
assert ("screenshot", "contract-dry-run-ready") in browser.calls
|
|
assert contract_no == ""
|
|
assert not [call for call in browser.calls if call[0] == "read_first"]
|