汉秋
cac22fcc56
为云端与素材管理页面接入首页路由，将共享云存储模型迁入 Assets 模块，并补充 API/ViewModel 测试。 Co-authored-by: Cursor <cursoragent@cursor.com>
365 lines
15 KiB
Swift
365 lines
15 KiB
Swift
//
|
||
// AssetsViewModelTests.swift
|
||
// suixinkanTests
|
||
//
|
||
// Created by Codex on 2026/6/23.
|
||
//
|
||
|
||
import XCTest
|
||
@testable import suixinkan
|
||
|
||
@MainActor
|
||
/// 资产 ViewModel 测试，覆盖云盘和素材管理核心业务逻辑。
|
||
final class AssetsViewModelTests: XCTestCase {
|
||
/// 测试云盘无上传权限时不会继续上传 OSS 或写入云盘。
|
||
func testCloudUploadStopsWhenPermissionDenied() async {
|
||
let api = MockAssetsService()
|
||
api.uploadPermission = false
|
||
api.uploadReason = "空间不足"
|
||
let uploader = MockAssetsUploader()
|
||
let viewModel = CloudStorageViewModel()
|
||
|
||
let success = await viewModel.upload(
|
||
localFiles: [CloudLocalUploadFile(data: Data([1]), fileName: "a.jpg", fileType: 2)],
|
||
scenicId: 9,
|
||
api: api,
|
||
uploadService: uploader,
|
||
transferStore: CloudTransferStore()
|
||
)
|
||
|
||
XCTAssertFalse(success)
|
||
XCTAssertEqual(viewModel.errorMessage, "空间不足")
|
||
XCTAssertEqual(uploader.uploadedFiles.count, 0)
|
||
XCTAssertEqual(api.cloudUploadRequests.count, 0)
|
||
}
|
||
|
||
/// 测试云盘上传会先上传 OSS，再登记云盘文件并刷新列表。
|
||
func testCloudUploadUploadsOSSBeforeCloudRecord() async {
|
||
let api = MockAssetsService()
|
||
let uploader = MockAssetsUploader()
|
||
let transferStore = CloudTransferStore()
|
||
let viewModel = CloudStorageViewModel()
|
||
|
||
let success = await viewModel.upload(
|
||
localFiles: [CloudLocalUploadFile(data: Data([1, 2]), fileName: "a.jpg", fileType: 2)],
|
||
scenicId: 9,
|
||
api: api,
|
||
uploadService: uploader,
|
||
transferStore: transferStore
|
||
)
|
||
|
||
XCTAssertTrue(success)
|
||
XCTAssertEqual(uploader.uploadedFiles.map(\.fileName), ["a.jpg"])
|
||
XCTAssertEqual(api.cloudUploadRequests.first?.fileUrl, "https://cdn.example.com/a.jpg")
|
||
XCTAssertEqual(transferStore.records.first?.status, .success)
|
||
}
|
||
|
||
/// 测试云盘分页、进入文件夹和返回文件夹。
|
||
func testCloudPaginationAndFolderNavigation() async {
|
||
let folder = CloudDriveFile(id: 2, parentFolderId: 0, fileUrl: "", name: "子目录", type: 99)
|
||
let api = MockAssetsService()
|
||
api.cloudPages = [
|
||
ListPayload(total: 2, list: [folder]),
|
||
ListPayload(total: 2, list: [CloudDriveFile(id: 3, parentFolderId: 0, fileUrl: "https://cdn/a.jpg", name: "a.jpg", type: 2)])
|
||
]
|
||
let viewModel = CloudStorageViewModel()
|
||
|
||
await viewModel.reload(api: api)
|
||
await viewModel.loadMore(api: api)
|
||
await viewModel.enterFolder(folder, api: api)
|
||
await viewModel.popToFolder(at: 0, api: api)
|
||
|
||
XCTAssertEqual(api.cloudListRequests.first?.parentFolderId, 0)
|
||
XCTAssertEqual(api.cloudListRequests[1].page, 2)
|
||
XCTAssertEqual(api.cloudListRequests[2].parentFolderId, 2)
|
||
XCTAssertEqual(viewModel.currentFolderId, 0)
|
||
}
|
||
|
||
/// 测试素材列表筛选和分页请求参数。
|
||
func testMediaListFilterAndPagination() async {
|
||
let api = MockAssetsService()
|
||
api.mediaListResponses = [
|
||
MediaLibraryListResponse.fixture(total: 2, ids: [1]),
|
||
MediaLibraryListResponse.fixture(total: 2, ids: [2])
|
||
]
|
||
let viewModel = MediaLibraryViewModel(kind: .material)
|
||
viewModel.keyword = " 航拍 "
|
||
viewModel.selectedAuditFilter = .approved
|
||
|
||
await viewModel.reload(api: api)
|
||
await viewModel.loadMore(api: api)
|
||
|
||
XCTAssertEqual(api.mediaListRequests.first?.keyword, "航拍")
|
||
XCTAssertEqual(api.mediaListRequests.first?.status, 1)
|
||
XCTAssertEqual(api.mediaListRequests.last?.page, 2)
|
||
XCTAssertEqual(viewModel.items.map(\.id), [1, 2])
|
||
}
|
||
|
||
/// 测试审核未通过素材禁止上下架。
|
||
func testRejectedMediaCannotToggleListing() async {
|
||
let api = MockAssetsService()
|
||
let viewModel = MediaLibraryViewModel()
|
||
let item = MediaLibraryItem.fixture(id: 1, status: 2, listingStatus: 0)
|
||
|
||
let success = await viewModel.toggleListing(item, api: api)
|
||
|
||
XCTAssertFalse(success)
|
||
XCTAssertEqual(api.mediaOperationRequests.count, 0)
|
||
XCTAssertEqual(viewModel.errorMessage, "审核通过后才能上下架")
|
||
}
|
||
|
||
/// 测试上传素材会先上传封面和媒体文件，再提交最终 OSS URL。
|
||
func testMediaUploadUploadsFilesBeforeSubmit() async {
|
||
let api = MockAssetsService()
|
||
let uploader = MockAssetsUploader()
|
||
let viewModel = MediaLibraryEditorViewModel()
|
||
viewModel.name = "航拍素材"
|
||
viewModel.selectedSpotId = 6
|
||
viewModel.tagsText = "航拍,夜景"
|
||
viewModel.setCover(MediaLocalUploadFile(data: Data([1]), fileName: "cover.jpg", fileType: 2, width: 100, height: 80))
|
||
viewModel.addMediaFiles([MediaLocalUploadFile(data: Data([2]), fileName: "a.jpg", fileType: 2, width: 120, height: 90)])
|
||
|
||
let success = await viewModel.submit(scenicId: 9, api: api, uploadService: uploader)
|
||
|
||
XCTAssertTrue(success)
|
||
XCTAssertEqual(uploader.uploadedFiles.map(\.fileName), ["cover.jpg", "a.jpg"])
|
||
XCTAssertEqual(api.mediaUploadRequests.first?.coverUrl, "https://cdn.example.com/cover.jpg")
|
||
XCTAssertEqual(api.mediaUploadRequests.first?.mediaList.first?.ossUrl, "https://cdn.example.com/a.jpg")
|
||
XCTAssertEqual(api.mediaUploadRequests.first?.materialTag, "航拍,夜景")
|
||
}
|
||
|
||
/// 测试素材上传失败时不会提交素材接口。
|
||
func testMediaUploadFailureDoesNotSubmit() async {
|
||
let api = MockAssetsService()
|
||
let uploader = MockAssetsUploader()
|
||
uploader.shouldFail = true
|
||
let viewModel = MediaLibraryEditorViewModel()
|
||
viewModel.name = "航拍素材"
|
||
viewModel.selectedSpotId = 6
|
||
viewModel.setCover(MediaLocalUploadFile(data: Data([1]), fileName: "cover.jpg", fileType: 2, width: 100, height: 80))
|
||
viewModel.addMediaFiles([MediaLocalUploadFile(data: Data([2]), fileName: "a.jpg", fileType: 2, width: 120, height: 90)])
|
||
|
||
let success = await viewModel.submit(scenicId: 9, api: api, uploadService: uploader)
|
||
|
||
XCTAssertFalse(success)
|
||
XCTAssertEqual(api.mediaUploadRequests.count, 0)
|
||
}
|
||
}
|
||
|
||
/// 资产服务测试替身，记录请求并返回预设数据。
|
||
@MainActor
|
||
private final class MockAssetsService: AssetsServing {
|
||
struct CloudListRequest: Equatable {
|
||
let parentFolderId: Int
|
||
let name: String
|
||
let type: Int
|
||
let orderBy: Int
|
||
let page: Int
|
||
let pageSize: Int
|
||
}
|
||
|
||
struct MediaListRequest: Equatable {
|
||
let keyword: String
|
||
let status: Int?
|
||
let page: Int
|
||
}
|
||
|
||
var cloudPages: [ListPayload<CloudDriveFile>] = [ListPayload(total: 0, list: [])]
|
||
var mediaListResponses: [MediaLibraryListResponse] = [MediaLibraryListResponse.fixture(total: 0, ids: [])]
|
||
var uploadPermission = true
|
||
var uploadReason = ""
|
||
var cloudListRequests: [CloudListRequest] = []
|
||
var cloudUploadRequests: [CloudFileUploadRequest] = []
|
||
var mediaListRequests: [MediaListRequest] = []
|
||
var mediaOperationRequests: [MediaAlbumOperationRequest] = []
|
||
var mediaUploadRequests: [MediaAlbumUploadRequest] = []
|
||
var mediaEditRequests: [MediaAlbumEditRequest] = []
|
||
|
||
/// 返回云盘文件列表。
|
||
func cloudFileList(parentFolderId: Int, name: String, type: Int, orderBy: Int, page: Int, pageSize: Int) async throws -> ListPayload<CloudDriveFile> {
|
||
cloudListRequests.append(CloudListRequest(parentFolderId: parentFolderId, name: name, type: type, orderBy: orderBy, page: page, pageSize: pageSize))
|
||
return cloudPages.isEmpty ? ListPayload(total: 0, list: []) : cloudPages.removeFirst()
|
||
}
|
||
|
||
/// 创建文件夹。
|
||
func cloudFolderCreate(_ request: CloudFolderCreateRequest) async throws {}
|
||
|
||
/// 登记云盘文件。
|
||
func cloudFileUpload(_ request: CloudFileUploadRequest) async throws {
|
||
cloudUploadRequests.append(request)
|
||
}
|
||
|
||
/// 删除云盘文件。
|
||
func cloudFileDelete(_ request: CloudFileDeleteRequest) async throws {}
|
||
|
||
/// 移动云盘文件。
|
||
func cloudFileMove(_ request: CloudFileMoveRequest) async throws {}
|
||
|
||
/// 重命名文件夹。
|
||
func cloudFolderEdit(_ request: CloudFolderModifyRequest) async throws {}
|
||
|
||
/// 重命名文件。
|
||
func cloudFileEdit(_ request: CloudFileModifyRequest) async throws {}
|
||
|
||
/// 检查上传权限。
|
||
func checkCloudUploadPermission() async throws -> CheckCloudUploadResponse {
|
||
try JSONDecoder().decode(CheckCloudUploadResponse.self, from: #"{"can_upload":\#(uploadPermission ? 1 : 0),"reason":"\#(uploadReason)"}"#.data(using: .utf8)!)
|
||
}
|
||
|
||
/// 返回素材列表。
|
||
func mediaAlbumList(kind: MediaLibraryKind, keyword: String, status: Int?, page: Int, pageSize: Int) async throws -> MediaLibraryListResponse {
|
||
mediaListRequests.append(MediaListRequest(keyword: keyword, status: status, page: page))
|
||
return mediaListResponses.isEmpty ? MediaLibraryListResponse.fixture(total: 0, ids: []) : mediaListResponses.removeFirst()
|
||
}
|
||
|
||
/// 返回素材详情。
|
||
func mediaAlbumDetail(id: Int) async throws -> MediaLibraryDetail {
|
||
MediaLibraryDetail.fixture(id: id)
|
||
}
|
||
|
||
/// 更新上下架。
|
||
func mediaAlbumOperation(_ request: MediaAlbumOperationRequest) async throws {
|
||
mediaOperationRequests.append(request)
|
||
}
|
||
|
||
/// 删除素材。
|
||
func mediaAlbumDelete(_ request: MediaAlbumDeleteRequest) async throws {}
|
||
|
||
/// 新增标签。
|
||
func mediaAlbumAddTag(_ request: MediaAlbumAddTagRequest) async throws {}
|
||
|
||
/// 上传素材。
|
||
func mediaAlbumUpload(_ request: MediaAlbumUploadRequest) async throws {
|
||
mediaUploadRequests.append(request)
|
||
}
|
||
|
||
/// 编辑素材。
|
||
func mediaAlbumEdit(_ request: MediaAlbumEditRequest) async throws {
|
||
mediaEditRequests.append(request)
|
||
}
|
||
}
|
||
|
||
/// OSS 上传测试替身，记录上传文件并返回固定 CDN URL。
|
||
@MainActor
|
||
private final class MockAssetsUploader: OSSUploadServing {
|
||
struct UploadedFile: Equatable {
|
||
let fileName: String
|
||
let fileType: Int
|
||
let scenicId: Int
|
||
}
|
||
|
||
var uploadedFiles: [UploadedFile] = []
|
||
var shouldFail = false
|
||
|
||
/// 上传用户头像。
|
||
func uploadUserAvatar(data: Data, fileName: String, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: 2, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传实名认证图片。
|
||
func uploadRealNameImage(data: Data, fileName: String, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: 2, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传云盘文件。
|
||
func uploadCloudFile(data: Data, fileName: String, fileType: Int, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: fileType, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传相册文件。
|
||
func uploadAlbumFile(data: Data, fileName: String, fileType: Int, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: fileType, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传任务附件。
|
||
func uploadTaskFile(data: Data, fileName: String, fileType: Int, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: fileType, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传打卡点图片。
|
||
func uploadPunchPointImage(data: Data, fileName: String, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: 2, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传景区申请图片。
|
||
func uploadScenicApplicationImage(data: Data, fileName: String, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: 2, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 上传银行卡图片。
|
||
func uploadBankCardImage(data: Data, fileName: String, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
try await upload(fileName: fileName, fileType: 2, scenicId: scenicId, onProgress: onProgress)
|
||
}
|
||
|
||
/// 通用测试上传流程。
|
||
private func upload(fileName: String, fileType: Int, scenicId: Int, onProgress: @escaping (Int) -> Void) async throws -> String {
|
||
if shouldFail {
|
||
throw APIError.networkFailed("上传失败")
|
||
}
|
||
uploadedFiles.append(UploadedFile(fileName: fileName, fileType: fileType, scenicId: scenicId))
|
||
onProgress(100)
|
||
return "https://cdn.example.com/\(fileName)"
|
||
}
|
||
}
|
||
|
||
private extension MediaLibraryListResponse {
|
||
/// 创建素材列表测试数据。
|
||
static func fixture(total: Int, ids: [Int]) -> MediaLibraryListResponse {
|
||
let list = ids.map { MediaLibraryItem.fixture(id: $0, status: 1, listingStatus: 0) }
|
||
let payload = [
|
||
"total": total,
|
||
"list": list.map { ["id": $0.id, "name": $0.name, "status": $0.status, "listing_status": $0.listingStatus] },
|
||
"order": ["total_num": 0, "avg_order_amount": 0, "refund_total": 0, "avg_change": 0]
|
||
] as [String: Any]
|
||
let data = try! JSONSerialization.data(withJSONObject: payload)
|
||
return try! JSONDecoder().decode(MediaLibraryListResponse.self, from: data)
|
||
}
|
||
}
|
||
|
||
private extension MediaLibraryItem {
|
||
/// 创建素材列表项测试数据。
|
||
static func fixture(id: Int, status: Int, listingStatus: Int) -> MediaLibraryItem {
|
||
let payload: [String: Any] = [
|
||
"id": id,
|
||
"name": "素材\(id)",
|
||
"cover_url": "",
|
||
"created_at": "",
|
||
"status": status,
|
||
"download_count": 0,
|
||
"likes_count": 0,
|
||
"collect_count": 0,
|
||
"share_count": 0,
|
||
"scenic_spot_name": "",
|
||
"project_name": "",
|
||
"listing_status": listingStatus
|
||
]
|
||
let data = try! JSONSerialization.data(withJSONObject: payload)
|
||
return try! JSONDecoder().decode(MediaLibraryItem.self, from: data)
|
||
}
|
||
}
|
||
|
||
private extension MediaLibraryDetail {
|
||
/// 创建素材详情测试数据。
|
||
static func fixture(id: Int) -> MediaLibraryDetail {
|
||
let payload: [String: Any] = [
|
||
"id": id,
|
||
"name": "素材\(id)",
|
||
"cover_url": "",
|
||
"description": "",
|
||
"status": 1,
|
||
"listing_status": 0,
|
||
"likes_count": 0,
|
||
"download_count": 0,
|
||
"collect_count": 0,
|
||
"share_count": 0,
|
||
"uploader_name": "",
|
||
"created_at": "",
|
||
"media_list": [],
|
||
"scenic_name": "",
|
||
"project_name": ""
|
||
]
|
||
let data = try! JSONSerialization.data(withJSONObject: payload)
|
||
return try! JSONDecoder().decode(MediaLibraryDetail.self, from: data)
|
||
}
|
||
}
|