1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2021 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 CodeLensTests: XCTestCase {
func testNoLenses() async throws {
var codeLensCapabilities = TextDocumentClientCapabilities.CodeLens()
codeLensCapabilities.supportedCommands = [
SupportedCodeLensCommand.run: "swift.run",
SupportedCodeLensCommand.debug: "swift.debug",
]
let capabilities = ClientCapabilities(textDocument: TextDocumentClientCapabilities(codeLens: codeLensCapabilities))
let project = try await SwiftPMTestProject(
files: [
"Test.swift": """
struct MyApp {
public static func main() {}
}
"""
],
capabilities: capabilities
)
let (uri, _) = try project.openDocument("Test.swift")
let response = try await project.testClient.send(
CodeLensRequest(textDocument: TextDocumentIdentifier(uri))
)
XCTAssertEqual(response, [])
}
func testNoClientCodeLenses() async throws {
let project = try await SwiftPMTestProject(
files: [
"Test.swift": """
@main
struct MyApp {
public static func main() {}
}
"""
]
)
let (uri, _) = try project.openDocument("Test.swift")
let response = try await project.testClient.send(
CodeLensRequest(textDocument: TextDocumentIdentifier(uri))
)
XCTAssertEqual(response, [])
}
func testSuccessfulCodeLensRequest() async throws {
var codeLensCapabilities = TextDocumentClientCapabilities.CodeLens()
codeLensCapabilities.supportedCommands = [
SupportedCodeLensCommand.run: "swift.run",
SupportedCodeLensCommand.debug: "swift.debug",
]
let capabilities = ClientCapabilities(textDocument: TextDocumentClientCapabilities(codeLens: codeLensCapabilities))
let project = try await SwiftPMTestProject(
files: [
"Test.swift": """
1️⃣@main2️⃣
struct MyApp {
public static func main() {}
}
"""
],
capabilities: capabilities
)
let (uri, positions) = try project.openDocument("Test.swift")
let response = try await project.testClient.send(
CodeLensRequest(textDocument: TextDocumentIdentifier(uri))
)
XCTAssertEqual(
response,
[
CodeLens(
range: positions["1️⃣"]..<positions["2️⃣"],
command: Command(title: "Run", command: "swift.run", arguments: nil)
),
CodeLens(
range: positions["1️⃣"]..<positions["2️⃣"],
command: Command(title: "Debug", command: "swift.debug", arguments: nil)
),
]
)
}
}
|