[Concurrency] Add test for objc type TaskExecutor

Author: ktoso

test/Concurrency/Runtime/async_task_executor_nsobject.swift

@@ -0,0 +1,49 @@
+// RUN: %target-run-simple-swift( -Xfrontend -disable-availability-checking %import-libdispatch -parse-as-library )
+
+// REQUIRES: executable_test
+// REQUIRES: concurrency
+// REQUIRES: libdispatch
+
+// REQUIRES: OS=macosx
+// REQUIRES: objc_interop
+
+// REQUIRES: concurrency_runtime
+// UNSUPPORTED: back_deployment_runtime
+
+import Dispatch
+import StdlibUnittest
+import _Concurrency
+
+import Foundation
+import Darwin
+
+// Sneaky trick to make this type objc reference counted.
+//
+// This test specifically checks that our reference counting accounts for existence of
+// objective-c types as TaskExecutors -- which was a bug where we'd swift_release
+// obj-c excecutors by accident (rdar://131151645).
+final class NSQueueTaskExecutor: NSData, TaskExecutor, @unchecked Sendable {
+  public func enqueue(_ _job: consuming ExecutorJob) {
+    let job = UnownedJob(_job)
+    DispatchQueue.main.async {
+      job.runSynchronously(on: self.asUnownedTaskExecutor())
+    }
+  }
+}
+
+@main struct Main {
+  static func main() async {
+    var taskExecutor: (any TaskExecutor)? = NSQueueTaskExecutor()
+
+    let task = Task(executorPreference: taskExecutor) {
+      dispatchPrecondition(condition: .onQueue(DispatchQueue.main))
+      try? await Task.sleep(for: .seconds(2))
+      return 12
+    }
+
+    taskExecutor = nil
+
+    let num = await task.value
+    assert(num == 12)
+  }
+}