// RUN: %empty-directory(%t)
// RUN: %target-clang -fobjc-arc %S/Inputs/objc_async.m -c -o %t/objc_async_objc.o
// RUN: %target-build-swift  -target %target-swift-5.1-abi-triple -target %target-swift-5.1-abi-triple -parse-as-library -module-name main -import-objc-header %S/Inputs/objc_async.h %s %t/objc_async_objc.o -o %t/objc_async
// RUN: %target-run %t/objc_async | %FileCheck %s

// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: objc_interop

// rdar://76038845
// REQUIRES: concurrency_runtime
// UNSUPPORTED: back_deployment_runtime

// Disable this test because it's flaky without a proper way to make the main
// Swift task await a background queue.
// REQUIRES: rdar77934626

func buttTest() async {
  let butt = Butt()
  let result = await butt.butt(1738)
  print("finishing \(result)")
}

func farmTest() async {
  let farm = Farm()
  let dogNumber = await farm.doggo
  print("dog number = \(dogNumber)")
  do {
    let _ = try await farm.catto
  } catch {
    print("caught exception")
  }
}

class Clbuttic: Butt {
    override func butt(_ x: Int) async -> Int {
        print("called into override")
        return 219
    }
}

class Buttertion: MutableButt_2Fast2Furious {
    override func butt(_ x: Int, completionHandler: @escaping (Int) -> Void) {
        print("called again into override")
        completionHandler(20721)
    }
}

@main struct Main {
  static func main() async {
    // CHECK: starting 1738
    // CHECK-NEXT: finishing 679
    await buttTest()

    // CHECK-NEXT: getting dog
    // CHECK-NEXT: dog number = 123
    // CHECK-NEXT: obtaining cat has failed!
    // CHECK-NEXT: caught exception
    await farmTest()

    // CHECK-NEXT: called into override
    // CHECK-NEXT: butt {{.*}} named clbuttic occurred at 219
    scheduleButt(Clbuttic(), "clbuttic")

    await Task.sleep(250_000)

    // CHECK-NEXT: called again into override
    // CHECK-NEXT: butt {{.*}} named buttertion occurred at 20721
    scheduleButt(Buttertion(), "buttertion")

    await Task.sleep(250_000)
  }
}