mirror of
https://github.com/kiddin9/op-packages.git
synced 2026-10-27 18:22:52 +08:00
87 lines
3.1 KiB
Lua
87 lines
3.1 KiB
Lua
-- core_link_test.lua — 订阅链接生成（core.generate_link）单元测试
|
|
-- 用法：lua5.1 tests/core_link_test.lua
|
|
|
|
package.path = "./root/usr/share/?.lua;" .. package.path
|
|
|
|
local core = require("substore.core")
|
|
local util = require("substore.util")
|
|
|
|
local passed, failed = 0, 0
|
|
|
|
local function check(name, cond)
|
|
if cond then
|
|
passed = passed + 1
|
|
print("PASS " .. name)
|
|
else
|
|
failed = failed + 1
|
|
print("FAIL " .. name)
|
|
end
|
|
end
|
|
|
|
-- 重定向数据目录到临时目录，避免污染系统
|
|
local tmp = os.tmpname() .. "_substore"
|
|
os.execute("mkdir -p " .. string.format("%q", tmp .. "/nodes"))
|
|
core.DATA_DIR = tmp
|
|
core.LIST_FILE = tmp .. "/subscriptions.json"
|
|
core.NODES_DIR = tmp .. "/nodes"
|
|
|
|
check("module exists", core ~= nil)
|
|
check("generate_link exists", type(core.generate_link) == "function")
|
|
check("ensure_token exists", type(core.ensure_token) == "function")
|
|
|
|
-- 创建订阅并写入节点
|
|
local id, aerr = core.add("测试订阅", "http://example.com/sub")
|
|
check("add subscription", id ~= nil, aerr or "")
|
|
|
|
local token = core.ensure_token(id)
|
|
check("token generated", type(token) == "string" and #token == 16)
|
|
check("token idempotent", core.ensure_token(id) == token)
|
|
|
|
local nodes = {
|
|
{ proto = "vmess", name = "NodeA", server = "1.1.1.1", port = 443, uuid = "u1", net = "tcp" },
|
|
{ proto = "shadowsocks", name = "NodeB", server = "2.2.2.2", port = 8388, method = "aes-256-gcm", password = "p1" },
|
|
{ proto = "wireguard", name = "NodeC", server = "3.3.3.3", port = 51820, ["private-key"] = "K", ["public-key"] = "P", ip = "10.0.0.1/32" },
|
|
}
|
|
check("write nodes", core.write_nodes(id, nodes) == true)
|
|
|
|
-- 各格式生成（列表与 output.FORMAT_OPTIONS 保持一致）
|
|
local output_mod = require("substore.output")
|
|
local targets = {}
|
|
for _, fo in ipairs(output_mod.FORMAT_OPTIONS) do targets[#targets + 1] = fo[1] end
|
|
local ok = true
|
|
for _, target in ipairs(targets) do
|
|
local content, ct, filename, err = core.generate_link(token, target)
|
|
if type(content) == "string" and content ~= "" and ct and filename then
|
|
-- ok
|
|
else
|
|
ok = false
|
|
print(" fail:", target, tostring(err))
|
|
end
|
|
end
|
|
check("all " .. #targets .. " targets generate", ok)
|
|
|
|
local yaml, ct, filename = core.generate_link(token, "ClashMeta")
|
|
check("clashmeta has proxies", yaml and yaml:find("proxies:") ~= nil)
|
|
check("clashmeta content-type", ct == "text/plain; charset=utf-8")
|
|
check("clashmeta filename .yaml", filename and filename:match("%.yaml$") ~= nil)
|
|
|
|
local sb = core.generate_link(token, "singbox")
|
|
check("singbox is json", sb and sb:sub(1, 1) == "{")
|
|
|
|
-- 错误 token
|
|
local bad, _, _, err = core.generate_link("deadbeef", "ClashMeta")
|
|
check("bad token returns nil", bad == nil)
|
|
check("bad token has err", err ~= nil)
|
|
|
|
-- 无节点订阅
|
|
local id2 = core.add("空订阅", "http://example.com/empty")
|
|
local token2 = core.ensure_token(id2)
|
|
local empty, _, _, err2 = core.generate_link(token2, "ClashMeta")
|
|
check("empty nodes returns nil", empty == nil)
|
|
check("empty nodes err", err2 ~= nil)
|
|
|
|
-- 清理
|
|
os.execute("rm -rf " .. string.format("%q", tmp))
|
|
|
|
print(string.format("\n%d passed, %d failed", passed, failed))
|
|
os.exit(failed == 0 and 0 or 1) |