# -*- coding: utf-8 -*-
"""打包后 EXE 冒烟测试。

确认这些确实进了包：批量导入入口与模板、列表编辑入口、弹窗样式，
以及「新建订单明细为空时仍能添加行」所依赖的 rowTpl 模板。

用法：python verify_exe.py 8080 [数据库路径]
      不传数据库路径时默认 po_app/dist/instance/po.db
"""
import os
import re
import sqlite3
import sys

import requests

BASE = "http://127.0.0.1:%s" % (sys.argv[1] if len(sys.argv) > 1 else "8080")
HERE = os.path.dirname(os.path.abspath(__file__))
DB = sys.argv[2] if len(sys.argv) > 2 else os.path.join(
    os.path.dirname(HERE), "po_app", "dist", "instance", "po.db")

s = requests.Session()
FAILS = []


def check(label, cond, extra=""):
    print(("  OK   " if cond else "  FAIL ") + label + ("  " + str(extra) if extra else ""))
    if not cond:
        FAILS.append(label)


def main():
    r = s.post(BASE + "/login", data={"username": "admin", "password": "admin123"})
    check("登录", r.status_code == 200, r.status_code)

    # 注意：url_for() 会被 Jinja 渲染成实际路径，页面里不会出现函数名
    for path, marks in (
        ("/materials", [u"批量导入 Excel", "matImpModal", "/materials/import_excel",
                        "/materials/import_template.xlsx", "openEdit", "editModal",
                        "/toggle", "/delete", u"停用", u"删除"]),
        ("/suppliers", [u"批量导入 Excel", "supImpModal", "/suppliers/import_excel",
                        "/suppliers/import_template.xlsx", "openEdit", "editModal",
                        "/toggle", "/delete", u"停用", u"删除"]),
    ):
        h = s.get(BASE + path).text
        missing = [m for m in marks if m not in h]
        check("%s 含导入/编辑/停用删除/置灰" % path, not missing, missing)

    css = s.get(BASE + "/static/style.css").text
    check("style.css 含弹窗样式", "modal-mask" in css and "modal-box" in css)

    for u in ("/materials/import_template.xlsx", "/suppliers/import_template.xlsx"):
        r = s.get(BASE + u)
        check("模板下载 %s" % u, r.status_code == 200 and len(r.content) > 3000,
              "%s bytes" % len(r.content))

    # 新建订单：明细为空时必须仍能添加行（依赖 rowTpl 模板存在）
    r = s.post(BASE + "/orders/new", data={"doc_type": "purchase"})
    h = r.text
    no = ""
    m = re.search(r"PO\d{12}", h)
    if m:
        no = m.group(0)
    check("新建订单页含 rowTpl 明细行模板", 'id="rowTpl"' in h)
    tpl = re.search(r'<template id="rowTpl">(.*?)</template>', h, re.S)
    check("rowTpl 内含完整空行", bool(tpl) and 'name="item_code"' in tpl.group(1))

    if no and os.path.exists(DB):
        con = sqlite3.connect(DB)
        cur = con.cursor()
        cur.execute("DELETE FROM order_items WHERE order_id IN "
                    "(SELECT id FROM orders WHERE order_no=?)", (no,))
        cur.execute("DELETE FROM orders WHERE order_no=?", (no,))
        con.commit()
        con.close()
        print("    已清理测试订单 %s" % no)

    print("\n" + ("打包版冒烟全部通过" if not FAILS else "失败: " + "; ".join(FAILS)))


if __name__ == "__main__":
    main()