Merge pull request #315 from kimdv/kimdv/add-identifier-expr-convenience-initializers

Add convenience initializers for IdentifierExpr

Author: ahoppen

Sources/SwiftSyntaxBuilder/IdentifierExprConvenienceInitializers.swift

@@ -0,0 +1,23 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2021 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
+//
+//===----------------------------------------------------------------------===//
+
+extension IdentifierExpr {
+  public init(_ identifier: String) {
+    self.init(identifier: .identifier(identifier))
+  }
+}
+
+extension IdentifierExpr: ExpressibleByStringLiteral {
+  public init(stringLiteral value: String) {
+    self.init(value)
+  }
+}

Tests/SwiftSyntaxBuilderTest/IdentifierExprTests.swift

@@ -0,0 +1,27 @@
+import XCTest
+import SwiftSyntax
+
+import SwiftSyntaxBuilder
+
+final class IdentifierExprTests: XCTestCase {
+  func testStringLiteral() {
+    let leadingTrivia = Trivia.garbageText("␣")
+
+    let testCases: [UInt: (IdentifierExpr, String)] = [
+      #line: (IdentifierExpr(identifier: .identifier("Test")), "␣Test"),
+      #line: (IdentifierExpr(stringLiteral: "Test"), "␣Test"),
+      #line: (IdentifierExpr("Test"), "␣Test"),
+      #line: ("Test", "␣Test")
+    ]
+
+    for (line, testCase) in testCases {
+      let (builder, expected) = testCase
+      let syntax = builder.buildSyntax(format: Format(), leadingTrivia: leadingTrivia)
+
+      var text = ""
+      syntax.write(to: &text)
+
+      XCTAssertEqual(text, expected, line: line)
+    }
+  }
+}