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
|
/*
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
#if canImport(FoundationNetworking)
import FoundationNetworking
#endif
import XCTest
@testable import SwiftDocC
fileprivate let baseURL = URL(string: "test://")!
fileprivate let helloWorldHTML = "<html><header><title>Hello Title</title></header><body>Hello world</body></html>".data(using: .utf8)!
class DocumentationSchemeHandlerTests: XCTestCase {
let templateURL = Bundle.module.url(
forResource: "TestBundle", withExtension: "docc", subdirectory: "Test Bundles")!
func testDocumentationSchemeHandler() {
#if !os(Linux) && !os(Android) && !os(Windows)
let topicSchemeHandler = DocumentationSchemeHandler(withTemplateURL: templateURL)
let request = URLRequest(url: baseURL.appendingPathComponent("/images/figure1.jpg"))
var (response, data) = topicSchemeHandler.response(to: request)
XCTAssertNotNil(data)
XCTAssertEqual(response.mimeType, "image/jpeg")
let failingRequest = URLRequest(url: baseURL.appendingPathComponent("/not/found.jpg"))
(response, data) = topicSchemeHandler.response(to: failingRequest)
XCTAssertNil(data)
topicSchemeHandler.fallbackHandler = { (request: URLRequest) -> (URLResponse, Data)? in
guard let url = request.url else { return nil }
let response = URLResponse(url: url, mimeType: "text/html", expectedContentLength: helloWorldHTML.count, textEncodingName: nil)
return (response, helloWorldHTML)
}
(response, data) = topicSchemeHandler.response(to: failingRequest)
XCTAssertEqual(data, helloWorldHTML)
XCTAssertEqual(response.mimeType, "text/html")
#endif
}
func testSetData() {
#if !os(Linux) && !os(Android) && !os(Windows)
let topicSchemeHandler = DocumentationSchemeHandler(withTemplateURL: templateURL)
let data = "hello!".data(using: .utf8)!
topicSchemeHandler.setData(data: ["a.txt": data])
XCTAssertEqual(
topicSchemeHandler.response(
to: URLRequest(url: baseURL.appendingPathComponent("/data/a.txt"))
).1,
data
)
topicSchemeHandler.setData(data: ["b.txt": data])
XCTAssertEqual(
topicSchemeHandler.response(
to: URLRequest(url: baseURL.appendingPathComponent("/data/b.txt"))
).1,
data
)
XCTAssertNil(
topicSchemeHandler.response(
to: URLRequest(url: baseURL.appendingPathComponent("/data/a.txt"))
).1,
"a.txt should have been deleted because we set the daata to b.txt."
)
#endif
}
}
|
