Swift网络请求函数在Playground可用但Xcode测试失效的解决方法

你的函数核心问题在于**URLSession.dataTask是异步执行的**：函数调用后会立刻启动网络任务，但不会等任务完成就直接返回初始值为nil的htmlString。

Playground有特殊机制，会自动等待异步任务执行完毕，所以看起来能拿到结果；但Xcode测试环境不会自动等待，导致网络请求的闭包还没执行，函数就已经返回了，自然看不到闭包里的逻辑执行。

下面是两种标准修复方案：

方案1：使用回调闭包（兼容旧版本Swift） #

把函数改成通过闭包异步返回结果，避免同步返回未完成的值：

private func fetch(url: URL, completion: @escaping (String?) -> Void) {
    let task = URLSession.shared.dataTask(with: url) { data, response, error in
        guard error == nil, let httpResponse = response as? HTTPURLResponse,
              (200...299).contains(httpResponse.statusCode) else {
            completion(nil)
            return
        }
        if let mimeType = httpResponse.mimeType, mimeType == "text/html",
           let data = data,
           let string = String(data: data, encoding: .utf8) {
            completion(string)
        } else {
            completion(nil)
        }
    }
    task.resume()
}

测试用例写法： #

测试时需要用XCTestExpectation等待异步任务完成：

func testFetchHTML() {
    let expectation = XCTestExpectation(description: "获取HTML内容")
    guard let testUrl = URL(string: "https://example.com") else {
        XCTFail("无效的测试URL")
        return
    }
    
    fetch(url: testUrl) { htmlContent in
        XCTAssertNotNil(htmlContent, "HTML内容不应为空")
        expectation.fulfill()
    }
    
    wait(for: [expectation], timeout: 10.0)
}

方案2：使用Async/Await（Swift 5.5+推荐） #

利用Swift的异步语法，让代码更简洁易读，同时天然支持异步测试：

private func fetch(url: URL) async -> String? {
    do {
        let (data, response) = try await URLSession.shared.data(from: url)
        guard let httpResponse = response as? HTTPURLResponse,
              (200...299).contains(httpResponse.statusCode) else {
            return nil
        }
        if let mimeType = httpResponse.mimeType, mimeType == "text/html",
           let string = String(data: data, encoding: .utf8) {
            return string
        }
        return nil
    } catch {
        return nil
    }
}

测试用例写法： #

直接把测试函数标记为async，用await等待结果：

func testFetchHTML() async {
    guard let testUrl = URL(string: "https://example.com") else {
        XCTFail("无效的测试URL")
        return
    }
    
    let htmlContent = await fetch(url: testUrl)
    XCTAssertNotNil(htmlContent, "HTML内容不应为空")
}

内容的提问来源于stack exchange，提问作者Priyatham