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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2023 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
//
extension Test.ID {
/// A selection of ``Test/ID`` elements.
///
/// This internally stores the test IDs using a `Graph` for more efficient
/// storage and membership querying.
///
/// Note that this type represents a selection in the abstract, and may
/// represent tests which have been selected to be run or be skipped.
struct Selection: Sendable {
/// A graph representing the test IDs in the selection.
///
/// If a child node has value `true`, the key path of that node represents a
/// test ID which was explicitly added. Child nodes with value `false` are
/// intermediary values.
///
/// A test ID's membership in the selection may be queried using one of the
/// `contains(_:)` methods.
private var _testIDsGraph = Graph<String, Bool>(value: false)
/// Initialize an instance of this type with the specified test IDs.
///
/// - Parameters:
/// - testIDs: The test IDs to include in the selection.
init(testIDs: some Collection<Test.ID>) {
_testIDsGraph = .init(value: false)
for testID in testIDs {
_testIDsGraph.insertValue(true, at: testID.keyPathRepresentation, intermediateValue: false)
}
}
}
}
// MARK: - Querying membership
extension Test.ID.Selection {
/// Determine if the selection contains the ID for the specified test.
///
/// - Parameters:
/// - test: The test whose ID should be queried.
/// - inferAncestors: Whether or not to infer inclusion for the ancestors
/// of explicitly included tests. If `false`, only explicitly included
/// tests and their descendants are included.
///
/// - Returns: Whether or not the selection contains the ID for the
/// specified test.
///
/// A test ID is considered contained in the selection if it has been
/// explicitly added or if it has a descendant or ancestor which has been
/// explicitly added.
func contains(_ test: Test, inferAncestors: Bool = true) -> Bool {
contains(test.id, inferAncestors: inferAncestors)
}
/// Determine if the selection contains a specified test ID.
///
/// - Parameters:
/// - testID: The test ID to query.
/// - inferAncestors: Whether or not to infer inclusion for the ancestors
/// of explicitly included tests. If `false`, only explicitly included
/// tests and their descendants are included.
///
/// - Returns: Whether or not the selection contains the specified test ID.
///
/// A test ID is considered contained in the selection if it has been
/// explicitly added or if it has a descendant or ancestor which has been
/// explicitly added.
func contains(_ testID: Test.ID, inferAncestors: Bool = true) -> Bool {
contains(testID.keyPathRepresentation, inferAncestors: inferAncestors)
}
/// Determine if the selection contains a test ID with the specified fully-
/// qualified name components.
///
/// - Parameters:
/// - fullyQualifiedNameComponents: The fully-qualified name components of
/// the test ID to query.
/// - inferAncestors: Whether or not to infer inclusion for the ancestors
/// of explicitly included tests. If `false`, only explicitly included
/// tests and their descendants are included.
///
/// - Returns: Whether or not the selection contains a test ID with the
/// specified fully-qualified name components.
///
/// A test ID is considered contained in the selection if it has been
/// explicitly added or if it has a descendant or ancestor which has been
/// explicitly added.
func contains(_ fullyQualifiedNameComponents: some Collection<String>, inferAncestors: Bool = true) -> Bool {
let values = _testIDsGraph.takeValues(at: fullyQualifiedNameComponents)
if inferAncestors {
var isContained = false
for value in values {
switch value {
case .some(false):
isContained = true
case .some(true):
return true
case nil:
return false
}
}
return isContained
} else {
// If ancestors aren't inferred to be contained in this selection, then
// we are only looking for any element that is explicitly included; if
// an ancestral test ID was passed, it won't have enough elements to reach
// the explicitly-included test ID when calling takeValues(at:).
return values.contains(true)
}
}
}
|
