红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
"""Tests for check_deps.py — focuses on parsing logic that doesn't need a server."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from check_deps import (
|
|
NODE_TO_PACKAGE,
|
|
model_present,
|
|
normalize_for_match,
|
|
suggest_install_command,
|
|
)
|
|
|
|
|
|
class TestNormalizeForMatch:
|
|
def test_basic(self):
|
|
s = normalize_for_match("model.safetensors")
|
|
assert "model.safetensors" in s
|
|
assert "model" in s
|
|
|
|
def test_subfolder(self):
|
|
s = normalize_for_match("subdir/model.pt")
|
|
assert "subdir/model.pt" in s
|
|
assert "model.pt" in s
|
|
assert "model" in s
|
|
|
|
|
|
class TestModelPresent:
|
|
def test_exact_match(self):
|
|
assert model_present("a.safetensors", {"a.safetensors", "b.safetensors"}) is True
|
|
|
|
def test_extension_difference(self):
|
|
# User said "model" but installed is "model.safetensors"
|
|
assert model_present("model", {"model.safetensors"}) is True
|
|
# Reverse direction — also matches
|
|
assert model_present("model.safetensors", {"model"}) is True
|
|
|
|
def test_subfolder_match(self):
|
|
# Installed list has "subdir/model.safetensors", workflow asks "model.safetensors"
|
|
assert model_present("model.safetensors", {"subdir/model.safetensors"}) is True
|
|
|
|
def test_missing(self):
|
|
assert model_present("missing.safetensors", {"a.safetensors", "b.safetensors"}) is False
|
|
|
|
def test_empty_installed(self):
|
|
assert model_present("anything.safetensors", set()) is False
|
|
|
|
|
|
class TestSuggestInstallCommand:
|
|
def test_known_node(self):
|
|
cmd = suggest_install_command("VHS_VideoCombine")
|
|
assert cmd == "comfy node install comfyui-videohelpersuite"
|
|
|
|
def test_unknown_node(self):
|
|
assert suggest_install_command("SomeRandomNodeName123") is None
|
|
|
|
|
|
class TestNodePackageMap:
|
|
def test_no_duplicates(self):
|
|
# Each node should map to exactly one package
|
|
keys = list(NODE_TO_PACKAGE.keys())
|
|
assert len(keys) == len(set(keys))
|
|
|
|
def test_packages_are_safe_for_shell(self):
|
|
# Registry slugs must be alphanumerics + hyphens/underscores only
|
|
# (passed straight to `comfy node install <pkg>`).
|
|
import re
|
|
safe = re.compile(r"^[A-Za-z0-9][A-Za-z0-9._\-]*$")
|
|
for pkg in NODE_TO_PACKAGE.values():
|
|
assert safe.match(pkg), f"Unsafe package slug: {pkg!r}"
|