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
|
/*
This source file is part of the Swift.org open source project
Copyright (c) 2022 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 XCTest
import Foundation
@testable import SymbolKit
class GraphCollectorTests: XCTestCase {
/// Verify that extension graphs are merged with the correct main graph depending on
/// the chosen strategy.
func testExtensionGraphMergingStrategies() throws {
let a = SymbolGraph(metadata: .init(formatVersion: .init(major: 1, minor: 0, patch: 0), generator: "unit-test"),
module: .init(name: "A", platform: .init()),
symbols: [
.init(identifier: .init(precise: "s:AA", interfaceLanguage: "swift"),
names: .init(title: "A", navigator: nil, subHeading: nil, prose: nil),
pathComponents: ["A"],
docComment: nil,
accessLevel: .init(rawValue: "public"),
kind: .init(parsedIdentifier: .class, displayName: "Class"),
mixins: [:])
],
relationships: [])
let b = SymbolGraph(metadata: .init(formatVersion: .init(major: 1, minor: 0, patch: 0), generator: "unit-test"),
module: .init(name: "B", platform: .init()),
symbols: [
.init(identifier: .init(precise: "s:BB", interfaceLanguage: "swift"),
names: .init(title: "B", navigator: nil, subHeading: nil, prose: nil),
pathComponents: ["B"],
docComment: nil,
accessLevel: .init(rawValue: "public"),
kind: .init(parsedIdentifier: .class, displayName: "Class"),
mixins: [:])
],
relationships: [])
let a_At_B = SymbolGraph(metadata: .init(formatVersion: .init(major: 1, minor: 0, patch: 0), generator: "unit-test"),
module: .init(name: "A", platform: .init()),
symbols: [
.init(identifier: .init(precise: "s:BBAatB", interfaceLanguage: "swift"),
names: .init(title: "AatB", navigator: nil, subHeading: nil, prose: nil),
pathComponents: ["B", "AatB"],
docComment: nil,
accessLevel: .init(rawValue: "public"),
kind: .init(parsedIdentifier: .class, displayName: "Class"),
mixins: [:])
],
relationships: [])
// test with default strategy (extension graphs get attached to extend**ed** graph
var collector = GraphCollector()
collector.mergeSymbolGraph(a, at: .init(fileURLWithPath: "A.symbols.json"))
collector.mergeSymbolGraph(b, at: .init(fileURLWithPath: "B.symbols.json"))
collector.mergeSymbolGraph(a_At_B, at: .init(fileURLWithPath: "A@B.symbols.json"))
var (unifiedGraphs, _) = collector.finishLoading()
XCTAssertEqual(unifiedGraphs.count, 2)
var graphA = try XCTUnwrap(unifiedGraphs["A"])
XCTAssertEqual(graphA.symbols.count, 1)
XCTAssert(graphA.symbols.keys.contains("s:AA"))
var graphB = try XCTUnwrap(unifiedGraphs["B"])
XCTAssertEqual(graphB.symbols.count, 2)
XCTAssert(graphB.symbols.keys.contains("s:BB"))
XCTAssert(graphB.symbols.keys.contains("s:BBAatB"))
// test with extendingGraph association strategy (extension graphs get attached to extend**ing** graph
collector = GraphCollector(extensionGraphAssociationStrategy: .extendingGraph)
collector.mergeSymbolGraph(a, at: .init(fileURLWithPath: "A.symbols.json"))
collector.mergeSymbolGraph(b, at: .init(fileURLWithPath: "B.symbols.json"))
collector.mergeSymbolGraph(a_At_B, at: .init(fileURLWithPath: "A@B.symbols.json"))
(unifiedGraphs, _) = collector.finishLoading()
XCTAssertEqual(unifiedGraphs.count, 2)
graphA = try XCTUnwrap(unifiedGraphs["A"])
XCTAssertEqual(graphA.symbols.count, 2)
XCTAssert(graphA.symbols.keys.contains("s:AA"))
XCTAssert(graphA.symbols.keys.contains("s:BBAatB"))
graphB = try XCTUnwrap(unifiedGraphs["B"])
XCTAssertEqual(graphB.symbols.count, 1)
XCTAssert(graphB.symbols.keys.contains("s:BB"))
}
}
|
