swift-argument-parser/Tests/ArgumentParserUnitTests/StringWrappingTests.swift

//===----------------------------------------------------------*- 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 XCTest
@testable import ArgumentParser

final class StringWrappingTests: XCTestCase {}

let shortSample = """
Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Lectus proin nibh nisl condimentum id. Semper feugiat nibh sed pulvinar proin gravida hendrerit. Massa id neque aliquam vestibulum morbi blandit cursus risus at. Iaculis urna id volutpat lacus laoreet. Netus et malesuada fames ac turpis egestas sed tempus urna.
"""

let longSample = """
Pretium vulputate sapien nec sagittis aliquam malesuada bibendum. Ut diam quam nulla porttitor.

Egestas egestas fringilla phasellus faucibus. Amet dictum sit amet justo donec enim diam. Consectetur adipiscing elit duis tristique.

Enim lobortis scelerisque fermentum dui.

Et leo duis ut diam quam.

Integer eget aliquet nibh praesent tristique magna sit. Faucibus turpis in eu mi bibendum neque egestas congue quisque.       Risus nec feugiat in fermentum posuere urna nec tincidunt.
"""

let jsonSample = """
{
  "level1": {
    "level2": {
      "level3": true
    }
  }
}
"""

// MARK: -

extension StringWrappingTests {
  func testShort() {
    XCTAssertEqual(shortSample.wrapped(to: 40), """
            Lorem ipsum dolor sit amet, consectetur
            adipiscing elit, sed do eiusmod tempor
            incididunt ut labore et dolore magna
            aliqua. Lectus proin nibh nisl
            condimentum id. Semper feugiat nibh sed
            pulvinar proin gravida hendrerit. Massa
            id neque aliquam vestibulum morbi
            blandit cursus risus at. Iaculis urna
            id volutpat lacus laoreet. Netus et
            malesuada fames ac turpis egestas sed
            tempus urna.
            """)

    XCTAssertEqual(shortSample.wrapped(to: 80), """
            Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor
            incididunt ut labore et dolore magna aliqua. Lectus proin nibh nisl condimentum
            id. Semper feugiat nibh sed pulvinar proin gravida hendrerit. Massa id neque
            aliquam vestibulum morbi blandit cursus risus at. Iaculis urna id volutpat
            lacus laoreet. Netus et malesuada fames ac turpis egestas sed tempus urna.
            """)
  }

  func testShortWithIndent() {
    XCTAssertEqual(shortSample.wrapped(to: 60, wrappingIndent: 10), """
                      Lorem ipsum dolor sit amet, consectetur
                      adipiscing elit, sed do eiusmod tempor incididunt
                      ut labore et dolore magna aliqua. Lectus proin
                      nibh nisl condimentum id. Semper feugiat nibh sed
                      pulvinar proin gravida hendrerit. Massa id neque
                      aliquam vestibulum morbi blandit cursus risus at.
                      Iaculis urna id volutpat lacus laoreet. Netus et
                      malesuada fames ac turpis egestas sed tempus urna.
            """)
  }

  func testLong() {
    XCTAssertEqual(longSample.wrapped(to: 40), """
            Pretium vulputate sapien nec sagittis
            aliquam malesuada bibendum. Ut diam
            quam nulla porttitor.

            Egestas egestas fringilla phasellus
            faucibus. Amet dictum sit amet justo
            donec enim diam. Consectetur adipiscing
            elit duis tristique.

            Enim lobortis scelerisque fermentum
            dui.

            Et leo duis ut diam quam.

            Integer eget aliquet nibh praesent
            tristique magna sit. Faucibus turpis in
            eu mi bibendum neque egestas congue
            quisque.       Risus nec feugiat in
            fermentum posuere urna nec tincidunt.
            """)

    XCTAssertEqual(longSample.wrapped(to: 80), """
            Pretium vulputate sapien nec sagittis aliquam malesuada bibendum. Ut diam quam
            nulla porttitor.

            Egestas egestas fringilla phasellus faucibus. Amet dictum sit amet justo donec
            enim diam. Consectetur adipiscing elit duis tristique.

            Enim lobortis scelerisque fermentum dui.

            Et leo duis ut diam quam.

            Integer eget aliquet nibh praesent tristique magna sit. Faucibus turpis in eu
            mi bibendum neque egestas congue quisque.       Risus nec feugiat in fermentum
            posuere urna nec tincidunt.
            """)
  }

  func testLongWithIndent() {
    XCTAssertEqual(longSample.wrapped(to: 60, wrappingIndent: 10), """
                      Pretium vulputate sapien nec sagittis aliquam
                      malesuada bibendum. Ut diam quam nulla porttitor.

                      Egestas egestas fringilla phasellus faucibus.
                      Amet dictum sit amet justo donec enim diam.
                      Consectetur adipiscing elit duis tristique.

                      Enim lobortis scelerisque fermentum dui.

                      Et leo duis ut diam quam.

                      Integer eget aliquet nibh praesent tristique
                      magna sit. Faucibus turpis in eu mi bibendum
                      neque egestas congue quisque.       Risus nec
                      feugiat in fermentum posuere urna nec tincidunt.
            """)

  }

  func testJSON() {
    XCTAssertEqual(jsonSample.wrapped(to: 80), """
            {
              "level1": {
                "level2": {
                  "level3": true
                }
              }
            }
            """)
  }

  func testJSONWithIndent() {
    XCTAssertEqual(jsonSample.wrapped(to: 80, wrappingIndent: 10), """
                      {
                        "level1": {
                          "level2": {
                            "level3": true
                          }
                        }
                      }
            """)
  }

  func testIndent() {
    XCTAssertEqual(
      shortSample.wrapped(to: 40).indentingEachLine(by: 10),
      shortSample.wrapped(to: 50, wrappingIndent: 10))
    XCTAssertEqual(
      longSample.wrapped(to: 40).indentingEachLine(by: 10),
      longSample.wrapped(to: 50, wrappingIndent: 10))

    XCTAssertEqual("".indentingEachLine(by: 10), "")
    XCTAssertEqual("\n".indentingEachLine(by: 10), "\n")
    XCTAssertEqual("a\n".indentingEachLine(by: 10), "          a\n")
    XCTAssertEqual("\na\n".indentingEachLine(by: 10), "\n          a\n")
    XCTAssertEqual("a\n\nb\n".indentingEachLine(by: 10),
                   "          a\n\n          b\n")
    XCTAssertEqual("\na\n\nb\n".indentingEachLine(by: 10),
                   "\n          a\n\n          b\n")
  }

  func testHangingIndent() {
    XCTAssertEqual(
      shortSample.wrapped(to: 40).hangingIndentingEachLine(by: 10),
      String(shortSample.wrapped(to: 50, wrappingIndent: 10).dropFirst(10)))
    XCTAssertEqual(
      longSample.wrapped(to: 40).hangingIndentingEachLine(by: 10),
      String(longSample.wrapped(to: 50, wrappingIndent: 10).dropFirst(10)))

    XCTAssertEqual("".hangingIndentingEachLine(by: 10), "")
    XCTAssertEqual("\n".hangingIndentingEachLine(by: 10), "\n")
    XCTAssertEqual("a\n".hangingIndentingEachLine(by: 10), "a\n")
    XCTAssertEqual("\na\n".hangingIndentingEachLine(by: 10), "\n          a\n")
    XCTAssertEqual("a\n\nb\n".hangingIndentingEachLine(by: 10),
                   "a\n\n          b\n")
    XCTAssertEqual("\na\n\nb\n".hangingIndentingEachLine(by: 10),
                   "\n          a\n\n          b\n")

  }
}