Track namespace and getattr aliases

This commit is contained in:
2026-07-02 17:19:07 +02:00
parent 065c9ae7ec
commit 7c4b83ed0e
2 changed files with 124 additions and 12 deletions
@@ -2303,6 +2303,56 @@ globals().get("GlobalsGetReturnTypesNode").RETURN_TYPES.clear()
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_getattr_class_attribute_alias_mutation_after_mapping_skips_node(self):
source = '''
class GetattrAttributeAliasNode:
RETURN_TYPES = ["IMAGE"]
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GetattrAttributeAliasNode": GetattrAttributeAliasNode,
}
RET = getattr(GetattrAttributeAliasNode, "RETURN_TYPES")
RET.clear()
'''
result = self._extract_source(source, "getattr-attribute-alias-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_getattr_namespace_class_attribute_alias_mutation_after_mapping_skips_node(self):
source = '''
class GetattrNamespaceAttributeAliasNode:
RETURN_TYPES = ["IMAGE"]
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GetattrNamespaceAttributeAliasNode": GetattrNamespaceAttributeAliasNode,
}
RET = getattr(globals()["GetattrNamespaceAttributeAliasNode"], "RETURN_TYPES")
RET.clear()
'''
result = self._extract_source(source, "getattr-namespace-attribute-alias-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_module_class_tuple_alias_patch_after_mapping_skips_node(self):
source = '''
class TupleAliasPatchedNode:
@@ -2451,6 +2501,56 @@ globals().update(NODE_CLASS_MAPPINGS={})
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_globals_subscript_alias_mutation_invalidates_static_node_mapping(self):
source = '''
class GlobalAliasMutatedMappingNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GlobalAliasMutatedMappingNode": GlobalAliasMutatedMappingNode,
}
ALIAS = globals()["NODE_CLASS_MAPPINGS"]
ALIAS.clear()
'''
result = self._extract_source(source, "global-alias-mutated-mapping-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_globals_get_alias_mutation_invalidates_static_node_mapping(self):
source = '''
class GlobalGetAliasMutatedMappingNode:
RETURN_TYPES = ("IMAGE",)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
}
NODE_CLASS_MAPPINGS = {
"GlobalGetAliasMutatedMappingNode": GlobalGetAliasMutatedMappingNode,
}
ALIAS = globals().get("NODE_CLASS_MAPPINGS")
ALIAS.clear()
'''
result = self._extract_source(source, "global-get-alias-mutated-mapping-pack")
self.assertEqual({}, result["nodes"])
self.assertEqual("no_static_nodes", result["pack"]["status"])
def test_unpacked_alias_mutation_invalidates_static_node_mapping(self):
source = '''
class UnpackedAliasMutatedMappingNode: