InitializerDeclTests.swift

//===----------------------------------------------------------------------===//
//
// 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 SwiftSyntax
import SwiftSyntaxBuilder
import XCTest

final class InitializerDeclTests: XCTestCase {
  func testInitializerDecl() {
    let builder = DeclSyntax(
      """
      public init(errorCode: Int) {
        self.code = errorCode
      }
      """
    )

    assertBuildResult(
      builder,
      """
      public init(errorCode: Int) {
        self.code = errorCode
      }
      """
    )
  }

  func testFailableInitializerDecl() {
    let builder = DeclSyntax(
      """
      public init?(errorCode: Int) {
        guard errorCode > 0 else { return nil }
        self.code = errorCode
      }
      """
    )

    assertBuildResult(
      builder,
      """
      public init?(errorCode: Int) {
        guard errorCode > 0 else {
            return nil
        }
        self.code = errorCode
      }
      """
    )
  }

  func testMultilineParameterList() {
    let builder = DeclSyntax(
      """
      init(
        _ p1: Int,
        p2: Int,
        _ p3: Int,
        p4: Int,
        _ p5: Int
      ) {
        self.init(p1 + p2 + p3 + p4 + p5)
      }
      """
    )

    assertBuildResult(
      builder,
      """
      init(
        _ p1: Int,
        p2: Int,
        _ p3: Int,
        p4: Int,
        _ p5: Int
      ) {
        self.init(p1 + p2 + p3 + p4 + p5)
      }
      """
    )
  }

  func testParseableInitializer() throws {
    assertBuildResult(
      try InitializerDeclSyntax("init() {}"),
      """
      init() {
      }
      """
    )
  }
}