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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
@_spi(Testing) import SourceKitLSP
import Swift
import SwiftBasicFormat
import SwiftParser
import SwiftSyntax
import SwiftSyntaxBuilder
import XCTest
fileprivate func assertFormatted<T: SyntaxProtocol>(
tree: T,
expected: String,
using format: ClosureCompletionFormat = ClosureCompletionFormat(indentationWidth: .spaces(4)),
file: StaticString = #filePath,
line: UInt = #line
) {
XCTAssertEqual(tree.formatted(using: format).description, expected, file: file, line: line)
}
fileprivate func assertFormatted(
source: String,
expected: String,
using format: ClosureCompletionFormat = ClosureCompletionFormat(indentationWidth: .spaces(4)),
file: StaticString = #filePath,
line: UInt = #line
) {
assertFormatted(
tree: Parser.parse(source: source),
expected: expected,
using: format,
file: file,
line: line
)
}
final class ClosureCompletionFormatTests: XCTestCase {
func testSingleStatementClosureArg() {
assertFormatted(
source: """
foo(bar: { baz in baz.quux })
""",
expected: """
foo(bar: { baz in baz.quux })
"""
)
}
func testSingleStatementClosureArgAlreadyMultiLine() {
assertFormatted(
source: """
foo(
bar: { baz in
baz.quux
}
)
""",
expected: """
foo(
bar: { baz in
baz.quux
}
)
"""
)
}
func testMultiStatmentClosureArg() {
assertFormatted(
source: """
foo(
bar: { baz in print(baz); return baz.quux }
)
""",
expected: """
foo(
bar: { baz in
print(baz);
return baz.quux
}
)
"""
)
}
func testMultiStatementClosureArgAlreadyMultiLine() {
assertFormatted(
source: """
foo(
bar: { baz in
print(baz)
return baz.quux
}
)
""",
expected: """
foo(
bar: { baz in
print(baz)
return baz.quux
}
)
"""
)
}
func testFormatClosureWithInitialIndentation() throws {
assertFormatted(
tree: ClosureExprSyntax(
statements: CodeBlockItemListSyntax([
CodeBlockItemSyntax(item: CodeBlockItemSyntax.Item(IntegerLiteralExprSyntax(integerLiteral: 2)))
])
),
expected: """
{ 2 }
""",
using: ClosureCompletionFormat(initialIndentation: .spaces(4))
)
}
}
|
