Files
inquiry_robot/inquiry-agent/tests/test_quote_sheet_filename.py

51 lines
1.7 KiB
Python

"""系统报价单对外文件名:工单号接后台模板名,并带文件后缀。"""
from __future__ import annotations
import os
import sys
import unittest
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
if ROOT not in sys.path:
sys.path.insert(0, ROOT)
from agent.quote_templates.filename import filename_from_render, quote_sheet_filename
class QuoteSheetFilenameTests(unittest.TestCase):
def test_joins_work_order_and_backend_template_name(self) -> None:
self.assertEqual(
quote_sheet_filename("WO202609100001", "陆运国内运输拼车报价单"),
"WO202609100001陆运国内运输拼车报价单.xlsx",
)
def test_drops_suffix_on_template_name(self) -> None:
self.assertEqual(
quote_sheet_filename("WO202609100001", "海运报价单模板.xlsx"),
"WO202609100001海运报价单模板.xlsx",
)
def test_pdf_uses_pdf_suffix(self) -> None:
self.assertEqual(
quote_sheet_filename("WO202609100001", "默认空运", suffix=".pdf"),
"WO202609100001默认空运.pdf",
)
def test_render_prefers_template_name_over_old_file_name(self) -> None:
shown = filename_from_render(
"WO202609100001",
{
"templateName": "国内运输拼车报价单",
"fileName": "WO202609100001_quote.xlsx",
},
)
self.assertEqual(shown, "WO202609100001国内运输拼车报价单.xlsx")
def test_blank_template_keeps_xlsx_suffix(self) -> None:
shown = filename_from_render(
"WO202609100001",
{"fileName": "WO202609100001_quote.xlsx"},
)
self.assertEqual(shown, "WO202609100001_quote.xlsx")