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
121
122
123
124
125
126
|
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
#if !os(Windows)
import Synchronization
class TestURLSessionFTP : LoopbackFTPServerTest {
let saveString = """
FTP implementation to test FTP
upload, download and data tasks. Instead of sending a file,
we are sending the hardcoded data.We are going to test FTP
data, download and upload tasks with delegates & completion handlers.
Creating the data here as we need to pass the count
as part of the header.\r\n
"""
func test_ftpDataTask() throws {
throw XCTSkip("<https://bugs.swift.org/browse/SR-10922> non-deterministic SEGFAULT in TestURLSessionFTP.test_ftpDataTask")
#if false
#if !DARWIN_COMPATIBILITY_TESTS
let ftpURL = "ftp://127.0.0.1:\(TestURLSessionFTP.serverPort)/test.txt"
let req = URLRequest(url: URL(string: ftpURL)!)
let configuration = URLSessionConfiguration.default
let expect = expectation(description: "URL test with custom protocol")
let sesh = URLSession(configuration: configuration)
let dataTask1 = sesh.dataTask(with: req, completionHandler: { data, res, error in
defer { expect.fulfill() }
XCTAssertNil(error)
XCTAssertEqual(self.saveString, String(data: data!, encoding: String.Encoding.utf8))
})
dataTask1.resume()
waitForExpectations(timeout: 60)
#endif
#endif
}
func test_ftpDataTaskDelegate() throws {
throw XCTSkip("<https://bugs.swift.org/browse/SR-10922> non-deterministic SEGFAULT in TestURLSessionFTP.test_ftpDataTask")
#if false
let urlString = "ftp://127.0.0.1:\(TestURLSessionFTP.serverPort)/test.txt"
let url = URL(string: urlString)!
let dataTask = FTPDataTask(with: expectation(description: "data task"))
dataTask.run(with: url)
waitForExpectations(timeout: 60)
if !dataTask.error {
XCTAssertNotNil(dataTask.fileData)
}
#endif
}
}
// Sendable note: Access to ivars is essentially serialized by the XCTestExpectation. It would be better to do it with a lock, but this is sufficient for now.
class FTPDataTask : NSObject, @unchecked Sendable {
let dataTaskExpectation: XCTestExpectation!
var fileData: NSMutableData = NSMutableData()
var session: URLSession! = nil
var task: URLSessionDataTask! = nil
var cancelExpectation: XCTestExpectation?
var responseReceivedExpectation: XCTestExpectation?
var hasTransferCompleted = false
private let _error = Mutex(false)
public var error: Bool {
get { _error.withLock { $0 } }
set { _error.withLock { $0 = newValue } }
}
init(with expectation: XCTestExpectation) {
dataTaskExpectation = expectation
}
func run(with request: URLRequest) {
let config = URLSessionConfiguration.default
config.timeoutIntervalForRequest = 8
session = URLSession(configuration: config, delegate: self, delegateQueue: nil)
task = session.dataTask(with: request)
task.resume()
}
func run(with url: URL) {
let config = URLSessionConfiguration.default
config.timeoutIntervalForRequest = 8
session = URLSession(configuration: config, delegate: self, delegateQueue: nil)
task = session.dataTask(with: url)
task.resume()
}
func cancel() {
task.cancel()
}
}
extension FTPDataTask : URLSessionDataDelegate {
public func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {
fileData.append(data)
responseReceivedExpectation?.fulfill()
}
public func urlSession(_ session: URLSession,
dataTask: URLSessionDataTask,
didReceive response: URLResponse,
completionHandler: @escaping (URLSession.ResponseDisposition) -> Void) {
guard responseReceivedExpectation != nil else { return }
responseReceivedExpectation!.fulfill()
}
}
extension FTPDataTask : URLSessionTaskDelegate {
public func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?) {
dataTaskExpectation.fulfill()
guard (error as? URLError) != nil else { return }
if let cancellation = cancelExpectation {
cancellation.fulfill()
}
self.error = true
}
}
#endif
|
