add unit test for mutable macro

Author: NullIsOne

Components/Macro/Sources/Macros/MutableMacro.swift

@@ -23,8 +23,9 @@ public struct MutableMacro: MemberMacro {
 
               let functions = try variables.compactMap { variableDecl -> FunctionDeclSyntax? in
                   guard let variableBinding = variableDecl.bindings.first,
-                        let variableType = variableBinding.typeAnnotation?.type else {
-                      throw MacroError.typeAnnotationRequiredFor(variableName: variableDecl.bindings.first?.pattern.description ?? "unknown")
+                        let variableType = variableBinding.typeAnnotation?.type.trimmed else {
+                      let variableName = variableDecl.bindings.first?.pattern.trimmedDescription
+                      throw MacroError.typeAnnotationRequiredFor(variableName: variableName ?? "unknown")
                   }
 
                   let variableName = TokenSyntax(stringLiteral: variableBinding.pattern.description)
@@ -34,7 +35,9 @@ public struct MutableMacro: MemberMacro {
                   let modifiers = ModifierListSyntax(arrayLiteral: .init(name: .keyword(.mutating)))
                   let bodyItem = CodeBlockItemSyntax.Item.expr(.init(stringLiteral: "self.\(variableName.text)=\(variableName.text)"))
                   let body = CodeBlockSyntax(statements: .init(arrayLiteral: .init(item: bodyItem)))
-                  return FunctionDeclSyntax(modifiers: modifiers,
+
+                  return FunctionDeclSyntax(leadingTrivia: .newlines(2),
+                                            modifiers: modifiers,
                                             identifier: .identifier("set"),
                                             signature: .init(input: .init(parameterList: parameterList)),
                                             body: body

Components/Macro/Tests/MutableMacroTests.swift

@@ -4,7 +4,7 @@ import XCTest
 import Macros
 
 let testMacros: [String: Macro.Type] = [
-    "mutable": MutableMacro.self
+    "Mutable": MutableMacro.self
 ]
 
 final class MutableMacroTests: XCTestCase {
@@ -18,8 +18,9 @@ final class MutableMacroTests: XCTestCase {
                 var someVar: Bool = false
             }
             """,
-            expandedSource: """
-            @Mutable
+            expandedSource:
+            """
+
             struct SomeStruct {
                 let id: String = ""
                 var someVar: Bool = false
@@ -42,13 +43,19 @@ final class MutableMacroTests: XCTestCase {
                 var someVar: Bool = false
             }
             """,
-            expandedSource: """
-            @Mutable
+            expandedSource:
+            """
+
             class SomeStruct {
                 let id: String = ""
                 var someVar: Bool = false
             }
             """,
+            diagnostics: [
+                .init(message: "onlyApplicableToStruct",
+                      line: 1,
+                      column: 1)
+            ],
             macros: testMacros
         )
     }
@@ -62,13 +69,19 @@ final class MutableMacroTests: XCTestCase {
                 var someVar = false
             }
             """,
-            expandedSource: """
-            @Mutable
+            expandedSource:
+            """
+
             struct SomeStruct {
                 let id: String = ""
                 var someVar = false
             }
             """,
+            diagnostics: [
+                .init(message: "typeAnnotationRequiredFor(variableName: \"someVar\")",
+                      line: 1,
+                      column: 1)
+            ],
             macros: testMacros
         )
     }