Convert lit-based syntax classification tests to XCTests

Author: StevenWong12

Sources/SwiftIDEUtils/SyntaxClassification.swift

@@ -39,8 +39,6 @@ public enum SyntaxClassification {
   case lineComment
   /// The token should not receive syntax coloring.
   case none
-  /// An image, color, etc. literal.
-  case objectLiteral
   /// An identifier referring to an operator.
   case operatorIdentifier
   /// A `#` token like `#warning`.
@@ -66,7 +64,7 @@ extension SyntaxClassification {
   internal static func classify(_ keyPath: AnyKeyPath) -> (SyntaxClassification, Bool)? {
     switch keyPath {
     case \AttributeSyntax.attributeName:
-      return (.attribute, false)
+      return (.attribute, true)
     case \PlatformVersionItemSyntax.availabilityVersionRestriction:
       return (.keyword, false)
     case \AvailabilityVersionRestrictionSyntax.platform:

Sources/SwiftIDEUtils/SyntaxClassifier.swift

@@ -196,12 +196,49 @@ private struct ClassificationVisitor {
 
     for case (let index, let child?) in children.enumerated() {
 
-      let classification: (SyntaxClassification, Bool)?
+      let classification: (classification: SyntaxClassification, force: Bool)?
       if case .layout(let layout) = descriptor.node.kind.syntaxNodeType.structure {
         classification = SyntaxClassification.classify(layout[index])
       } else {
         classification = nil
       }
+
+      if let classification, classification.force {
+        // Leading trivia.
+        if let triviaPieces = child.leadingTriviaPieces {
+          for triviaPiece in triviaPieces {
+            let range = triviaPiece.classify(offset: byteOffset)
+            report(range: range)
+            byteOffset += triviaPiece.byteLength
+          }
+        }
+        child.leadingTriviaPieces?.compactMap { $0 }.forEach {
+          let range = $0.classify(offset: byteOffset)
+          report(range: range)
+        }
+        // Layout node text.
+        let layoutNodeTextLength = child.byteLength - child.leadingTriviaByteLength - child.trailingTriviaByteLength
+        let range = SyntaxClassifiedRange(
+          kind: classification.classification,
+          range: ByteSourceRange(
+            offset: byteOffset,
+            length: layoutNodeTextLength
+          )
+        )
+        report(range: range)
+        byteOffset += layoutNodeTextLength
+
+        // Trailing trivia.
+        if let triviaPieces = child.trailingTriviaPieces {
+          for triviaPiece in triviaPieces {
+            let range = triviaPiece.classify(offset: byteOffset)
+            report(range: range)
+            byteOffset += triviaPiece.byteLength
+          }
+        }
+        continue
+      }
+
       let result = visit(
         .init(
           node: child,

Sources/SwiftSyntax/Raw/RawSyntax.swift

@@ -505,6 +505,16 @@ extension RawSyntax {
     lastToken(viewMode: .sourceAccurate)?.trailingTriviaByteLength ?? 0
   }
 
+  @_spi(RawSyntax)
+  public var leadingTriviaPieces: [RawTriviaPiece]? {
+    firstToken(viewMode: .sourceAccurate)?.leadingRawTriviaPieces
+  }
+
+  @_spi(RawSyntax)
+  public var trailingTriviaPieces: [RawTriviaPiece]? {
+    lastToken(viewMode: .sourceAccurate)?.trailingRawTriviaPieces
+  }
+
   /// The length of this node’s content, without the first leading and the last
   /// trailing trivia. Intermediate trivia inside a layout node is included in
   /// this.

Sources/lit-test-helper/ClassifiedSyntaxTreePrinter.swift

@@ -30,7 +30,6 @@ extension SyntaxClassification {
     case .poundDirective: return "#kw"
     case .buildConfigId: return "#id"
     case .attribute: return "attr-builtin"
-    case .objectLiteral: return "object-literal"
     case .editorPlaceholder: return "placeholder"
     case .lineComment: return "comment-line"
     case .blockComment: return "comment-block"

Tests/SwiftIDEUtilsTest/Assertions.swift

@@ -71,7 +71,7 @@ func assertClassification(
       line: spec.line
     )
 
-    lastRangeUpperBound = source.index(source.startIndex, offsetBy: range.endOffset)
+    lastRangeUpperBound = source.utf8.index(source.utf8.startIndex, offsetBy: range.endOffset)
   }
 }