// RUN: %target-run-simple-swift
// REQUIRES: executable_test
// REQUIRES: objc_interop

import StdlibUnittest
import ObjectiveC // for autoreleasepool

var suite = TestSuite("AutoreleasingUnsafeMutablePointer")

/// Call `body` passing it an AutoreleasingUnsafeMutablePointer whose pointee
/// has the specified value, allocated in a temporary buffer.
func withAUMP<Pointee: AnyObject>(
  as type: Pointee.Type = Pointee.self,
  initialValue: Optional<Unmanaged<Pointee>> = nil,
  _ body: (AutoreleasingUnsafeMutablePointer<Optional<Pointee>>) -> Void
) {
  let p =
    UnsafeMutablePointer<Optional<Unmanaged<Pointee>>>.allocate(capacity: 1)
  p.initialize(to: initialValue)
  body(AutoreleasingUnsafeMutablePointer(p))
  p.deallocate()
}

/// Call `body` passing it an AutoreleasingUnsafeMutablePointer whose pointee
/// has the specified value, allocated in a temporary buffer.
func withAUMP<Pointee: AnyObject>(
  initialValues: [Unmanaged<Pointee>?],
  _ body: (AutoreleasingUnsafeMutablePointer<Optional<Pointee>>) -> Void
) {
  let p = UnsafeMutablePointer<Optional<Unmanaged<Pointee>>>.allocate(
    capacity: initialValues.count
  )
  for i in 0 ..< initialValues.count {
    (p + i).initialize(to: initialValues[i])
  }

  let aump = AutoreleasingUnsafeMutablePointer<Optional<Pointee>>(p)
  body(aump)
  p.deallocate()
}

suite.test("helper calls its closure exactly once") {
  // `withAUMP` is expected to call its closure exactly once, with an argument
  // pointing to a nil value.
  var runCount = 0
  withAUMP(as: LifetimeTracked.self) { p in
    runCount += 1
    expectNil(p.pointee)
  }
  expectEqual(runCount, 1)
}

suite.test("init doesn't autorelease") {
  let originalInstances = LifetimeTracked.instances
  let unmanaged = Unmanaged.passRetained(LifetimeTracked(42))
  expectEqual(LifetimeTracked.instances, originalInstances + 1)

  withAUMP(initialValue: unmanaged) { p in noop(p) }

  // Releasing the original reference should immediately deallocate the
  // object.
  expectEqual(LifetimeTracked.instances, originalInstances + 1)
  unmanaged.release()
  expectEqual(LifetimeTracked.instances, originalInstances)
}

suite.test("getter initially returns the initial value") {
  withAUMP(as: LifetimeTracked.self) { p in
    expectNil(p.pointee)
  }

  let unmanaged = Unmanaged.passRetained(LifetimeTracked(42))
  withAUMP(initialValue: unmanaged) { p in
    expectTrue(p.pointee === unmanaged.takeUnretainedValue())
  }
  unmanaged.release()
}

suite.test("pointee getter returns the last value set") {
  autoreleasepool {
    withAUMP(as: LifetimeTracked.self) { p in
      expectNil(p.pointee)
      let object = LifetimeTracked(23)
      p.pointee = object
      expectTrue(p.pointee === object)
      let other = LifetimeTracked(42)
      p.pointee = other
      expectTrue(p.pointee === other)
      p.pointee = nil
      expectNil(p.pointee)
    }
  }
}

suite.test("pointee getter doesn't autorelease") {
  let originalInstances = LifetimeTracked.instances
  autoreleasepool {
    let unmanaged = Unmanaged.passRetained(LifetimeTracked(42))
    expectEqual(LifetimeTracked.instances, originalInstances + 1)

    withAUMP(initialValue: unmanaged) { p in
      expectTrue(p.pointee === unmanaged.takeUnretainedValue())
    }
    // Releasing the original reference should immediately deallocate the
    // object.
    expectEqual(LifetimeTracked.instances, originalInstances + 1)
    unmanaged.release()
    expectEqual(LifetimeTracked.instances, originalInstances)
  }
}

suite.test("pointee setter autoreleases the new value") {
  let originalInstances = LifetimeTracked.instances
  autoreleasepool {
    withAUMP(as: LifetimeTracked.self) { p in
      let object = LifetimeTracked(42)
      // There is one more instance now.
      expectEqual(LifetimeTracked.instances, originalInstances + 1)
      p.pointee = object
    }
    // The strong reference to `object` is gone, but the autoreleasepool
    // is still holding onto it.
    expectEqual(LifetimeTracked.instances, originalInstances + 1)
  }
  // Draining the pool deallocates the instance.
  expectEqual(LifetimeTracked.instances, originalInstances)
}

suite.test("AutoreleasingUnsafeMutablePointer doesn't retain its value") {
  let originalInstances = LifetimeTracked.instances
  withAUMP(as: LifetimeTracked.self) { p in
    autoreleasepool {
      let object = LifetimeTracked(42)
      p.pointee = object
      expectEqual(LifetimeTracked.instances, originalInstances + 1)
    }
    // Draining the pool deallocates the instance, even though
    // the autoreleasing pointer still points to it.
    // This is okay as long as the pointer isn't dereferenced.
    expectEqual(LifetimeTracked.instances, originalInstances)
  }
  expectEqual(LifetimeTracked.instances, originalInstances)
}

suite.test("subscript[0] is the same as pointee.getter") {
  withAUMP(as: LifetimeTracked.self) { p in
    expectNil(p[0])
  }

  let unmanaged = Unmanaged.passRetained(LifetimeTracked(42))
  withAUMP(initialValue: unmanaged) { p in
    expectTrue(p[0] === unmanaged.takeUnretainedValue())
  }
  unmanaged.release()
}


suite.test("subscript and advanced(by:) works") {
  let originalInstances = LifetimeTracked.instances
  let count = 100

  var refs = 0
  let values: [Unmanaged<LifetimeTracked>?] = (0 ..< count).map { i in
    if i % 10 == 0 {
      refs += 1
      return Unmanaged.passRetained(LifetimeTracked(i))
    } else {
      return nil
    }
  }

  withAUMP(initialValues: values) { p in
    for i in 0 ..< count {
      expectTrue(p[i] === values[i]?.takeUnretainedValue())
      expectTrue(p.advanced(by: i).pointee === values[i]?.takeUnretainedValue())
    }
  }

  expectEqual(LifetimeTracked.instances, originalInstances + refs)
  for unmanaged in values {
    unmanaged?.release()
  }
  expectEqual(LifetimeTracked.instances, originalInstances)
}

runAllTests()