Files
container/Tests/ContainerizationExtrasTests/TestTimeout.swift
T
Danny Canter 259c841591 ContainerizationExtras: Timeout adjustments (#254)
Three things:

1. Timeout.run today didn't allow a throwing closure.
2. Add a new constructor for Timeout that takes in a Duration directly
so we can be more granular than seconds.
3. Add unit tests because we all love those :)
2025-08-07 14:34:45 -04:00

58 lines
1.8 KiB
Swift

//===----------------------------------------------------------------------===//
// Copyright © 2025 Apple Inc. and the Containerization project authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//===----------------------------------------------------------------------===//
import Foundation
import Testing
@testable import ContainerizationExtras
final class TestTimeoutType {
@Test
func testNoCancellation() async throws {
await #expect(throws: Never.self) {
try await Timeout.run(
for: .seconds(5),
operation: {
return
})
}
}
@Test
func testCancellationError() async throws {
await #expect(throws: CancellationError.self) {
try await Timeout.run(
for: .milliseconds(50),
operation: {
try await Task.sleep(for: .seconds(2))
})
}
}
@Test
func testClosureError() async throws {
// Check that we get the closures error if we don't timeout, but
// the closure does throw before.
await #expect(throws: POSIXError.self) {
try await Timeout.run(
for: .seconds(10),
operation: {
throw POSIXError(.E2BIG)
})
}
}
}