1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/*
This source file is part of the Swift.org open source project
Copyright (c) 2021 Apple Inc. and the Swift project authors
Licensed under Apache License v2.0 with Runtime Library Exception
See https://swift.org/LICENSE.txt for license information
See https://swift.org/CONTRIBUTORS.txt for Swift project authors
*/
import Foundation
import XCTest
/// Asynchronous expectation that works on Linux.
class AsynchronousExpectation {
enum Result {
case success
case timedOut
static func fromDispatchResult(_ result: DispatchTimeoutResult) -> Result {
switch result {
case .success: return .success
case .timedOut: return .timedOut
}
}
}
let description: String
private(set) var waitGroup: DispatchGroup? = DispatchGroup()
private let groupLock = NSLock()
init(description: String) {
self.description = description
}
func fulfill() {
groupLock.lock()
defer { groupLock.unlock() }
waitGroup?.leave()
waitGroup = nil
}
func wait(timeout: TimeInterval) -> Result {
groupLock.lock()
precondition(waitGroup != nil)
waitGroup!.enter()
return DispatchQueue.global().sync { [group = self.waitGroup!] in
groupLock.unlock()
let waitResult = group.wait(timeout: .now() + timeout)
return Result.fromDispatchResult(waitResult)
}
}
}
class AsynchronousExpectationTests: XCTestCase {
func testExpectationTimesOut() throws {
let expectation1 = AsynchronousExpectation(description: "expectation")
let result = expectation1.wait(timeout: 2)
XCTAssertEqual(result, .timedOut)
}
func testExpectationFulfills() throws {
let expectation1 = AsynchronousExpectation(description: "expectation")
DispatchQueue.global(qos: .utility).asyncAfter(deadline: .now() + 0.5) {
expectation1.fulfill()
}
let result = expectation1.wait(timeout: 2)
XCTAssertEqual(result, .success)
}
}
|
