//===--- Join.swift.gyb ---------------------------------------*- swift -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

// RUN: rm -rf %t && mkdir -p %t && %S/../../utils/gyb %s -o %t/Join.swift
// RUN: %S/../../utils/line-directive %t/Join.swift -- %target-build-swift %t/Join.swift -o %t/a.out
// RUN: %S/../../utils/line-directive %t/Join.swift -- %target-run %t/a.out
// REQUIRES: executable_test

import StdlibUnittest
import StdlibCollectionUnittest

// Also import modules which are used by StdlibUnittest internally. This
// workaround is needed to link all required libraries in case we compile
// StdlibUnittest with -sil-serialize-all.
#if _runtime(_ObjC)
import ObjectiveC
#endif

public struct JoinWithSeparatorTest {
  public let elements: [[Int]]
  public let separator: [Int]
  public let expected: [Int]
  public let loc: SourceLoc

  init(
    elements: [[Int]], separator: [Int], expected: [Int],
    file: String = #file, line: UInt = #line
  ) {
    self.elements = elements
    self.separator = separator
    self.expected = expected
    self.loc = SourceLoc(file, line, comment: "prefix() test data")
  }
}

public let joinWithSeparatorTests = [
  // Empty separator.
  JoinWithSeparatorTest(
    elements: [],
    separator: [],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[]],
    separator: [],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[], []],
    separator: [],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[], [], []],
    separator: [],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[ 1010 ]],
    separator: [],
    expected: [ 1010 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ]],
    separator: [],
    expected: [ 1010, 2020 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], [ 3030 ]],
    separator: [],
    expected: [ 1010, 2020, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020 ], [ 3030 ]],
    separator: [],
    expected: [ 2020, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [], [ 3030 ]],
    separator: [],
    expected: [ 1010, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], []],
    separator: [],
    expected: [ 1010, 2020 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [],
    expected: [ 1010, 1011, 2020, 2021, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [],
    expected: [ 2020, 2021, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [], [ 3030, 3031 ]],
    separator: [],
    expected: [ 1010, 1011, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], []],
    separator: [],
    expected: [ 1010, 1011, 2020, 2021 ]),

  // 1-element separator.
  JoinWithSeparatorTest(
    elements: [],
    separator: [ 99 ],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[]],
    separator: [ 99 ],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[], []],
    separator: [ 99 ],
    expected: [ 99 ]),

  JoinWithSeparatorTest(
    elements: [[], [], []],
    separator: [ 99 ],
    expected: [ 99, 99 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ]],
    separator: [ 99 ],
    expected: [ 1010 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ]],
    separator: [ 99 ],
    expected: [ 1010, 99, 2020 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], [ 3030 ]],
    separator: [ 99 ],
    expected: [ 1010, 99, 2020, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020 ], [ 3030 ]],
    separator: [ 99 ],
    expected: [ 99, 2020, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [], [ 3030 ]],
    separator: [ 99 ],
    expected: [ 1010, 99, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], []],
    separator: [ 99 ],
    expected: [ 1010, 99, 2020, 99 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [ 99 ],
    expected: [ 1010, 1011, 99, 2020, 2021, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [ 99 ],
    expected: [ 99, 2020, 2021, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [], [ 3030, 3031 ]],
    separator: [ 99 ],
    expected: [ 1010, 1011, 99, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], []],
    separator: [ 99 ],
    expected: [ 1010, 1011, 99, 2020, 2021, 99 ]),

  // 2-element separator.
  JoinWithSeparatorTest(
    elements: [],
    separator: [ 98, 99 ],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[]],
    separator: [ 98, 99 ],
    expected: []),

  JoinWithSeparatorTest(
    elements: [[], []],
    separator: [ 98, 99 ],
    expected: [ 98, 99 ]),

  JoinWithSeparatorTest(
    elements: [[], [], []],
    separator: [ 98, 99 ],
    expected: [ 98, 99, 98, 99 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ]],
    separator: [ 98, 99 ],
    expected: [ 1010 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ]],
    separator: [ 98, 99 ],
    expected: [ 1010, 98, 99, 2020 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], [ 3030 ]],
    separator: [ 98, 99 ],
    expected: [ 1010, 98, 99, 2020, 98, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020 ], [ 3030 ]],
    separator: [ 98, 99 ],
    expected: [ 98, 99, 2020, 98, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [], [ 3030 ]],
    separator: [ 98, 99 ],
    expected: [ 1010, 98, 99, 98, 99, 3030 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010 ], [ 2020 ], []],
    separator: [ 98, 99 ],
    expected: [ 1010, 98, 99, 2020, 98, 99 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [ 98, 99 ],
    expected: [ 1010, 1011, 98, 99, 2020, 2021, 98, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[], [ 2020, 2021 ], [ 3030, 3031 ]],
    separator: [ 98, 99 ],
    expected: [ 98, 99, 2020, 2021, 98, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [], [ 3030, 3031 ]],
    separator: [ 98, 99 ],
    expected: [ 1010, 1011, 98, 99, 98, 99, 3030, 3031 ]),

  JoinWithSeparatorTest(
    elements: [[ 1010, 1011 ], [ 2020, 2021 ], []],
    separator: [ 98, 99 ],
    expected: [ 1010, 1011, 98, 99, 2020, 2021, 98, 99 ]),
]

