ARK-OS/arkos/frameworks/swift-argument-parser/Tests/ArgumentParserExampleTests/RepeatExampleTests.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift Argument Parser open source project
//
// Copyright (c) 2020 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
//
//===----------------------------------------------------------------------===//

import ArgumentParserTestHelpers
import XCTest

@testable import ArgumentParser

final class RepeatExampleTests: XCTestCase {
  override func setUp() {
    Platform.Environment[.columns] = nil
  }

  func testRepeat() throws {
    try AssertExecuteCommand(
      command: "repeat hello",
      expected: """
        hello
        hello

        """)
  }

  func testRepeat_include_counter() throws {
    try AssertExecuteCommand(
      command: "repeat --include-counter hello",
      expected: """
        1: hello
        2: hello

        """)
  }

  func testRepeat_Count() throws {
    try AssertExecuteCommand(
      command: "repeat hello --count 6",
      expected: """
        hello
        hello
        hello
        hello
        hello
        hello

        """)
  }

  func testRepeat_Help() throws {
    let helpText = """
      USAGE: repeat [--count <count>] [--include-counter] <phrase>

      ARGUMENTS:
        <phrase>                The phrase to repeat.

      OPTIONS:
        --count <count>         How many times to repeat 'phrase'.
        --include-counter       Include a counter with each repetition.
        -h, --help              Show help information.


      """

    try AssertExecuteCommand(command: "repeat -h", expected: helpText)
    try AssertExecuteCommand(command: "repeat --help", expected: helpText)
  }

  func testRepeat_Fail() throws {
    try AssertExecuteCommand(
      command: "repeat",
      expected: """
        Error: Missing expected argument '<phrase>'

        USAGE: repeat [--count <count>] [--include-counter] <phrase>

        ARGUMENTS:
          <phrase>                The phrase to repeat.

        OPTIONS:
          --count <count>         How many times to repeat 'phrase'.
          --include-counter       Include a counter with each repetition.
          -h, --help              Show help information.


        """,
      exitCode: .validationFailure)

    try AssertExecuteCommand(
      command: "repeat hello --count",
      expected: """
        Error: Missing value for '--count <count>'
        Help:  --count <count>  How many times to repeat 'phrase'.
        Usage: repeat [--count <count>] [--include-counter] <phrase>
          See 'repeat --help' for more information.

        """,
      exitCode: .validationFailure)

    try AssertExecuteCommand(
      command: "repeat hello --count ZZZ",
      expected: """
        Error: The value 'ZZZ' is invalid for '--count <count>'
        Help:  --count <count>  How many times to repeat 'phrase'.
        Usage: repeat [--count <count>] [--include-counter] <phrase>
          See 'repeat --help' for more information.

        """,
      exitCode: .validationFailure)

    try AssertExecuteCommand(
      command: "repeat --version hello",
      expected: """
        Error: Unknown option '--version'
        Usage: repeat [--count <count>] [--include-counter] <phrase>
          See 'repeat --help' for more information.

        """,
      exitCode: .validationFailure)
  }
}