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
|
//===----------------------------------------------------------------------===//
//
// 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 MacroExamplesImplementation
import SwiftSyntaxMacros
import SwiftSyntaxMacrosTestSupport
import XCTest
final class DefaultFatalErrorImplementationMacroTests: XCTestCase {
private let macros = ["defaultFatalErrorImplementation": DefaultFatalErrorImplementationMacro.self]
func testExpansionWhenAttachedToProtocolExpandsCorrectly() {
assertMacroExpansion(
"""
@defaultFatalErrorImplementation
protocol MyProtocol {
func foo()
func bar() -> Int
}
""",
expandedSource: """
protocol MyProtocol {
func foo()
func bar() -> Int
}
extension MyProtocol {
func foo() {
fatalError("whoops 😅")
}
func bar() -> Int {
fatalError("whoops 😅")
}
}
""",
macros: macros,
indentationWidth: .spaces(2)
)
}
func testExpansionWhenNotAttachedToProtocolProducesDiagnostic() {
assertMacroExpansion(
"""
@defaultFatalErrorImplementation
class MyClass {}
""",
expandedSource: """
class MyClass {}
""",
diagnostics: [
DiagnosticSpec(
message: "Macro `defaultFatalErrorImplementation` can only be applied to a protocol",
line: 1,
column: 1
)
],
macros: macros,
indentationWidth: .spaces(2)
)
}
func testExpansionWhenAttachedToEmptyProtocolDoesNotAddExtension() {
assertMacroExpansion(
"""
@defaultFatalErrorImplementation
protocol EmptyProtocol {}
""",
expandedSource: """
protocol EmptyProtocol {}
""",
macros: macros,
indentationWidth: .spaces(2)
)
}
}
|