🎉 Sync 2026-09-27 21:03:26

luci-app-substore/tests/node_group_test.lua

@@ -0,0 +1,132 @@
+-- node_group_test.lua — 节点分组与标签测试
+package.path = "./root/usr/share/?.lua;" .. package.path
+
+local node = require("substore.node")
+
+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 function eq(a, b)
+	if type(a) ~= type(b) then return false end
+	if type(a) == "table" then
+		for k, v in pairs(a) do if not eq(v, b[k]) then return false end end
+		for k in pairs(b) do if a[k] == nil then return false end end
+		return true
+	end
+	return a == b
+end
+
+-- 测试数据
+local nodes = {
+	{ name = "HK VMESS 1", proto = "vmess", server = "1.1.1.1", port = 443 },
+	{ name = "HK VMESS 2", proto = "vmess", server = "1.1.1.2", port = 443 },
+	{ name = "US TROJAN 1", proto = "trojan", server = "2.2.2.2", port = 443 },
+	{ name = "JP SS 1", proto = "shadowsocks", server = "3.3.3.3", port = 8388 },
+	{ name = "HK TROJAN 2", proto = "trojan", server = "1.1.1.1", port = 443 },
+}
+
+-- ---------- group_by ----------
+check("group_by proto exists", type(node.group_by) == "function")
+if type(node.group_by) == "function" then
+	local groups = node.group_by(nodes, "proto")
+	check("group_by proto count", groups ~= nil and groups.vmess ~= nil and #groups.vmess == 2)
+	check("group_by proto trojan", groups ~= nil and groups.trojan ~= nil and #groups.trojan == 2)
+	check("group_by proto shadowsocks", groups ~= nil and groups.shadowsocks ~= nil and #groups.shadowsocks == 1)
+
+	local groups_server = node.group_by(nodes, "server")
+	check("group_by server 1.1.1.1", groups_server ~= nil and groups_server["1.1.1.1"] ~= nil and #groups_server["1.1.1.1"] == 2)
+	check("group_by server 2.2.2.2", groups_server ~= nil and groups_server["2.2.2.2"] ~= nil and #groups_server["2.2.2.2"] == 1)
+end
+
+-- ---------- add_group ----------
+check("add_group exists", type(node.add_group) == "function")
+if type(node.add_group) == "function" then
+	local nodes_copy = {}
+	for _, n in ipairs(nodes) do
+		nodes_copy[#nodes_copy + 1] = { name = n.name, proto = n.proto, server = n.server, port = n.port }
+	end
+
+	-- 按协议分组
+	local rules_proto = {
+		{ field = "proto", prefix = "proto_" }
+	}
+	local result = node.add_group(nodes_copy, rules_proto)
+	check("add_group proto field exists", result[1].group ~= nil)
+	check("add_group proto value", result[1].group == "proto_vmess")
+
+	-- 按服务器分组
+	local nodes_copy2 = {}
+	for _, n in ipairs(nodes) do
+		nodes_copy2[#nodes_copy2 + 1] = { name = n.name, proto = n.proto, server = n.server, port = n.port }
+	end
+	local rules_server = {
+		{ field = "server", prefix = "server_" }
+	}
+	local result2 = node.add_group(nodes_copy2, rules_server)
+	check("add_group server field exists", result2[1].group ~= nil)
+	check("add_group server value", result2[1].group == "server_1.1.1.1")
+end
+
+-- ---------- add_tags ----------
+check("add_tags exists", type(node.add_tags) == "function")
+if type(node.add_tags) == "function" then
+	local nodes_copy = {}
+	for _, n in ipairs(nodes) do
+		nodes_copy[#nodes_copy + 1] = { name = n.name, proto = n.proto, server = n.server, port = n.port }
+	end
+
+	-- 按关键词添加标签
+	local rules_keyword = {
+		{ type = "keyword", value = "HK", tag = "hongkong" }
+	}
+	local result = node.add_tags(nodes_copy, rules_keyword)
+	check("add_tags keyword exists", result[1].tags ~= nil and type(result[1].tags) == "table")
+	check("add_tags keyword value", result[1].tags ~= nil and result[1].tags.hongkong == true)
+	check("add_tags keyword not match", result[3].tags ~= nil and result[3].tags.hongkong == nil)
+end
+
+-- ---------- filter_by_group ----------
+check("filter_by_group exists", type(node.filter_by_group) == "function")
+if type(node.filter_by_group) == "function" then
+	local nodes_copy = {}
+	for _, n in ipairs(nodes) do
+		nodes_copy[#nodes_copy + 1] = { name = n.name, proto = n.proto, server = n.server, port = n.port, group = (n.proto == "vmess") and "proto_vmess" or "other" }
+	end
+
+	local filtered = node.filter_by_group(nodes_copy, "proto_vmess")
+	check("filter_by_group count", filtered ~= nil and #filtered == 2)
+
+	local filtered_none = node.filter_by_group(nodes_copy, "proto_trojan")
+	check("filter_by_group none", filtered_none ~= nil and #filtered_none == 0)
+end
+
+-- ---------- filter_by_tags ----------
+check("filter_by_tags exists", type(node.filter_by_tags) == "function")
+if type(node.filter_by_tags) == "function" then
+	local nodes_copy = {}
+	for _, n in ipairs(nodes) do
+		nodes_copy[#nodes_copy + 1] = { name = n.name, proto = n.proto, server = n.server, port = n.port, tags = {} }
+	end
+	nodes_copy[1].tags.hongkong = true
+	nodes_copy[2].tags.hongkong = true
+	nodes_copy[3].tags.usa = true
+
+	local filtered = node.filter_by_tags(nodes_copy, { "hongkong" })
+	check("filter_by_tags count", filtered ~= nil and #filtered == 2)
+
+	local filtered_none = node.filter_by_tags(nodes_copy, { "japan" })
+	check("filter_by_tags none", filtered_none ~= nil and #filtered_none == 0)
+end
+
+-- 结果
+print(string.format("\n%d passed, %d failed", passed, failed))
+os.exit(failed == 0 and 0 or 1)