var JoinTestSuite = TestSuite("Join")

% from itertools import product
% traversals = [ 'Forward', 'Bidirectional', 'RandomAccess' ]
% base_kinds = [ 'Defaulted', 'Minimal' ]
% collections = [
%   ('{}{}RangeReplaceableCollection'.format(base_kind, traversal), traversal)
%     for base_kind, traversal in product(base_kinds, traversals)
% ]
% other_array_types = [ 'Array', 'ArraySlice', 'ContiguousArray' ]
% collections += [(array, 'RandomAccess') for array in other_array_types]

% for Base, traversal in collections:
%   if Base not in other_array_types:
%     label = 'elements:'
%   else:
%     label = ''
%   end

JoinTestSuite.test("${Base}.join()") {
  for test in joinWithSeparatorTests {
    let elements = ${Base}(${label} test.elements.map {
      ${Base}(${label} $0)
    })
    let separator = ${Base}(${label} test.separator)
    let r = Array(elements.joined(separator: separator))
    checkSequence(test.expected, r)
  }
}

JoinTestSuite.test("${Base}.join()/_copyToNativeArrayBuffer()") {
  for test in joinWithSeparatorTests {
    let elements = ${Base}(${label} test.elements.map {
      ${Base}(${label} $0)
    })
    let separator = ${Base}(${label} test.separator)
    let r = Array(elements.joined(separator: separator))
    checkSequence(test.expected, r)
  }
}

% end

func join(separator: String, _ elements: [String]) -> String {
  return elements.joined(separator: separator)
}

JoinTestSuite.test("String.joined(separator:)") {
  //
  // Test the free function.
  //

  // Empty separator.
  expectEqual("",    join("", []))
  expectEqual("",    join("", [""]))
  expectEqual("",    join("", ["", ""]))
  expectEqual("",    join("", ["", "", ""]))
  expectEqual("a",   join("", ["a"]))
  expectEqual("ab",  join("", ["a", "b"]))
  expectEqual("abc", join("", ["a", "b", "c"]))
  expectEqual("abcdef", join("", ["ab", "cd", "ef"]))

  // 1-element separator.
  expectEqual("",      join("x", [""]))
  expectEqual("x",     join("x", ["", ""]))
  expectEqual("xx",    join("x", ["", "", ""]))
  expectEqual("a",     join("x", ["a"]))
  expectEqual("axb",   join("x", ["a", "b"]))
  expectEqual("axbxc", join("x", ["a", "b", "c"]))
  expectEqual("abxcdxef", join("x", ["ab", "cd", "ef"]))

  // 2-element separator.
  expectEqual("",        join("xy", [""]))
  expectEqual("xy",      join("xy", ["", ""]))
  expectEqual("xyxy",    join("xy", ["", "", ""]))
  expectEqual("a",       join("xy", ["a"]))
  expectEqual("axyb",    join("xy", ["a", "b"]))
  expectEqual("axybxyc", join("xy", ["a", "b", "c"]))
  expectEqual("abxycdxyef", join("xy", ["ab", "cd", "ef"]))

  //
  // Test forwarding instance function.
  //

  expectEqual("abxycdxyef", ["ab", "cd", "ef"].joined(separator: "xy"))
}

runAllTests()