Add a request to list all the tests within the current workspace

Fixes swiftlang#611
rdar://98710526

Author: ahoppen

Sources/LanguageServerProtocol/Messages.swift

@@ -82,6 +82,7 @@ public let builtinRequests: [_RequestType.Type] = [
   WorkspaceSemanticTokensRefreshRequest.self,
   WorkspaceSymbolResolveRequest.self,
   WorkspaceSymbolsRequest.self,
+  WorkspaceTestsRequest.self,
 ]
 
 /// The set of known notifications.

Sources/LanguageServerProtocol/Requests/WorkspaceTestsRequest.swift

@@ -0,0 +1,22 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+/// A request that returns symbols for all the test classes and test methods within the current workspace.
+///
+/// **(LSP Extension)**
+public struct WorkspaceTestsRequest: RequestType, Hashable {
+
+  public static let method: String = "workspace/tests"
+  public typealias Response = [WorkspaceSymbolItem]?
+
+  public init() {}
+}

Sources/SKTestSupport/SwiftPMTestWorkspace.swift

@@ -44,13 +44,15 @@ public class SwiftPMTestWorkspace: MultiFileTestWorkspace {
     var filesByPath: [RelativeFileLocation: String] = [:]
     for (fileLocation, contents) in files {
       let directories =
-        if fileLocation.directories.isEmpty {
+        switch fileLocation.directories.first {
+        case "Sources", "Tests":
+          fileLocation.directories
+        case nil:
           ["Sources", "MyLibrary"]
-        } else if fileLocation.directories.first != "Sources" {
+        default:
           ["Sources"] + fileLocation.directories
-        } else {
-          fileLocation.directories
         }
+
       filesByPath[RelativeFileLocation(directories: directories, fileLocation.fileName)] = contents
     }
     filesByPath["Package.swift"] = manifest
@@ -75,6 +77,7 @@ public class SwiftPMTestWorkspace: MultiFileTestWorkspace {
       swift.path,
       "build",
       "--package-path", path.path,
+      "--build-tests",
       "-Xswiftc", "-index-ignore-system-modules",
       "-Xcc", "-index-ignore-system-symbols",
     ]

Sources/SourceKitLSP/SourceKitServer.swift

@@ -335,6 +335,8 @@ fileprivate enum TaskMetadata: DependencyTracker {
       self = .freestanding
     case is WorkspaceSymbolsRequest:
       self = .freestanding
+    case is WorkspaceTestsRequest:
+      self = .freestanding
     default:
       logger.error(
         """
@@ -836,6 +838,8 @@ extension SourceKitServer: MessageHandler {
       await request.reply { try await shutdown(request.params) }
     case let request as RequestAndReply<WorkspaceSymbolsRequest>:
       await request.reply { try await workspaceSymbols(request.params) }
+    case let request as RequestAndReply<WorkspaceTestsRequest>:
+      await request.reply { try await workspaceTests(request.params) }
     case let request as RequestAndReply<PollIndexRequest>:
       await request.reply { try await pollIndex(request.params) }
     case let request as RequestAndReply<BarrierRequest>:
@@ -1490,33 +1494,30 @@ extension SourceKitServer {
   /// Handle a workspace/symbol request, returning the SymbolInformation.
   /// - returns: An array with SymbolInformation for each matching symbol in the workspace.
   func workspaceSymbols(_ req: WorkspaceSymbolsRequest) async throws -> [WorkspaceSymbolItem]? {
-    let symbols = findWorkspaceSymbols(
-      matching: req.query
-    ).map({ symbolOccurrence -> WorkspaceSymbolItem in
-      let symbolPosition = Position(
-        line: symbolOccurrence.location.line - 1,  // 1-based -> 0-based
-        // FIXME: we need to convert the utf8/utf16 column, which may require reading the file!
-        utf16index: symbolOccurrence.location.utf8Column - 1
-      )
-
-      let symbolLocation = Location(
-        uri: DocumentURI(URL(fileURLWithPath: symbolOccurrence.location.path)),
-        range: Range(symbolPosition)
-      )
-
-      return .symbolInformation(
-        SymbolInformation(
-          name: symbolOccurrence.symbol.name,
-          kind: symbolOccurrence.symbol.kind.asLspSymbolKind(),
-          deprecated: nil,
-          location: symbolLocation,
-          containerName: symbolOccurrence.getContainerName()
-        )
-      )
-    })
+    let symbols = findWorkspaceSymbols(matching: req.query).map(WorkspaceSymbolItem.init)
     return symbols
   }
 
+  func workspaceTests(_ req: WorkspaceTestsRequest) async throws -> [WorkspaceSymbolItem]? {
+    let testSymbols = workspaces.flatMap { (workspace) -> [SymbolOccurrence] in
+      guard let index = workspace.index else {
+        return []
+      }
+      let xctestSubclasses = index.occurrences(ofUSR: "c:objc(cs)XCTest", roles: [.baseOf])
+        .flatMap(\.relations)
+        .filter { $0.roles.contains(.baseOf) }
+        .flatMap { index.occurrences(ofUSR: $0.symbol.usr, roles: [.definition]) }
+      let testMethods = xctestSubclasses.flatMap { xctestSubclass in
+        index.occurrences(relatedToUSR: xctestSubclass.symbol.usr, roles: [.childOf])
+      }.filter { testMethodSymbol in
+        return testMethodSymbol.symbol.kind == .instanceMethod && testMethodSymbol.symbol.name.starts(with: "test")
+          && testMethodSymbol.roles.contains(.definition)
+      }
+      return xctestSubclasses + testMethods
+    }
+    return testSymbols.map(WorkspaceSymbolItem.init)
+  }
+
   /// Forwards a SymbolInfoRequest to the appropriate toolchain service for this document.
   func symbolInfo(
     _ req: SymbolInfoRequest,
@@ -2213,3 +2214,27 @@ fileprivate struct DocumentNotificationRequestQueue {
     queue = []
   }
 }
+fileprivate extension WorkspaceSymbolItem {
+  init(_ symbolOccurrence: SymbolOccurrence) {
+    let symbolPosition = Position(
+      line: symbolOccurrence.location.line - 1,  // 1-based -> 0-based
+      // FIXME: we need to convert the utf8/utf16 column, which may require reading the file!
+      utf16index: symbolOccurrence.location.utf8Column - 1
+    )
+
+    let symbolLocation = Location(
+      uri: DocumentURI(URL(fileURLWithPath: symbolOccurrence.location.path)),
+      range: Range(symbolPosition)
+    )
+
+    self = .symbolInformation(
+      SymbolInformation(
+        name: symbolOccurrence.symbol.name,
+        kind: symbolOccurrence.symbol.kind.asLspSymbolKind(),
+        deprecated: nil,
+        location: symbolLocation,
+        containerName: symbolOccurrence.getContainerName()
+      )
+    )
+  }
+}

Tests/SourceKitLSPTests/WorkspaceTestsTests.swift

@@ -0,0 +1,72 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+import LanguageServerProtocol
+import SKTestSupport
+import XCTest
+
+final class WorkspaceTestsTests: XCTestCase {
+  func testWorkspaceTests() async throws {
+    let ws = try await SwiftPMTestWorkspace(
+      files: [
+        "Tests/MyLibraryTests/MyTests.swift": """
+        import XCTest
+
+        class 1️⃣MyTests: XCTest {
+          func 2️⃣testMyLibrary() {}
+          func unrelatedFunc() {}
+          var testVariable: Int = 0
+        }
+        """
+      ],
+      manifest: """
+        // swift-tools-version: 5.7
+
+        import PackageDescription
+
+        let package = Package(
+          name: "MyLibrary",
+          targets: [.testTarget(name: "MyLibraryTests")]
+        )
+        """,
+      build: true
+    )
+
+    let workspaceTests = try await ws.testClient.send(WorkspaceTestsRequest())
+    XCTAssertEqual(
+      workspaceTests,
+      [
+        WorkspaceSymbolItem.symbolInformation(
+          SymbolInformation(
+            name: "MyTests",
+            kind: .class,
+            location: Location(
+              uri: try ws.uri(for: "MyTests.swift"),
+              range: Range(try ws.position(of: "1️⃣", in: "MyTests.swift"))
+            )
+          )
+        ),
+        WorkspaceSymbolItem.symbolInformation(
+          SymbolInformation(
+            name: "testMyLibrary()",
+            kind: .method,
+            location: Location(
+              uri: try ws.uri(for: "MyTests.swift"),
+              range: Range(try ws.position(of: "2️⃣", in: "MyTests.swift"))
+            ),
+            containerName: "MyTests"
+          )
+        ),
+      ]
+    )
+  }
+}