Merge pull request #3264 from austinzheng/az-tests2

Author: swift-ci

validation-test/stdlib/UnsafeBufferPointer.swift.gyb

@@ -90,6 +90,178 @@ let subscriptGetTests : [SubscriptGetTest] = [
   SubscriptGetTest(rangeSelection: .offsets(-100, 100)),
 ]
 
+struct SubscriptSetTest {
+  // SubscriptSetTest operates on a `(end - start)` sized buffer containing
+  // monotonically increasing integers from `start` to `end - 1`.
+  static let start = 0
+  static let end = 10
+
+  let rangeSelection: RangeSelection
+
+  let replacementValues: [OpaqueValue<Int>]
+  let replacementValuesClosed: [OpaqueValue<Int>]
+
+  /// The values that should be expected by slicing the UBP, or `nil` if the
+  /// test is expected to crash.
+  let expectedValues: [Int]?
+  let expectedValuesClosed: [Int]?
+
+  let loc: SourceLoc
+
+  static var elementCount = (end - start)
+
+  /// Create and populate an `UnsafeMutableBufferPointer` for use with unit
+  /// tests.
+  /// PRECONDITION: `memory` must be allocated with space for
+  /// `SubscriptSetTest.elementCount` elements. 
+  func createUnsafeMutableBufferPointer(
+    from memory: UnsafeMutablePointer<OpaqueValue<Int>>
+  ) -> UnsafeMutableBufferPointer<OpaqueValue<Int>> 
+  {
+    for i in SubscriptSetTest.start..<SubscriptSetTest.end {
+      memory[i] = OpaqueValue(i)  
+    }
+    return UnsafeMutableBufferPointer(
+      start: memory, 
+      count: SubscriptSetTest.elementCount)
+  }
+
+  /// Create and populate a mutable buffer pointer slice for use with unit
+  /// tests.
+  /// PRECONDITION: `memory` must be allocated with space for
+  /// `replacementValues.count` elements.
+  func replacementValuesSlice(
+    from memory: UnsafeMutablePointer<OpaqueValue<Int>>,
+    replacementValues: [OpaqueValue<Int>]
+  ) -> MutableRandomAccessSlice<UnsafeMutableBufferPointer<OpaqueValue<Int>>>
+  {
+    for (i, value) in replacementValues.enumerated() {
+      memory[i] = value
+    }
+    let buffer = UnsafeMutableBufferPointer(
+      start: memory, 
+      count: replacementValues.count)
+    let fullRange = RangeSelection.full.range(in: buffer)
+    return buffer[fullRange]
+  }
+
+  init(
+    rangeSelection: RangeSelection, 
+    replacementValues: [Int],
+    replacementValuesClosed: [Int]? = nil,
+    expectedValues: [Int]? = nil,
+    expectedValuesClosed: [Int]? = nil,
+    file: String = #file, line: UInt = #line
+  ) {
+    self.rangeSelection = rangeSelection
+    self.replacementValues = replacementValues.map { OpaqueValue($0) }
+    if let replacements = replacementValuesClosed {
+      self.replacementValuesClosed = replacements.map { OpaqueValue($0) }
+    } else {
+      self.replacementValuesClosed = self.replacementValues
+    }
+    self.expectedValues = expectedValues
+    self.expectedValuesClosed = expectedValuesClosed ?? expectedValues
+    self.loc = SourceLoc(file, line, comment: "test data")
+  }
+}
+
+let subscriptSetTests : [SubscriptSetTest] = [
+  // Valid, empty.
+  SubscriptSetTest(
+    rangeSelection: .emptyRange,
+    replacementValues: [],
+    expectedValues: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]),
+
+  // Valid, edges.
+  SubscriptSetTest(
+    rangeSelection: .leftEdge,
+    replacementValues: [],
+    replacementValuesClosed: [9001],
+    expectedValues: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
+    expectedValuesClosed: [9001, 1, 2, 3, 4, 5, 6, 7, 8, 9]),
+  SubscriptSetTest(
+    rangeSelection: .rightEdge,
+    replacementValues: [],
+    replacementValuesClosed: [9001],
+    expectedValues: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
+    expectedValuesClosed: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9001]),
+
+  // Valid, internal.
+  SubscriptSetTest(
+    rangeSelection: .leftHalf,
+    replacementValues: [10, 11, 12, 13, 14],
+    expectedValues: [10, 11, 12, 13, 14, 5, 6, 7, 8, 9]),
+  SubscriptSetTest(
+    rangeSelection: .rightHalf,
+    replacementValues: [10, 11, 12, 13, 14],
+    expectedValues: [0, 1, 2, 3, 4, 10, 11, 12, 13, 14]),
+  SubscriptSetTest(
+    rangeSelection: .middle,
+    replacementValues: [10, 11, 12, 13, 14, 15],
+    expectedValues: [0, 1, 10, 11, 12, 13, 14, 15, 8, 9]),
+  SubscriptSetTest(
+    rangeSelection: .full,
+    replacementValues: [10, 11, 12, 13, 14, 15, 16, 17, 18, 19],
+    expectedValues: [10, 11, 12, 13, 14, 15, 16, 17, 18, 19]),
+
+  // Invalid, range and replacement count mismatch.
+  SubscriptSetTest(
+    rangeSelection: .leftEdge,
+    replacementValues: [],
+    replacementValuesClosed: [9]),
+  SubscriptSetTest(
+    rangeSelection: .leftEdge,
+    replacementValues: [9, 9],
+    replacementValuesClosed: [9, 9, 9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(1, 2),
+    replacementValues: [],
+    replacementValuesClosed: [9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(1, 2),
+    replacementValues: [9, 9],
+    replacementValuesClosed: [9, 9, 9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(2, 5),
+    replacementValues: [9, 9],
+    replacementValuesClosed: [9, 9, 9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(2, 5),
+    replacementValues: [9, 9, 9, 9],
+    replacementValuesClosed: [9, 9, 9, 9, 9]),
+
+  // Invalid, bottom out of bounds.
+  SubscriptSetTest(
+    rangeSelection: .offsets(-1, -1),
+    replacementValues: []),
+  SubscriptSetTest(
+    rangeSelection: .offsets(-1, 0),
+    replacementValues: [9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(-3, 5),
+    replacementValues: [9, 9, 9, 9, 9, 9, 9, 9]),
+
+  // Invalid, top out of bounds.
+  SubscriptSetTest(
+    rangeSelection: .offsets(10, 10),
+    replacementValues: []),
+  SubscriptSetTest(
+    rangeSelection: .offsets(9, 10),
+    replacementValues: [9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(8, 15),
+    replacementValues: [9, 9, 9, 9, 9, 9, 9]),
+
+  // Invalid, both out of bounds.
+  SubscriptSetTest(
+    rangeSelection: .offsets(-1, 10),
+    replacementValues: [9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9]),
+  SubscriptSetTest(
+    rangeSelection: .offsets(-2, 11),
+    replacementValues: [9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9]),
+]
+
 
 // Test Suites
 
@@ -228,6 +400,50 @@ ${SelfName}TestSuite.test("subscript/${RangeName}/get").forEach(in: subscriptGet
 
 % end
 
+%   for RangeName in ['range', 'countableRange', 'closedRange', 'countableClosedRange']:
+UnsafeMutableBufferPointerTestSuite.test("subscript/${RangeName}/set")
+  .forEach(in: subscriptSetTests) { (test) in
+
+  let expectedValues: [Int]?
+  let replacementValues: [OpaqueValue<Int>]
+%      if 'closed' in RangeName.lower():
+  if test.rangeSelection.isEmpty {
+    return
+  }
+  expectedValues = test.expectedValuesClosed
+  replacementValues = test.replacementValuesClosed
+%      else:
+  expectedValues = test.expectedValues
+  replacementValues = test.replacementValues
+%      end
+
+  let elementCount = SubscriptSetTest.elementCount
+
+  var memory = UnsafeMutablePointer<OpaqueValue<Int>>(
+    allocatingCapacity: elementCount)
+  var sliceMemory = UnsafeMutablePointer<OpaqueValue<Int>>(
+    allocatingCapacity: replacementValues.count)
+  var buffer = test.createUnsafeMutableBufferPointer(from: memory)
+  defer { 
+    memory.deallocateCapacity(elementCount) 
+    sliceMemory.deallocateCapacity(replacementValues.count)
+  }
+
+  let range = test.rangeSelection.${RangeName}(in: buffer)
+  let replacementSlice = test.replacementValuesSlice(
+    from: sliceMemory,
+    replacementValues: replacementValues)
+
+  if expectedValues == nil { expectCrashLater() }
+  buffer[range] = replacementSlice
+  expectEqual(
+    expectedValues!,
+    buffer.map { $0.value },
+    stackTrace: SourceLocStack().with(test.loc)
+  )
+}
+% end
+
 UnsafeMutableBufferPointerTestSuite.test("changeElementViaBuffer") {
   let count = 4
   let allocated = UnsafeMutablePointer<Float>(allocatingCapacity: count)