mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
a339c82c1e
* [Experiment] Make Syntax nodes structs instead of classes * [Experiment] Add Hashable conformance to concrete types * Fix pep8 violation * Remove AnySyntax, explicitly specialize SyntaxCollection nodes * Refine the comment for SyntaxCollection nodes
120 lines
3.8 KiB
Swift
120 lines
3.8 KiB
Swift
// RUN: %target-run-simple-swift
|
|
// REQUIRES: executable_test
|
|
// REQUIRES: OS=macosx
|
|
// REQUIRES: objc_interop
|
|
|
|
import Foundation
|
|
import StdlibUnittest
|
|
import SwiftSyntax
|
|
|
|
func cannedStructDecl() -> StructDeclSyntax {
|
|
let structKW = SyntaxFactory.makeStructKeyword(trailingTrivia: .spaces(1))
|
|
let fooID = SyntaxFactory.makeIdentifier("Foo", trailingTrivia: .spaces(1))
|
|
let rBrace = SyntaxFactory.makeRightBraceToken(leadingTrivia: .newlines(1))
|
|
let members = MemberDeclBlockSyntax {
|
|
$0.useLeftBrace(SyntaxFactory.makeLeftBraceToken())
|
|
$0.useRightBrace(rBrace)
|
|
}
|
|
return StructDeclSyntax {
|
|
$0.useStructKeyword(structKW)
|
|
$0.useIdentifier(fooID)
|
|
$0.useMembers(members)
|
|
}
|
|
}
|
|
|
|
var SyntaxFactoryAPI = TestSuite("SyntaxFactoryAPI")
|
|
|
|
SyntaxFactoryAPI.test("Generated") {
|
|
|
|
let structDecl = cannedStructDecl()
|
|
|
|
expectEqual("\(structDecl)",
|
|
"""
|
|
struct Foo {
|
|
}
|
|
""")
|
|
|
|
let forType = SyntaxFactory.makeIdentifier("for",
|
|
leadingTrivia: .backticks(1),
|
|
trailingTrivia: [
|
|
.backticks(1), .spaces(1)
|
|
])
|
|
let newBrace = SyntaxFactory.makeRightBraceToken(leadingTrivia: .newlines(2))
|
|
|
|
let renamed = structDecl.withIdentifier(forType)
|
|
.withMembers(structDecl.members
|
|
.withRightBrace(newBrace))
|
|
|
|
expectEqual("\(renamed)",
|
|
"""
|
|
struct `for` {
|
|
|
|
}
|
|
""")
|
|
|
|
expectNotEqual(structDecl.members, renamed.members)
|
|
expectEqual(structDecl, structDecl.root as? StructDeclSyntax)
|
|
expectNil(structDecl.parent)
|
|
expectNotNil(structDecl.members.parent)
|
|
expectEqual(structDecl.members.parent as? StructDeclSyntax, structDecl)
|
|
|
|
// Ensure that accessing children via named identifiers is exactly the
|
|
// same as accessing them as their underlying data.
|
|
expectEqual(structDecl.members,
|
|
structDecl.child(at: 7) as? MemberDeclBlockSyntax)
|
|
|
|
expectEqual("\(structDecl.members.rightBrace)",
|
|
"""
|
|
|
|
}
|
|
""")
|
|
}
|
|
|
|
SyntaxFactoryAPI.test("TokenSyntax") {
|
|
let tok = SyntaxFactory.makeStructKeyword()
|
|
expectEqual("\(tok)", "struct")
|
|
expectTrue(tok.isPresent)
|
|
|
|
let preSpacedTok = tok.withLeadingTrivia(.spaces(3))
|
|
expectEqual("\(preSpacedTok)", " struct")
|
|
|
|
let postSpacedTok = tok.withTrailingTrivia(.spaces(6))
|
|
expectEqual("\(postSpacedTok)", "struct ")
|
|
|
|
let prePostSpacedTok = preSpacedTok.withTrailingTrivia(.spaces(4))
|
|
expectEqual("\(prePostSpacedTok)", " struct ")
|
|
}
|
|
|
|
SyntaxFactoryAPI.test("FunctionCallSyntaxBuilder") {
|
|
let string = SyntaxFactory.makeStringLiteralExpr("Hello, world!")
|
|
let printID = SyntaxFactory.makeVariableExpr("print")
|
|
let arg = FunctionCallArgumentSyntax {
|
|
$0.useExpression(string)
|
|
}
|
|
let call = FunctionCallExprSyntax {
|
|
$0.useCalledExpression(printID)
|
|
$0.useLeftParen(SyntaxFactory.makeLeftParenToken())
|
|
$0.addFunctionCallArgument(arg)
|
|
$0.useRightParen(SyntaxFactory.makeRightParenToken())
|
|
}
|
|
expectEqual("\(call)", "print(\"Hello, world!\")")
|
|
|
|
let terminatorArg = FunctionCallArgumentSyntax {
|
|
$0.useLabel(SyntaxFactory.makeIdentifier("terminator"))
|
|
$0.useColon(SyntaxFactory.makeColonToken(trailingTrivia: .spaces(1)))
|
|
$0.useExpression(SyntaxFactory.makeStringLiteralExpr(" "))
|
|
}
|
|
let callWithTerminator = call.withArgumentList(
|
|
SyntaxFactory.makeFunctionCallArgumentList([
|
|
arg.withTrailingComma(
|
|
SyntaxFactory.makeCommaToken(trailingTrivia: .spaces(1))),
|
|
terminatorArg
|
|
])
|
|
)
|
|
|
|
expectEqual("\(callWithTerminator)",
|
|
"print(\"Hello, world!\", terminator: \" \")")
|
|
}
|
|
|
|
runAllTests()
|