From eb80dc08f3888f500e9cedc3719c3a5198e71771 Mon Sep 17 00:00:00 2001
From: Eric Masseran <rico.masseran@gmail.com>
Date: Mon, 5 May 2025 00:37:38 +0200
Subject: [PATCH] Add decorator tests - sandwich mode

---
 test/test_inference/test_signature.py | 48 ++++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/test/test_inference/test_signature.py b/test/test_inference/test_signature.py
index 64d830d6..47e89d20 100644
--- a/test/test_inference/test_signature.py
+++ b/test/test_inference/test_signature.py
@@ -346,6 +346,20 @@ def test_wraps_signature(Script, code, signature):
             ["y"],
             True,
         ],
+        [
+            dedent(
+                """
+         @dataclass
+         class Y():
+             y: int
+         class Z(Y): # Not included
+             z = 5
+         @dataclass
+         class X(Z):"""
+            ),
+            ["y"],
+            True,
+        ],
         # init=False
         [
             dedent(
@@ -384,6 +398,7 @@ def test_wraps_signature(Script, code, signature):
         "transformed_with_params",
         "subclass_transformed",
         "both_transformed",
+        "intermediate_not_transformed",
         "init_false",
         "init_false_multiple",
         "custom_init",
@@ -469,6 +484,34 @@ dataclass_transform_cases = [
             pass
         @create_model
         class X:'''), [], True],
+    [dedent('''
+        @dataclass_transform
+        def create_model():
+            pass
+        class Y:
+            y: int
+        @create_model
+        class X(Y):'''), [], True],
+    [dedent('''
+        @dataclass_transform
+        def create_model():
+            pass
+        @create_model
+        class Y:
+            y: int
+        @create_model
+        class X(Y):'''), ["y"], True],
+    [dedent('''
+        @dataclass_transform
+        def create_model():
+            pass
+        @create_model
+        class Y:
+            y: int
+        class Z(Y):
+            z: int
+        @create_model
+        class X(Z):'''), ["y"], True],
     # Metaclass based
     [dedent('''
         @dataclass_transform
@@ -547,7 +590,10 @@ ids = [
     "subclass_transformer",
     "base_transformed",
     "base_transformed_with_params",
-    "decorator_transformed",
+    "decorator_transformed_direct",
+    "decorator_transformed_subclass",
+    "decorator_transformed_both",
+    "decorator_transformed_intermediate_not",
     "metaclass_transformed",
     "custom_init",
     "base_transformed_init_false",