Fail closed on class-body signature references

2026-07-02 20:02:55 +02:00
parent 447a5c72dd
commit 25b3f69d0d
2 changed files with 181 additions and 2 deletions
@@ -1960,6 +1960,80 @@ NODE_CLASS_MAPPINGS = {
        self.assertEqual({}, result["nodes"])
        self.assertEqual("no_static_nodes", result["pack"]["status"])

+    def test_bare_return_types_reference_skips_node(self):
+        source = '''
+class BareReturnTypesReferenceNode:
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_TYPES
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+            },
+        }
+
+
+NODE_CLASS_MAPPINGS = {
+    "BareReturnTypesReferenceNode": BareReturnTypesReferenceNode,
+}
+'''
+        result = self._extract_source(source, "bare-return-types-reference-pack")
+
+        self.assertEqual({}, result["nodes"])
+        self.assertEqual("no_static_nodes", result["pack"]["status"])
+
+    def test_bare_return_types_alias_reference_skips_node(self):
+        source = '''
+class BareReturnTypesAliasReferenceNode:
+    RETURN_TYPES = ("IMAGE",)
+    ALIAS = RETURN_TYPES
+    ALIAS
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+            },
+        }
+
+
+NODE_CLASS_MAPPINGS = {
+    "BareReturnTypesAliasReferenceNode": BareReturnTypesAliasReferenceNode,
+}
+'''
+        result = self._extract_source(source, "bare-return-types-alias-reference-pack")
+
+        self.assertEqual({}, result["nodes"])
+        self.assertEqual("no_static_nodes", result["pack"]["status"])
+
+    def test_bare_return_names_reference_skips_node(self):
+        source = '''
+class BareReturnNamesReferenceNode:
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("image",)
+    RETURN_NAMES
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+            },
+        }
+
+
+NODE_CLASS_MAPPINGS = {
+    "BareReturnNamesReferenceNode": BareReturnNamesReferenceNode,
+}
+'''
+        result = self._extract_source(source, "bare-return-names-reference-pack")
+
+        self.assertEqual({}, result["nodes"])
+        self.assertEqual("no_static_nodes", result["pack"]["status"])
+
    def test_return_types_chained_alias_mutation_skips_node(self):
        source = '''
 class ChainedAliasMutatedReturnTypesNode:
@@ -4595,6 +4669,56 @@ NODE_CLASS_MAPPINGS = {
        self.assertEqual({}, result["nodes"])
        self.assertEqual("no_static_nodes", result["pack"]["status"])

+    def test_bare_input_types_reference_skips_node(self):
+        source = '''
+class BareInputTypesReferenceNode:
+    RETURN_TYPES = ("IMAGE",)
+    INPUT_TYPES
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+            },
+        }
+
+
+NODE_CLASS_MAPPINGS = {
+    "BareInputTypesReferenceNode": BareInputTypesReferenceNode,
+}
+'''
+        result = self._extract_source(source, "bare-input-types-reference-pack")
+
+        self.assertEqual({}, result["nodes"])
+        self.assertEqual("no_static_nodes", result["pack"]["status"])
+
+    def test_bare_input_types_alias_reference_skips_node(self):
+        source = '''
+class BareInputTypesAliasReferenceNode:
+    RETURN_TYPES = ("IMAGE",)
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "image": ("IMAGE",),
+            },
+        }
+
+    ALIAS = INPUT_TYPES
+    ALIAS
+
+
+NODE_CLASS_MAPPINGS = {
+    "BareInputTypesAliasReferenceNode": BareInputTypesAliasReferenceNode,
+}
+'''
+        result = self._extract_source(source, "bare-input-types-alias-reference-pack")
+
+        self.assertEqual({}, result["nodes"])
+        self.assertEqual("no_static_nodes", result["pack"]["status"])
+
    def test_input_types_default_observed_by_arbitrary_call_skips_node(self):
        source = '''
 class DefaultObservedInputTypesNode: