mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-31 03:17:23 +08:00
35943e01ab
Amp-Thread-ID: https://ampcode.com/threads/T-019e4ca5-b71a-7168-8f56-58b2325f34c3 Co-authored-by: Amp <amp@ampcode.com>
70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
"""Unit tests for app.assets.helpers."""
|
|
|
|
from app.assets.helpers import expand_bucket_prefixes
|
|
|
|
|
|
class TestExpandBucketPrefixes:
|
|
def test_flat_category_unchanged(self):
|
|
# `checkpoints` is already a standalone token, no expansion needed.
|
|
assert expand_bucket_prefixes(["models", "checkpoints"]) == [
|
|
"models",
|
|
"checkpoints",
|
|
]
|
|
|
|
def test_nested_category_inserts_bucket(self):
|
|
# Path-derived shape for `models/checkpoints/flux/foo.safetensors` —
|
|
# the standalone bucket has to be present so the FE set-membership
|
|
# filter (`include_tags=models,checkpoints`) matches the asset.
|
|
assert expand_bucket_prefixes(["models", "checkpoints/flux"]) == [
|
|
"models",
|
|
"checkpoints/flux",
|
|
"checkpoints",
|
|
]
|
|
|
|
def test_deeply_nested_only_first_segment_expands(self):
|
|
# Only the FIRST slash segment ever gets emitted as a standalone —
|
|
# intermediate path segments don't have routing significance.
|
|
assert expand_bucket_prefixes(
|
|
["models", "diffusers/kolors/text_encoder"]
|
|
) == ["models", "diffusers/kolors/text_encoder", "diffusers"]
|
|
|
|
def test_unknown_prefix_does_not_expand(self):
|
|
# Free-form user labels with slashes whose first segment is not a
|
|
# registered bucket pass through opaquely.
|
|
assert expand_bucket_prefixes(["models", "my-org/team-a"]) == [
|
|
"models",
|
|
"my-org/team-a",
|
|
]
|
|
|
|
def test_idempotent(self):
|
|
# Re-applying the helper is a no-op once the bucket is in the set.
|
|
expanded = expand_bucket_prefixes(["models", "checkpoints/flux"])
|
|
assert expand_bucket_prefixes(expanded) == expanded
|
|
|
|
def test_does_not_duplicate_existing_bucket(self):
|
|
# If the caller already supplied the standalone bucket, don't add a
|
|
# second copy.
|
|
assert expand_bucket_prefixes(
|
|
["models", "checkpoints/flux", "checkpoints"]
|
|
) == ["models", "checkpoints/flux", "checkpoints"]
|
|
|
|
def test_preserves_caller_order(self):
|
|
# User tags after path tags must stay after; the inserted bucket
|
|
# token slots in immediately after its slash-joined parent so the
|
|
# microsecond stagger lands it at path-tier before user-tier.
|
|
assert expand_bucket_prefixes(
|
|
["models", "loras/style", "favorite", "v2"]
|
|
) == ["models", "loras/style", "loras", "favorite", "v2"]
|
|
|
|
def test_empty_input(self):
|
|
assert expand_bucket_prefixes([]) == []
|
|
|
|
def test_input_root_with_subpath_no_expansion(self):
|
|
# `portraits` isn't a registered model category, so the input
|
|
# subpath stays opaque (FE filter doesn't have a checkpoint-loader
|
|
# analogue for input subfolders).
|
|
assert expand_bucket_prefixes(["input", "portraits/2026"]) == [
|
|
"input",
|
|
"portraits/2026",
|
|
]
|