152 lines
4.0 KiB
Python
152 lines
4.0 KiB
Python
import yaml
|
|
from expander import filter_proxy, expand_config, build_mihomo_config
|
|
|
|
PROXY_FULL = {
|
|
"name": "node1",
|
|
"type": "ss",
|
|
"server": "1.2.3.4",
|
|
"port": 443,
|
|
"password": "secret",
|
|
"cipher": "aes-256-gcm",
|
|
"alive": True, # runtime field — should be stripped
|
|
"history": [], # runtime field — should be stripped
|
|
"extra": {"foo": "bar"}, # runtime field — should be stripped
|
|
}
|
|
|
|
BASE_YAML = """
|
|
proxies: []
|
|
proxy-providers:
|
|
provider1:
|
|
type: http
|
|
url: https://example.com/sub
|
|
interval: 3600
|
|
proxy-groups:
|
|
- name: Proxy
|
|
type: select
|
|
use:
|
|
- provider1
|
|
rules:
|
|
- MATCH,DIRECT
|
|
"""
|
|
|
|
|
|
def test_filter_proxy_removes_runtime_fields():
|
|
filtered = filter_proxy(PROXY_FULL)
|
|
assert "alive" not in filtered
|
|
assert "history" not in filtered
|
|
assert "extra" not in filtered
|
|
assert filtered["name"] == "node1"
|
|
assert filtered["server"] == "1.2.3.4"
|
|
|
|
|
|
def test_expand_config_replaces_providers_with_proxies():
|
|
proxies = [
|
|
{"name": "node1", "type": "ss", "server": "1.2.3.4", "port": 443,
|
|
"password": "pwd", "cipher": "aes-256-gcm"},
|
|
]
|
|
result = expand_config(BASE_YAML, {"provider1": proxies})
|
|
cfg = yaml.safe_load(result)
|
|
|
|
assert "proxy-providers" not in cfg
|
|
assert any(p["name"] == "node1" for p in cfg["proxies"])
|
|
|
|
|
|
def test_expand_config_fills_use_in_proxy_groups():
|
|
proxies = [
|
|
{"name": "node1", "type": "ss", "server": "1.2.3.4", "port": 443,
|
|
"password": "pwd", "cipher": "aes-256-gcm"},
|
|
{"name": "node2", "type": "ss", "server": "5.6.7.8", "port": 443,
|
|
"password": "pwd", "cipher": "aes-256-gcm"},
|
|
]
|
|
result = expand_config(BASE_YAML, {"provider1": proxies})
|
|
cfg = yaml.safe_load(result)
|
|
|
|
proxy_group = next(g for g in cfg["proxy-groups"] if g["name"] == "Proxy")
|
|
assert "use" not in proxy_group
|
|
assert "node1" in proxy_group["proxies"]
|
|
assert "node2" in proxy_group["proxies"]
|
|
|
|
|
|
def test_expand_config_preserves_existing_proxies():
|
|
base = """
|
|
proxies:
|
|
- name: static-node
|
|
type: direct
|
|
proxy-providers:
|
|
p1:
|
|
type: http
|
|
url: https://example.com/sub
|
|
interval: 3600
|
|
proxy-groups: []
|
|
rules: []
|
|
"""
|
|
result = expand_config(base, {"p1": [{"name": "dynamic-node", "type": "ss",
|
|
"server": "1.2.3.4", "port": 443}]})
|
|
cfg = yaml.safe_load(result)
|
|
names = [p["name"] for p in cfg["proxies"]]
|
|
assert "static-node" in names
|
|
assert "dynamic-node" in names
|
|
|
|
|
|
def test_expand_config_empty_providers():
|
|
result = expand_config(BASE_YAML, {})
|
|
cfg = yaml.safe_load(result)
|
|
assert "proxy-providers" not in cfg
|
|
|
|
|
|
def test_build_mihomo_config_merges_providers():
|
|
yaml1 = """
|
|
proxy-providers:
|
|
p1:
|
|
type: http
|
|
url: https://a.example.com/sub
|
|
interval: 3600
|
|
health-check:
|
|
enable: true
|
|
url: http://www.gstatic.com/generate_204
|
|
interval: 300
|
|
"""
|
|
yaml2 = """
|
|
proxy-providers:
|
|
p2:
|
|
type: http
|
|
url: https://b.example.com/sub
|
|
interval: 3600
|
|
"""
|
|
result = build_mihomo_config([yaml1, yaml2], "mysecret")
|
|
cfg = yaml.safe_load(result)
|
|
|
|
assert cfg["external-controller"] == "0.0.0.0:9090"
|
|
assert cfg["secret"] == "mysecret"
|
|
assert cfg["mixed-port"] == 7890
|
|
assert cfg["allow-lan"] is False
|
|
assert "p1" in cfg["proxy-providers"]
|
|
assert "p2" in cfg["proxy-providers"]
|
|
assert cfg["proxy-providers"]["p1"]["health-check"]["enable"] is True
|
|
|
|
|
|
def test_build_mihomo_config_deduplicates_by_name():
|
|
yaml1 = """
|
|
proxy-providers:
|
|
p1:
|
|
type: http
|
|
url: https://original.example.com/sub
|
|
interval: 3600
|
|
"""
|
|
yaml2 = """
|
|
proxy-providers:
|
|
p1:
|
|
type: http
|
|
url: https://duplicate.example.com/sub
|
|
interval: 3600
|
|
"""
|
|
result = build_mihomo_config([yaml1, yaml2], "s")
|
|
cfg = yaml.safe_load(result)
|
|
assert cfg["proxy-providers"]["p1"]["url"] == "https://original.example.com/sub"
|
|
|
|
|
|
def test_build_mihomo_config_no_providers():
|
|
result = build_mihomo_config(["proxies: []"], "s")
|
|
cfg = yaml.safe_load(result)
|
|
assert "proxy-providers" not in cfg
|