Handle complete function names with raw pieces

Author: grynspan

Sources/TestingMacros/Support/Additions/FunctionDeclSyntaxAdditions.swift

@@ -9,6 +9,7 @@
 //
 
 import SwiftSyntax
+import SwiftSyntaxBuilder
 import SwiftSyntaxMacros
 
 extension FunctionDeclSyntax {
@@ -35,16 +36,24 @@ extension FunctionDeclSyntax {
 
   /// The name of this function including parentheses, parameter labels, and
   /// colons.
-  var completeName: String {
-    var result = [name.textWithoutBackticks, "(",]
-
-    for parameter in signature.parameterClause.parameters {
-      result.append(parameter.firstName.textWithoutBackticks)
-      result.append(":")
+  var completeName: DeclReferenceExprSyntax {
+    func possiblyRaw(_ token: TokenSyntax) -> TokenSyntax {
+      if let rawIdentifier = token.rawIdentifier {
+        return .identifier("`\(rawIdentifier)`")
+      }
+      return .identifier(token.textWithoutBackticks)
     }
-    result.append(")")
 
-    return result.joined()
+    return DeclReferenceExprSyntax(
+      baseName: possiblyRaw(name),
+      argumentNames: DeclNameArgumentsSyntax(
+        arguments: DeclNameArgumentListSyntax {
+          for parameter in signature.parameterClause.parameters {
+            DeclNameArgumentSyntax(name: possiblyRaw(parameter.firstName))
+          }
+        }
+      )
+    )
   }
 
   /// An array of tuples representing this function's parameters.

Sources/TestingMacros/TestDeclarationMacro.swift

@@ -407,7 +407,7 @@ public struct TestDeclarationMacro: PeerMacro, Sendable {
     var testsBody: CodeBlockItemListSyntax = """
     return [
       .__function(
-        named: \(literal: functionDecl.completeName),
+        named: \(literal: functionDecl.completeName.trimmedDescription),
         in: \(typeNameExpr),
         xcTestCompatibleSelector: \(selectorExpr ?? "nil"),
         \(raw: attributeInfo.functionArgumentList(in: context)),
@@ -433,7 +433,7 @@ public struct TestDeclarationMacro: PeerMacro, Sendable {
         private \(_staticKeyword(for: typeName)) nonisolated func \(unavailableTestName)() async -> [Testing.Test] {
           [
             .__function(
-              named: \(literal: functionDecl.completeName),
+              named: \(literal: functionDecl.completeName.trimmedDescription),
               in: \(typeNameExpr),
               xcTestCompatibleSelector: \(selectorExpr ?? "nil"),
               \(raw: attributeInfo.functionArgumentList(in: context)),

Tests/TestingMacrosTests/TestDeclarationMacroTests.swift

@@ -271,6 +271,15 @@ struct TestDeclarationMacroTests {
     #expect(TokenSyntax.identifier("`class struct`").rawIdentifier != nil)
   }
 
+  @Test("Raw function name components")
+  func rawFunctionNameComponents() throws {
+    let decl = """
+    func `__raw__$hello`(`__raw__$world`: T, etc: U, `blah`: V) {}
+    """ as DeclSyntax
+    let functionDecl = try #require(decl.as(FunctionDeclSyntax.self))
+    #expect(functionDecl.completeName.trimmedDescription == "`hello`(`world`:etc:blah:)")
+  }
+
   @Test("Warning diagnostics emitted on API misuse",
     arguments: [
       // return types