feat(models): 模型自动下载(我的·模型管理) + 断点续传 + 旁路导入
实现 spec(2026-05-29-model-download-design)的模型分发功能: - ModelManifest: 硬编码功能文件清单 + base URL https://file.myv0.com/ - FileDownloader: URLSessionDataDelegate 分块写盘,HTTP Range 断点续传 + 大小校验 (根因修复:URL.resourceValues 会缓存文件大小,续传时先读 offset 再读 finalSize 会拿到下载前的陈旧值导致校验误判;改用 FileManager.attributesOfItem) - ModelDownloadService: @MainActor @Observable 编排逐文件下载,聚合进度/速度, 支持下载全部/暂停/重试,以及旁路文件导入 - ModelStore: 新增 fileURL/localBytes/isComplete(可注入清单)/importModel(补 VL) - ModelManagementView: 分模型卡片(状态/进度/速度) + 下载全部/暂停 + NWPathMonitor 蜂窝提示 + 从文件导入(离线兜底) - MeView: 模型管理卡改 NavigationLink + 动态状态(已就绪/下载中/N就绪) 测试(Swift Testing): Manifest 清单/字节数、Store 路径/校验/导入、 DownloadState、FileDownloader(URLProtocol mock:下载/Range续传/大小校验) Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
136
康康Tests/ModelDownloadCoreTests.swift
Normal file
136
康康Tests/ModelDownloadCoreTests.swift
Normal file
@@ -0,0 +1,136 @@
|
||||
import Testing
|
||||
import Foundation
|
||||
@testable import 康康
|
||||
|
||||
// MARK: - Mock 网络层
|
||||
|
||||
/// 按 URL 注册完整响应体,startLoading 时按请求的 Range header 自动切片返回(206)或全量(200)。
|
||||
/// 每个测试用唯一 URL 注册自己的内容 → 测试间不会互相覆盖,无需依赖执行顺序或可见性。
|
||||
final class MockURLProtocol: URLProtocol, @unchecked Sendable {
|
||||
private static let lock = NSLock()
|
||||
private static var bodies: [String: Data] = [:]
|
||||
|
||||
static func register(_ url: URL, body: Data) {
|
||||
lock.lock(); defer { lock.unlock() }
|
||||
bodies[url.path] = body
|
||||
}
|
||||
static func reset() {
|
||||
lock.lock(); defer { lock.unlock() }
|
||||
bodies.removeAll()
|
||||
}
|
||||
private static func body(forPath path: String) -> Data? {
|
||||
lock.lock(); defer { lock.unlock() }
|
||||
return bodies[path]
|
||||
}
|
||||
|
||||
override class func canInit(with request: URLRequest) -> Bool { true }
|
||||
override class func canonicalRequest(for request: URLRequest) -> URLRequest { request }
|
||||
|
||||
override func startLoading() {
|
||||
guard let url = request.url, let full = Self.body(forPath: url.path) else {
|
||||
client?.urlProtocol(self, didFailWithError: URLError(.fileDoesNotExist))
|
||||
return
|
||||
}
|
||||
var data = full
|
||||
var status = 200
|
||||
var headers: [String: String] = [:]
|
||||
if let range = request.value(forHTTPHeaderField: "Range"),
|
||||
let start = Self.parseRangeStart(range), start <= full.count {
|
||||
data = Data(full.suffix(from: start))
|
||||
status = 206
|
||||
headers["Content-Range"] = "bytes \(start)-\(full.count - 1)/\(full.count)"
|
||||
}
|
||||
let response = HTTPURLResponse(
|
||||
url: url, statusCode: status, httpVersion: nil, headerFields: headers)!
|
||||
client?.urlProtocol(self, didReceive: response, cacheStoragePolicy: .notAllowed)
|
||||
client?.urlProtocol(self, didLoad: data)
|
||||
client?.urlProtocolDidFinishLoading(self)
|
||||
}
|
||||
|
||||
override func stopLoading() {}
|
||||
|
||||
/// "bytes=2-" → 2
|
||||
private static func parseRangeStart(_ s: String) -> Int? {
|
||||
guard let eq = s.firstIndex(of: "="), let dash = s.firstIndex(of: "-") else { return nil }
|
||||
return Int(s[s.index(after: eq)..<dash])
|
||||
}
|
||||
}
|
||||
|
||||
private func mockConfiguration() -> URLSessionConfiguration {
|
||||
let config = URLSessionConfiguration.ephemeral
|
||||
config.protocolClasses = [MockURLProtocol.self]
|
||||
return config
|
||||
}
|
||||
|
||||
private func tempFile() -> URL {
|
||||
FileManager.default.temporaryDirectory
|
||||
.appendingPathComponent(UUID().uuidString, isDirectory: true)
|
||||
.appendingPathComponent("a.bin")
|
||||
}
|
||||
|
||||
private func uniqueURL() -> URL {
|
||||
URL(string: "https://mock.test/\(UUID().uuidString).bin")!
|
||||
}
|
||||
|
||||
// MARK: - DownloadState
|
||||
|
||||
struct DownloadStateTests {
|
||||
@Test func fractionZeroWhenTotalZero() {
|
||||
let s = DownloadState(phase: .idle, receivedBytes: 0, totalBytes: 0, bytesPerSecond: 0)
|
||||
#expect(s.fraction == 0)
|
||||
}
|
||||
|
||||
@Test func fractionComputed() {
|
||||
let s = DownloadState(phase: .downloading, receivedBytes: 50, totalBytes: 200, bytesPerSecond: 0)
|
||||
#expect(s.fraction == 0.25)
|
||||
}
|
||||
}
|
||||
|
||||
// MARK: - FileDownloader
|
||||
|
||||
/// 串行执行:这些测试共享全局 URLProtocol / URLSession 状态,并行会互相干扰。
|
||||
@Suite(.serialized)
|
||||
struct FileDownloaderTests {
|
||||
|
||||
@Test func downloadsFileContent() async throws {
|
||||
let url = uniqueURL()
|
||||
MockURLProtocol.register(url, body: Data("hello".utf8))
|
||||
let dst = tempFile()
|
||||
defer { try? FileManager.default.removeItem(at: dst.deletingLastPathComponent()) }
|
||||
|
||||
let dl = FileDownloader(configuration: mockConfiguration())
|
||||
try await dl.download(from: url, to: dst, expectedBytes: 5)
|
||||
|
||||
#expect(try Data(contentsOf: dst) == Data("hello".utf8))
|
||||
#expect(!FileManager.default.fileExists(atPath: dst.appendingPathExtension("part").path))
|
||||
}
|
||||
|
||||
@Test func resumesFromPartialFile() async throws {
|
||||
let url = uniqueURL()
|
||||
MockURLProtocol.register(url, body: Data("hello".utf8))
|
||||
let dst = tempFile()
|
||||
defer { try? FileManager.default.removeItem(at: dst.deletingLastPathComponent()) }
|
||||
// 预置已下载的一半,download 应从 offset 2 续传
|
||||
try FileManager.default.createDirectory(
|
||||
at: dst.deletingLastPathComponent(), withIntermediateDirectories: true)
|
||||
try Data("he".utf8).write(to: dst.appendingPathExtension("part"))
|
||||
|
||||
let dl = FileDownloader(configuration: mockConfiguration())
|
||||
try await dl.download(from: url, to: dst, expectedBytes: 5)
|
||||
|
||||
#expect(try Data(contentsOf: dst) == Data("hello".utf8))
|
||||
}
|
||||
|
||||
@Test func throwsOnSizeMismatch() async throws {
|
||||
let url = uniqueURL()
|
||||
MockURLProtocol.register(url, body: Data("hi".utf8)) // 仅 2 字节,期望 5
|
||||
let dst = tempFile()
|
||||
defer { try? FileManager.default.removeItem(at: dst.deletingLastPathComponent()) }
|
||||
|
||||
let dl = FileDownloader(configuration: mockConfiguration())
|
||||
await #expect(throws: (any Error).self) {
|
||||
try await dl.download(from: url, to: dst, expectedBytes: 5)
|
||||
}
|
||||
#expect(!FileManager.default.fileExists(atPath: dst.path))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user