moxiaonian
d9da4301cd
副本编号 LY-20261007-AC63（水印指纹已嵌入） 含 README、.gitignore；排除 .DS_Store / __pycache__ / .pytest_cache
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""软授权锁：母版放行、副本扣次、用尽/过期/篡改/总开关。"""
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))
|
|
import _license_gate as gate
|
|
|
|
|
|
@pytest.fixture
|
|
def root(tmp_path):
|
|
return tmp_path
|
|
|
|
|
|
def make_state(root, **kw):
|
|
kw.setdefault("copy_id", "LY-20261007-A3F9")
|
|
kw.setdefault("licensee", "张三")
|
|
kw.setdefault("max_uses", 2)
|
|
return gate.write_state(root, **kw)
|
|
|
|
|
|
class TestMasterCopy:
|
|
def test_no_state_file_passes(self, root):
|
|
assert gate.check(root) is None
|
|
|
|
def test_consume_without_state_is_noop(self, root):
|
|
gate.consume(root)
|
|
assert not (root / "data" / ".license.json").exists()
|
|
|
|
|
|
class TestLicensedCopy:
|
|
def test_decrements_per_use_then_denies(self, root):
|
|
make_state(root)
|
|
assert gate.check(root) is None
|
|
gate.consume(root)
|
|
st = json.loads((root / "data" / ".license.json").read_text("utf-8"))
|
|
assert st["remaining_uses"] == 1
|
|
assert gate.check(root) is None
|
|
gate.consume(root)
|
|
assert "次数已用尽" in gate.check(root)
|
|
|
|
def test_expired_denies(self, root):
|
|
make_state(root, expires_at="2020-01-01")
|
|
assert "到期" in gate.check(root)
|
|
|
|
def test_future_expiry_passes(self, root):
|
|
make_state(root, expires_at="2999-01-01")
|
|
assert gate.check(root) is None
|
|
|
|
def test_tampered_state_denies(self, root):
|
|
make_state(root)
|
|
p = root / "data" / ".license.json"
|
|
st = json.loads(p.read_text("utf-8"))
|
|
st["remaining_uses"] = 999 # 手改次数但不重算签名
|
|
p.write_text(json.dumps(st, ensure_ascii=False), "utf-8")
|
|
assert "校验失败" in gate.check(root)
|
|
|
|
def test_corrupted_state_denies(self, root):
|
|
(root / "data").mkdir()
|
|
(root / "data" / ".license.json").write_text("not json", "utf-8")
|
|
assert gate.check(root) is not None
|
|
|
|
|
|
class TestMasterSwitch:
|
|
def test_disabled_gate_passes_exhausted_copy(self, root, monkeypatch):
|
|
make_state(root, max_uses=0)
|
|
monkeypatch.setattr(gate, "_ENABLED", False)
|
|
assert gate.check(root) is None
|
|
gate.consume(root) # 不扣次
|
|
st = json.loads((root / "data" / ".license.json").read_text("utf-8"))
|
|
assert st["remaining_uses"] == 0
|