82 lines
2.3 KiB
Swift
82 lines
2.3 KiB
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
public import SignalServiceKit
|
|
public import XCTest
|
|
|
|
extension Thenable {
|
|
func expect(timeout: TimeInterval, file: StaticString = #filePath, line: UInt = #line) -> Value {
|
|
let expectation = XCTestExpectation(description: "\(file):\(line)")
|
|
var result: Value?
|
|
self.done {
|
|
result = $0
|
|
expectation.fulfill()
|
|
}.catch {
|
|
XCTFail("\($0)", file: file, line: line)
|
|
}
|
|
_ = XCTWaiter.wait(for: [expectation], timeout: timeout)
|
|
return try! XCTUnwrap(result, file: file, line: line)
|
|
}
|
|
}
|
|
|
|
extension XCTestCase {
|
|
@discardableResult
|
|
public func expect<T>(
|
|
_ promise: Promise<T>,
|
|
description: String,
|
|
timeout: TimeInterval
|
|
) -> Result<T, Error> {
|
|
let expectation = self.expectation(description: description)
|
|
var result: Result<T, Error>!
|
|
promise.observe {
|
|
result = $0
|
|
expectation.fulfill()
|
|
}
|
|
wait(for: [expectation], timeout: timeout)
|
|
return result
|
|
}
|
|
|
|
@discardableResult
|
|
public func expectSuccess<T>(
|
|
_ promise: Promise<T>,
|
|
description: String,
|
|
timeout: TimeInterval
|
|
) -> T {
|
|
let expectation = self.expectation(description: description)
|
|
var result: T!
|
|
promise.observe {
|
|
switch $0 {
|
|
case .success(let v):
|
|
result = v
|
|
case .failure(let e):
|
|
XCTFail("Expected success, got error: \(e)")
|
|
}
|
|
expectation.fulfill()
|
|
}
|
|
wait(for: [expectation], timeout: timeout)
|
|
return result
|
|
}
|
|
|
|
@discardableResult
|
|
public func expectFailure<T>(
|
|
_ promise: Promise<T>,
|
|
description: String,
|
|
timeout: TimeInterval
|
|
) -> Error {
|
|
let expectation = self.expectation(description: description)
|
|
var result: Error!
|
|
promise.observe {
|
|
switch $0 {
|
|
case .success(let v):
|
|
XCTFail("Expected failure, got success: \(v)")
|
|
case .failure(let e):
|
|
result = e
|
|
}
|
|
expectation.fulfill()
|
|
}
|
|
wait(for: [expectation], timeout: timeout)
|
|
return result
|
|
}
|
|
}
|