Test CustomAttribute initializer in the context of variables

Author: fwcd

Tests/SwiftSyntaxBuilderTest/VariableTests.swift

@@ -107,29 +107,56 @@ final class VariableTests: XCTestCase {
   }
 
   func testAttributedVariables() {
-    let attributedVar = VariableDecl(
-      attributes: CustomAttribute(attributeName: "Test", argumentList: nil),
-      .var,
-      name: "x",
-      type: "Int"
-    )
-
-    XCTAssertEqual(attributedVar.buildSyntax(format: Format()).description, """
-      @Test var x: Int
-      """)
+    let testCases: [UInt: (VariableDecl, String)] = [
+      #line: (
+        VariableDecl(
+          attributes: CustomAttribute("Test"),
+          .var,
+          name: "x",
+          type: "Int"
+        ),
+        """
+        @Test var x: Int
+        """
+      ),
+      #line: (
+        VariableDecl(
+          attributes: CustomAttribute("Test"),
+          name: "y",
+          type: "String"
+        ) {
+          StringLiteralExpr("Hello world!")
+        },
+        """
+        @Test var y: String {
+            "Hello world!"
+        }
+        """
+      ),
+      #line: (
+        VariableDecl(
+          attributes: CustomAttribute("WithArgs") {
+            TupleExprElement(expression: "value1")
+            TupleExprElement(label: "label", expression: "value2")
+          },
+          name: "z",
+          type: "Float"
+        ) {
+          FloatLiteralExpr(0.0)
+        },
+        """
+        @WithArgs(value1, label: value2) var z: Float {
+            0.0
+        }
+        """
+      ),
+    ]
 
-    let attributedProperty = VariableDecl(
-      attributes: CustomAttribute(attributeName: "Test", argumentList: nil),
-      name: "y",
-      type: "String"
-    ) {
-      StringLiteralExpr("Hello world!")
-    }
+    for (line, testCase) in testCases {
+      let (builder, expected) = testCase
+      let syntax = builder.buildSyntax(format: Format())
 
-    XCTAssertEqual(attributedProperty.buildSyntax(format: Format()).description, """
-      @Test var y: String {
-          "Hello world!"
-      }
-      """)
+      XCTAssertEqual(syntax.description, expected, line: line)
+    }
   }
 }