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
|
import SwiftFormat
@_spi(Rules) @_spi(Testing) import SwiftFormat
import SwiftParser
import XCTest
class ImportsXCTestVisitorTests: XCTestCase {
func testDoesNotImportXCTest() throws {
XCTAssertEqual(
try makeContextAndSetImportsXCTest(
source: """
import Foundation
"""
),
.doesNotImportXCTest
)
}
func testImportsXCTest() throws {
XCTAssertEqual(
try makeContextAndSetImportsXCTest(
source: """
import Foundation
import XCTest
"""
),
.importsXCTest
)
}
func testImportsSpecificXCTestDecl() throws {
XCTAssertEqual(
try makeContextAndSetImportsXCTest(
source: """
import Foundation
import class XCTest.XCTestCase
"""
),
.importsXCTest
)
}
func testImportsXCTestInsideConditional() throws {
XCTAssertEqual(
try makeContextAndSetImportsXCTest(
source: """
import Foundation
#if SOME_FEATURE_FLAG
import XCTest
#endif
"""
),
.importsXCTest
)
}
/// Parses the given source, makes a new `Context`, then populates and returns its `XCTest`
/// import state.
private func makeContextAndSetImportsXCTest(source: String) throws -> Context.XCTestImportState {
let sourceFile = Parser.parse(source: source)
let context = Context(
configuration: Configuration(),
operatorTable: .standardOperators,
findingConsumer: { _ in },
fileURL: URL(fileURLWithPath: "/tmp/test.swift"),
sourceFileSyntax: sourceFile,
ruleNameCache: ruleNameCache
)
setImportsXCTest(context: context, sourceFile: sourceFile)
return context.importsXCTest
}
}
|
