28 lines
764 B
Swift
28 lines
764 B
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
import XCTest
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
final class CancellableContinuationTest: XCTestCase {
|
|
func testNestedCancellation() async throws {
|
|
let cancellableTask = Task {
|
|
let cancellableContinuation = CancellableContinuation<Void>()
|
|
try await withTaskCancellationHandler(
|
|
operation: cancellableContinuation.wait,
|
|
onCancel: { /* do nothing */ }
|
|
)
|
|
}
|
|
cancellableTask.cancel()
|
|
do {
|
|
try await cancellableTask.value
|
|
XCTFail("Task must fail.")
|
|
} catch is CancellationError {
|
|
// this is fine
|
|
}
|
|
}
|
|
}
|