File size: 950 Bytes
a38ca42 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """Smoke test for validate_beta_key logic (local mode, no HF required)."""
import os
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
os.environ["VTX_KEYS_SOURCE"] = "local"
os.environ["VTX_KEYS_PATH"] = str(Path(__file__).resolve().parents[1] / "keys.json")
from app import validate_beta_key
active = validate_beta_key("VTX-BETA-7K4P-X92M", track_activation=False)
revoked = validate_beta_key("VTX-BETA-R8JL-Q4WT", track_activation=False)
unknown = validate_beta_key("VTX-BETA-NOT-REAL", track_activation=False)
empty = validate_beta_key("", track_activation=False)
assert active == {"valid": True, "active": True, "status": "ACTIVE"}, active
assert revoked == {"valid": True, "active": False, "status": "REVOKED"}, revoked
assert unknown == {"valid": False, "active": False, "status": "INVALID"}, unknown
assert empty == {"valid": False, "active": False, "status": "INVALID"}, empty
print("OK") |