Fail closed on namespace dunders and metadata types

This commit is contained in:
2026-07-02 18:29:47 +02:00
parent f7143e7bac
commit 3cf4a5eb52
2 changed files with 173 additions and 7 deletions
@@ -1187,6 +1187,52 @@ NODE_CLASS_MAPPINGS = {
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_input_types_with_non_string_input_name_skips_node(self):
source = '''
class NonStringInputNameNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
1: ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"NonStringInputNameNode": NonStringInputNameNode,
}
'''
result = self._extract_source(source, "non-string-input-name-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_input_types_with_non_string_input_type_skips_node(self):
source = '''
class NonStringInputTypeNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": (2,),
},
}
NODE_CLASS_MAPPINGS = {
"NonStringInputTypeNode": NonStringInputTypeNode,
}
'''
result = self._extract_source(source, "non-string-input-type-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_dynamic_return_types_reassignment_skips_node(self):
source = '''
def build_outputs():
@@ -1876,6 +1922,53 @@ NODE_CLASS_MAPPINGS = {
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_non_string_return_type_entry_skips_node(self):
source = '''
class NonStringReturnTypeNode:
RETURN_TYPES = (123,)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"NonStringReturnTypeNode": NonStringReturnTypeNode,
}
'''
result = self._extract_source(source, "non-string-return-type-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_non_string_return_name_entry_skips_node(self):
source = '''
class NonStringReturnNameNode:
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = (456,)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"NonStringReturnNameNode": NonStringReturnNameNode,
}
'''
result = self._extract_source(source, "non-string-return-name-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_missing_return_names_is_allowed(self):
source = '''
class MissingReturnNamesNode:
@@ -3028,6 +3121,55 @@ G.update(NODE_CLASS_MAPPINGS={})
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_globals_dunder_setitem_invalidates_static_node_mapping(self):
source = '''
class GlobalDunderSetitemNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GlobalDunderSetitemNode": GlobalDunderSetitemNode,
}
globals().__setitem__("NODE_CLASS_MAPPINGS", {})
'''
result = self._extract_source(source, "global-dunder-setitem-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_globals_alias_dunder_setitem_invalidates_static_node_mapping(self):
source = '''
class GlobalAliasDunderSetitemNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GlobalAliasDunderSetitemNode": GlobalAliasDunderSetitemNode,
}
ns = globals()
ns.__setitem__("NODE_CLASS_MAPPINGS", {})
'''
result = self._extract_source(source, "global-alias-dunder-setitem-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_arbitrary_call_invalidates_static_node_mapping(self):
source = '''
class ArbitraryCallMappingNode: