IxMxAMAR
15e4b0da4e
Critical:
- rename nodes/ -> mp_nodes/, shared/ -> mp_shared/ to avoid shadowing
ComfyUI core's nodes.py via sys.path injection (root cause of v0.1.1
'nodes is not a package' startup failure)
- http_post now streams response and enforces 50MB cap mid-stream (DoS)
- AES key derivation: HKDF -> PBKDF2-HMAC-SHA256 600k iters + random salt
(work factor against weak-key brute force; ciphertext format changed)
- fs_mkdir/copy/move/delete confined to ComfyUI input/output/temp + ~ by
default; UTILITY_MEGAPACK_ALLOW_ARBITRARY_PATHS=1 to override
UX / correctness:
- combo settings now declare options array (themes dropdown was empty)
- LATENT default 1x4x8x8 zeros -> {} (avoids Flux/SD3 shape crashes)
- @op decorator rejects reserved widget names mode/theme at register time
- _pad_outputs docstring updated to match relaxed opaque-type behavior
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""Tests for the Encoding category of ProgrammingNode."""
|
|
|
|
import pytest
|
|
|
|
from mp_nodes.programming import ProgrammingNode
|
|
|
|
|
|
def run(mode, **kwargs):
|
|
return ProgrammingNode().process(mode=mode, theme="(use pack default)", **kwargs)
|
|
|
|
|
|
class TestBase64:
|
|
def test_encode_round_trip(self):
|
|
encoded = run("base64_encode", text="hello world")[0]
|
|
assert encoded == "aGVsbG8gd29ybGQ="
|
|
|
|
def test_decode_round_trip(self):
|
|
decoded = run("base64_decode", text="aGVsbG8gd29ybGQ=")[0]
|
|
assert decoded == "hello world"
|
|
|
|
def test_decode_invalid_raises(self):
|
|
with pytest.raises(RuntimeError, match=r"\[ProgrammingNode/base64_decode\]"):
|
|
run("base64_decode", text="not-base64!@#$")
|
|
|
|
def test_encode_unicode(self):
|
|
out = run("base64_encode", text="café")[0]
|
|
assert run("base64_decode", text=out)[0] == "café"
|
|
|
|
|
|
class TestUrlEncode:
|
|
def test_url_encode_quotes_reserved(self):
|
|
out = run("url_encode", text="hello world & friends")[0]
|
|
assert out == "hello%20world%20%26%20friends"
|
|
|
|
def test_url_decode_handles_plus_as_space(self):
|
|
out = run("url_decode", text="hello+world%26friends")[0]
|
|
assert out == "hello world&friends"
|
|
|
|
def test_url_round_trip(self):
|
|
original = "a b/c?d=1&e=2"
|
|
encoded = run("url_encode", text=original)[0]
|
|
decoded = run("url_decode", text=encoded)[0]
|
|
assert decoded == original
|
|
|
|
|
|
class TestSlug:
|
|
def test_basic_slug(self):
|
|
assert run("slug", text="Hello, World!")[0] == "hello-world"
|
|
|
|
def test_strips_leading_trailing_hyphens(self):
|
|
assert run("slug", text="!!! Hello !!!")[0] == "hello"
|
|
|
|
def test_max_length_truncates(self):
|
|
assert run("slug", text="a" * 200, max_length=10)[0] == "a" * 10
|
|
|
|
def test_max_length_does_not_leave_trailing_hyphen(self):
|
|
# 'foo-bar-baz' truncated to 4 would be 'foo-' but we strip trailing hyphens
|
|
out = run("slug", text="foo bar baz", max_length=4)[0]
|
|
assert not out.endswith("-")
|
|
|
|
def test_unicode_collapses_to_hyphens(self):
|
|
# Non-ASCII letters are not in [a-z0-9], so they collapse to hyphens
|
|
out = run("slug", text="café-bistro")[0]
|
|
# 'café-bistro' -> 'caf-bistro' (the é + - both collapse)
|
|
assert out == "caf-bistro"
|