Files
sourcekit-lsp/Tests/SourceKitLSPTests/RewriteSourceKitPlaceholdersTests.swift
2025-10-31 14:11:11 -07:00

69 lines
2.5 KiB
Swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2024 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 SKLogging
import SKTestSupport
import SourceKitLSP
@_spi(Testing) import SwiftLanguageService
import XCTest
final class RewriteSourceKitPlaceholdersTests: SourceKitLSPTestCase {
func testClientDoesNotSupportSnippets() {
let input = "foo(bar: <#T##Int##Int#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: false)
XCTAssertEqual(rewritten, "foo(bar: )")
}
func testInputWithoutPlaceholders() {
let input = "foo()"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, "foo()")
}
func testPlaceholderWithType() {
let input = "foo(bar: <#T##bar##Int#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, "foo(bar: ${1:Int})")
}
func testMultiplePlaceholders() {
let input = "foo(bar: <#T##Int##Int#>, baz: <#T##String##String#>, quux: <#T##String##String#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, "foo(bar: ${1:Int}, baz: ${2:String}, quux: ${3:String})")
}
func testClosurePlaceholderReturnType() {
let input = "foo(bar: <#{ <#T##Int##Int#> }#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, #"foo(bar: ${1:{ ${2:Int} \}})"#)
}
func testClosurePlaceholderArgumentType() {
let input = "foo(bar: <#{ <#T##Int##Int#> in <#T##Void##Void#> }#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, #"foo(bar: ${1:{ ${2:Int} in ${3:Void} \}})"#)
}
func testMultipleClosurePlaceholders() {
let input = "foo(<#{ <#T##Int##Int#> }#>, baz: <#{ <#Int#> in <#T##Bool##Bool#> }#>)"
let rewritten = rewriteSourceKitPlaceholders(in: input, clientSupportsSnippets: true)
XCTAssertEqual(rewritten, #"foo(${1:{ ${2:Int} \}}, baz: ${3:{ ${4:Int} in ${5:Bool} \}})"#)
}
}