Replace MockPackageRegistryURLProtocol with HTTPClient handlers

Author: mattt

Diff for: Sources/SPMTestSupport/MockPackageRegistryURLProtocol.swift

@@ -33,10 +33,87 @@ final class RegistryManagerTests: XCTestCase {
             return MockArchiver()
         }
 
-        RegistryManager.clientFactory = { _ in
-            let configuration = URLSessionConfiguration.default
-            configuration.protocolClasses = [MockPackageRegistryURLProtocol.self] + (configuration.protocolClasses ?? [])
-            return URLSessionHTTPClient(configuration: configuration)
+        RegistryManager.clientFactory = { diagnosticsEngine in
+            let configuration = HTTPClientConfiguration()
+            let handler: HTTPClient.Handler = { request, completion in
+                var headers: HTTPClientHeaders = [
+                    "Content-Version": "1",
+                ]
+
+                let result: Result<HTTPClientResponse, Error>
+                switch (request.method, request.url.absoluteString.lowercased()) {
+                case (.head, "https://github.com/mona/linkedlist"):
+                    headers.add(name: "Location", value: "https://pkg.swift.github.com/github.com/mona/LinkedList")
+
+                    result = .success(HTTPClientResponse(statusCode: 303, headers: headers, body: nil))
+                case (.get, "https://pkg.swift.github.com/github.com/mona/linkedlist"):
+                    let data = #"""
+                    {
+                        "releases": {
+                            "1.1.1": {
+                                "url": "https://packages.example.com/mona/LinkedList/1.1.1"
+                            },
+                            "1.1.0": {
+                                "url": "https://packages.example.com/mona/LinkedList/1.1.0",
+                                "problem": {
+                                    "status": 410,
+                                    "title": "Gone",
+                                    "detail": "this release was removed from the registry"
+                                }
+                            },
+                            "1.0.0": {
+                                "url": "https://packages.example.com/mona/LinkedList/1.0.0"
+                            }
+                        }
+                    }
+
+                    """#.data(using: .utf8)!
+
+                    headers.add(name: "Content-Type", value: "application/json")
+                    headers.add(name: "Content-Length", value: "\(data.count)")
+
+                    result = .success(HTTPClientResponse(statusCode: 200, headers: headers, body: data))
+                case (.get, "https://pkg.swift.github.com/github.com/mona/linkedlist/1.1.1/package.swift"):
+                    let data = #"""
+                    // swift-tools-version:5.0
+                    import PackageDescription
+
+                    let package = Package(
+                        name: "LinkedList",
+                        products: [
+                            .library(name: "LinkedList", targets: ["LinkedList"])
+                        ],
+                        targets: [
+                            .target(name: "LinkedList"),
+                            .testTarget(name: "LinkedListTests", dependencies: ["LinkedList"]),
+                        ],
+                        swiftLanguageVersions: [.v4, .v5]
+                    )
+
+                    """#.data(using: .utf8)!
+
+                    headers.add(name: "Content-Type", value: "text/x-swift")
+                    headers.add(name: "Content-Disposition", value: #"attachment; filename="Package.swift""#)
+                    headers.add(name: "Content-Length", value: "\(data.count)")
+
+                    result = .success(HTTPClientResponse(statusCode: 200, headers: headers, body: data))
+                case (.get, "https://pkg.swift.github.com/github.com/mona/linkedlist/1.1.1.zip"):
+                    let archive = emptyZipFile
+
+                    headers.add(name: "Content-Type", value: "application/zip")
+                    headers.add(name: "Content-Disposition", value: #"attachment; filename="LinkedList-1.1.1.zip""#)
+                    headers.add(name: "Content-Length", value: "22")
+                    headers.add(name: "Digest", value: "sha-256=bc6c9a5d2f2226cfa1ef4fad8344b10e1cc2e82960f468f70d9ed696d26b3283")
+
+                    result = .success(HTTPClientResponse(statusCode: 200, headers: headers, body: Data(archive.contents)))
+                default:
+                    result = .failure(StringError("Unhandled request: \(request)"))
+                }
+
+                completion(result)
+            }
+
+            return HTTPClient(configuration: configuration, handler: handler, diagnosticsEngine: diagnosticsEngine)
         }
 
         super.setUp()