"""
企微加解密自检（无外网）。

运行：在 agent 目录 .venv 下 python -m agent.channel.wecom.selftest
需已配置 WECOM_CALLBACK_TOKEN / WECOM_ENCODING_AES_KEY / WECOM_CORP_ID。
"""

from __future__ import annotations

from agent.channel.wecom.crypto import WeComCrypto
from agent.config import get_settings


def main() -> None:
    s = get_settings()
    c = WeComCrypto(s.wecom_callback_token, s.wecom_encoding_aes_key, s.wecom_corp_id)
    plain = "<xml><FromUserName><![CDATA[Alice]]></FromUserName><MsgType><![CDATA[text]]></MsgType><Content><![CDATA[hi]]></Content><MsgId>1</MsgId></xml>"
    enc = c.encrypt(plain)
    dec = c.decrypt(enc)
    assert "Alice" in dec and "hi" in dec
    print("crypto_ok")


if __name__ == "__main__":
    main()