368 lines
20 KiB
Swift
368 lines
20 KiB
Swift
//
|
|
// TravelAlbumAPITests.swift
|
|
// suixinkanTests
|
|
//
|
|
|
|
import XCTest
|
|
@testable import suixinkan
|
|
|
|
@MainActor
|
|
/// 旅拍相册 API 测试。
|
|
final class TravelAlbumAPITests: XCTestCase {
|
|
func testListBuildsPathAndQuery() async throws {
|
|
let data = envelopeJSON(#"{"total":1,"list":[]}"#)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
_ = try await api.list(page: 2, pageSize: 30)
|
|
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/list")
|
|
let query = try XCTUnwrap(URLComponents(url: try XCTUnwrap(request.url), resolvingAgainstBaseURL: false)?.queryItems)
|
|
XCTAssertEqual(query.first { $0.name == "page" }?.value, "2")
|
|
XCTAssertEqual(query.first { $0.name == "page_size" }?.value, "30")
|
|
}
|
|
|
|
func testCreateEncodesBody() async throws {
|
|
let data = envelopeJSON(#"{"id":8}"#)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
_ = try await api.create(
|
|
TravelAlbumCreateRequest(
|
|
name: "2026-07-07-001",
|
|
type: 1,
|
|
orderNumber: nil,
|
|
materialNum: 2,
|
|
materialPrice: 10.5,
|
|
materialPackagePrice: 88,
|
|
photoPrice: 0
|
|
)
|
|
)
|
|
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "POST")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/create")
|
|
let body = try JSONSerialization.jsonObject(with: try XCTUnwrap(request.httpBody)) as? [String: Any]
|
|
XCTAssertEqual(body?["name"] as? String, "2026-07-07-001")
|
|
XCTAssertEqual(body?["type"] as? Int, 1)
|
|
XCTAssertEqual(body?["material_num"] as? Int, 2)
|
|
XCTAssertEqual(body?["material_price"] as? Double, 10.5)
|
|
}
|
|
|
|
func testMaterialListAndDeleteAndMpCode() async throws {
|
|
let materialList = envelopeJSON(
|
|
#"{"total":1,"list":[{"id":6,"user_equity_travel_id":3,"status":1,"order_number":"","user_id":9,"file_name":"A.JPG","file_type":2,"file_url":"https://cdn/a.jpg","file_size":1024,"cover_url":"","is_purchased":false,"ai_retouch_status":3,"ai_retouch_status_name":"AI已修","ai_refined_url":"https://cdn/refined.jpg","ai_atmosphere_url":"https://cdn/atmosphere.jpg","created_at":"","updated_at":""}]}"#
|
|
)
|
|
let empty = envelopeJSON("{}")
|
|
let code = envelopeJSON(#"{"mp_code_oss_url":"https://cdn/qr.png"}"#)
|
|
let session = MockURLSession(responses: [materialList, empty, code])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let materials = try await api.materialList(
|
|
userEquityTravelId: 3,
|
|
page: 1,
|
|
pageSize: 30,
|
|
orderBy: 4,
|
|
isPurchased: 1
|
|
)
|
|
try await api.deleteAlbum(id: 3)
|
|
let response = try await api.mpCode(id: 3)
|
|
|
|
XCTAssertEqual(materials.list.first?.aiRetouchStatus, 3)
|
|
XCTAssertEqual(materials.list.first?.aiRefinedURL, "https://cdn/refined.jpg")
|
|
XCTAssertEqual(materials.list.first?.aiAtmosphereURL, "https://cdn/atmosphere.jpg")
|
|
XCTAssertEqual(response.mpCodeOssUrl, "https://cdn/qr.png")
|
|
XCTAssertEqual(session.requests[0].url?.path, "/api/yf-handset-app/photog/travel-album/material-list")
|
|
let query = URLComponents(url: session.requests[0].url!, resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "user_equity_travel_id" }?.value, "3")
|
|
XCTAssertEqual(query?.first { $0.name == "order_by" }?.value, "4")
|
|
XCTAssertEqual(query?.first { $0.name == "is_purchased" }?.value, "1")
|
|
XCTAssertEqual(session.requests[1].url?.path, "/api/yf-handset-app/photog/travel-album/delete")
|
|
let body = try JSONSerialization.jsonObject(with: session.requests[1].httpBody!) as? [String: Any]
|
|
XCTAssertEqual(body?["id"] as? Int, 3)
|
|
XCTAssertEqual(session.requests[2].url?.path, "/api/yf-handset-app/photog/travel-album/mp-code")
|
|
}
|
|
|
|
func testMaterialInfoBuildsQueryAndDecodesMaterial() async throws {
|
|
let data = envelopeJSON(
|
|
#"{"id":6,"user_equity_travel_id":3,"status":1,"order_number":"","user_id":9,"file_name":"A.JPG","file_type":2,"file_url":"https://cdn/a.jpg","file_size":1024,"cover_url":"","is_purchased":false,"ai_retouch_status":3,"ai_retouch_status_name":"AI已修","ai_retouch_batch_id":60,"ai_refined_url":"https://cdn/refined.jpg","ai_atmosphere_url":"https://cdn/atmosphere.jpg","created_at":"","updated_at":""}"#
|
|
)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let material = try await api.materialInfo(userEquityTravelId: 3, materialId: 6)
|
|
|
|
XCTAssertEqual(material.id, 6)
|
|
XCTAssertEqual(material.aiRetouchBatchId, 60)
|
|
XCTAssertEqual(material.aiRefinedURL, "https://cdn/refined.jpg")
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "GET")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/material-info")
|
|
let query = URLComponents(url: try XCTUnwrap(request.url), resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "user_equity_travel_id" }?.value, "3")
|
|
XCTAssertEqual(query?.first { $0.name == "material_id" }?.value, "6")
|
|
}
|
|
|
|
func testBatchDeleteMaterialsBuildsPathAndBody() async throws {
|
|
let session = MockURLSession(responses: [envelopeJSON("{}")])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
try await api.batchDeleteMaterials(ids: [2031, 2032, 2033])
|
|
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "POST")
|
|
XCTAssertEqual(
|
|
request.url?.path,
|
|
"/api/yf-handset-app/photog/travel-album/batch-delete-material"
|
|
)
|
|
let body = try JSONSerialization.jsonObject(with: try XCTUnwrap(request.httpBody)) as? [String: Any]
|
|
XCTAssertEqual(body?["ids"] as? [Int], [2031, 2032, 2033])
|
|
}
|
|
|
|
func testUploadMaterialBuildsPathAndBody() async throws {
|
|
let material = envelopeJSON(#"{"id":6,"user_equity_travel_id":3,"status":1,"order_number":"","user_id":9,"file_name":"A.JPG","file_type":1,"file_url":"https://cdn/a.jpg","file_size":1024,"cover_url":"","is_purchased":false,"created_at":"2026-07-08 10:00:00","updated_at":"2026-07-08 10:00:00"}"#)
|
|
let session = MockURLSession(responses: [material])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let response = try await api.uploadMaterial(
|
|
TravelAlbumUploadMaterialRequest(
|
|
userEquityTravelId: 3,
|
|
fileName: "A.JPG",
|
|
fileUrl: "https://cdn/a.jpg",
|
|
clientPhotoId: "2a1d96b1-c0cc-489f-9f42-419ff1439a62"
|
|
)
|
|
)
|
|
|
|
XCTAssertEqual(response.id, 6)
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "POST")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/upload-material")
|
|
let body = try JSONSerialization.jsonObject(with: try XCTUnwrap(request.httpBody)) as? [String: Any]
|
|
XCTAssertEqual(body?["user_equity_travel_id"] as? Int, 3)
|
|
XCTAssertEqual(body?["file_name"] as? String, "A.JPG")
|
|
XCTAssertEqual(body?["file_url"] as? String, "https://cdn/a.jpg")
|
|
XCTAssertEqual(body?["client_photo_id"] as? String, "2a1d96b1-c0cc-489f-9f42-419ff1439a62")
|
|
}
|
|
|
|
func testMaterialClientPhotoIdsBuildsPathQueryAndDecodesResponse() async throws {
|
|
let data = envelopeJSON(#"{"client_photo_ids":["one","two"]}"#)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let response = try await api.materialClientPhotoIds(userEquityTravelId: 3)
|
|
|
|
XCTAssertEqual(response.clientPhotoIds, ["one", "two"])
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "GET")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/material-client-photo-ids")
|
|
let query = URLComponents(url: request.url!, resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "user_equity_travel_id" }?.value, "3")
|
|
}
|
|
|
|
func testAIRetouchTemplatesBuildsQueryAndDecodesGroups() async throws {
|
|
let data = envelopeJSON(
|
|
#"{"refined_templates":[{"id":1,"name":"清透","preview_url":"https://cdn/refined.jpg","before_url":"https://cdn/refined-before.jpg","after_url":"https://cdn/refined-after.jpg"}],"atmosphere_templates":[{"id":2,"name":"暖阳","preview_url":"https://cdn/atmosphere.jpg","before_url":"https://cdn/atmosphere-before.jpg","after_url":"https://cdn/atmosphere-after.jpg"}],"cover_templates":[{"id":3,"name":"杂志","preview_url":"https://cdn/cover.jpg","before_url":"https://cdn/cover-before.jpg","after_url":"https://cdn/cover-after.jpg"}],"remaining_quota":12}"#
|
|
)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let response = try await api.aiRetouchTemplates(scenicId: 18)
|
|
|
|
XCTAssertEqual(response.refinedTemplates.first?.name, "清透")
|
|
XCTAssertEqual(response.refinedTemplates.first?.beforeURL, "https://cdn/refined-before.jpg")
|
|
XCTAssertEqual(response.refinedTemplates.first?.afterURL, "https://cdn/refined-after.jpg")
|
|
XCTAssertEqual(response.atmosphereTemplates.first?.id, 2)
|
|
XCTAssertEqual(response.coverTemplates.first?.previewURL, "https://cdn/cover.jpg")
|
|
XCTAssertEqual(response.remainingQuota, 12)
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "GET")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/ai-retouch-templates")
|
|
let query = URLComponents(url: try XCTUnwrap(request.url), resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "scenic_id" }?.value, "18")
|
|
}
|
|
|
|
func testSubmitAIRetouchEncodesRequiredAndSelectedOptionalTemplates() async throws {
|
|
let session = MockURLSession(responses: [jobSubmissionJSON(batchId: 9, albumId: 6)])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
try await api.submitAIRetouch(
|
|
TravelAlbumAIRetouchRequest(
|
|
userEquityTravelId: 6,
|
|
materialIds: [11, 12, 13, 14],
|
|
refinedTemplateId: 21,
|
|
atmosphereTemplateId: 22,
|
|
coverTemplateId: 31
|
|
)
|
|
)
|
|
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.httpMethod, "POST")
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/ai-retouch")
|
|
let body = try JSONSerialization.jsonObject(with: try XCTUnwrap(request.httpBody)) as? [String: Any]
|
|
XCTAssertEqual(body?["user_equity_travel_id"] as? Int, 6)
|
|
XCTAssertEqual(body?["material_ids"] as? [Int], [11, 12, 13, 14])
|
|
XCTAssertEqual(body?["refined_template_id"] as? Int, 21)
|
|
XCTAssertEqual(body?["atmosphere_template_id"] as? Int, 22)
|
|
XCTAssertEqual(body?["cover_template_id"] as? Int, 31)
|
|
XCTAssertEqual(Set(body?.keys.map { $0 } ?? []), [
|
|
"user_equity_travel_id",
|
|
"material_ids",
|
|
"refined_template_id",
|
|
"atmosphere_template_id",
|
|
"cover_template_id",
|
|
])
|
|
}
|
|
|
|
func testSubmitAIRetouchOmitsAllOptionalTemplatesWhenAbsent() async throws {
|
|
let session = MockURLSession(responses: [jobSubmissionJSON(batchId: 10, albumId: 6)])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
try await api.submitAIRetouch(
|
|
TravelAlbumAIRetouchRequest(
|
|
userEquityTravelId: 6,
|
|
materialIds: [11],
|
|
refinedTemplateId: 21,
|
|
atmosphereTemplateId: nil,
|
|
coverTemplateId: nil
|
|
)
|
|
)
|
|
|
|
let body = try JSONSerialization.jsonObject(with: try XCTUnwrap(session.requests.first?.httpBody)) as? [String: Any]
|
|
XCTAssertNil(body?["atmosphere_template_id"])
|
|
XCTAssertNil(body?["cover_template_id"])
|
|
XCTAssertEqual(Set(body?.keys.map { $0 } ?? []), [
|
|
"user_equity_travel_id",
|
|
"material_ids",
|
|
"refined_template_id",
|
|
])
|
|
}
|
|
|
|
func testSubmitAIReretouchEncodesOnlyFieldsRequiredByEachType() async throws {
|
|
let session = MockURLSession(responses: [
|
|
jobSubmissionJSON(batchId: 51, albumId: 6),
|
|
jobSubmissionJSON(batchId: 52, albumId: 6),
|
|
jobSubmissionJSON(batchId: 53, albumId: 6),
|
|
jobSubmissionJSON(batchId: 54, albumId: 6),
|
|
])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
try await api.submitAIReretouch(
|
|
TravelAlbumAIReretouchRequest(
|
|
id: 11,
|
|
aiRetouchBatchId: 51,
|
|
type: .refined,
|
|
refinedTemplateId: 21,
|
|
atmosphereTemplateId: nil
|
|
)
|
|
)
|
|
try await api.submitAIReretouch(
|
|
TravelAlbumAIReretouchRequest(
|
|
id: 12,
|
|
aiRetouchBatchId: 52,
|
|
type: .atmosphere,
|
|
refinedTemplateId: nil,
|
|
atmosphereTemplateId: 22
|
|
)
|
|
)
|
|
try await api.submitAIReretouch(
|
|
TravelAlbumAIReretouchRequest(
|
|
id: 13,
|
|
aiRetouchBatchId: 53,
|
|
type: .all,
|
|
refinedTemplateId: 21,
|
|
atmosphereTemplateId: 22
|
|
)
|
|
)
|
|
try await api.submitAIReretouch(
|
|
TravelAlbumAIReretouchRequest(
|
|
id: 14,
|
|
aiRetouchBatchId: 54,
|
|
type: .all,
|
|
refinedTemplateId: 21,
|
|
atmosphereTemplateId: nil
|
|
)
|
|
)
|
|
|
|
let bodies = try session.requests.map { request in
|
|
try JSONSerialization.jsonObject(with: XCTUnwrap(request.httpBody)) as? [String: Any]
|
|
}
|
|
XCTAssertEqual(session.requests.map { $0.url?.path }, Array(
|
|
repeating: "/api/yf-handset-app/photog/travel-album/ai-reretouch",
|
|
count: 4
|
|
))
|
|
XCTAssertEqual(Set(bodies[0]?.keys.map { $0 } ?? []), ["id", "ai_retouch_batch_id", "type", "refined_template_id"])
|
|
XCTAssertEqual(Set(bodies[1]?.keys.map { $0 } ?? []), ["id", "ai_retouch_batch_id", "type", "atmosphere_template_id"])
|
|
XCTAssertEqual(Set(bodies[2]?.keys.map { $0 } ?? []), [
|
|
"id",
|
|
"ai_retouch_batch_id",
|
|
"type",
|
|
"refined_template_id",
|
|
"atmosphere_template_id",
|
|
])
|
|
XCTAssertEqual(Set(bodies[3]?.keys.map { $0 } ?? []), [
|
|
"id",
|
|
"ai_retouch_batch_id",
|
|
"type",
|
|
"refined_template_id",
|
|
])
|
|
XCTAssertEqual(bodies[0]?["type"] as? Int, 1)
|
|
XCTAssertEqual(bodies[1]?["type"] as? Int, 2)
|
|
XCTAssertEqual(bodies[2]?["type"] as? Int, 3)
|
|
XCTAssertEqual(bodies[3]?["type"] as? Int, 3)
|
|
}
|
|
|
|
func testAIJobListBuildsCursorQueryAndDecodesUnknownStatusSafely() async throws {
|
|
let data = envelopeJSON(
|
|
#"{"items":[{"ai_retouch_batch_id":91,"user_equity_travel_id":6,"scope":"album","status":"future_status","album":{"id":6,"name":"九寨沟旅拍","user_phone":"138****0000","cover_url":""},"source_count":1,"outputs":[{"type":"refined","count":1}],"preview_images":[],"progress":{"total":1,"queued":1,"processing":0,"succeeded":0,"failed":0,"canceled":0},"estimated_finish_at":null,"failure_summary":null,"created_at":"2026-08-14T06:00:00Z","started_at":null,"finished_at":null}],"next_cursor":"cursor-2","has_more":true}"#
|
|
)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let response = try await api.aiRetouchJobList(
|
|
statusGroup: .inProgress,
|
|
limit: 20,
|
|
cursor: "cursor-1"
|
|
)
|
|
|
|
XCTAssertEqual(response.items.first?.status, .unknown("future_status"))
|
|
XCTAssertEqual(response.nextCursor, "cursor-2")
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/ai-retouch-job-list")
|
|
let query = URLComponents(url: try XCTUnwrap(request.url), resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "status_group" }?.value, "in_progress")
|
|
XCTAssertEqual(query?.first { $0.name == "limit" }?.value, "20")
|
|
XCTAssertEqual(query?.first { $0.name == "cursor" }?.value, "cursor-1")
|
|
}
|
|
|
|
func testAIJobInfoDecodesPerOutputFailureReason() async throws {
|
|
let data = envelopeJSON(
|
|
#"{"ai_retouch_batch_id":91,"user_equity_travel_id":6,"scope":"album","status":"partially_succeeded","album":{"id":6,"name":"九寨沟旅拍","user_phone":"138****0000","cover_url":""},"source_count":1,"outputs":[{"type":"refined","count":1}],"progress":{"total":1,"queued":0,"processing":0,"succeeded":0,"failed":1,"canceled":0},"quota_settlement":{"status":"settled","reserved_units":1,"consumed_units":0,"released_units":1,"cover_units":0},"targets":[{"target_id":1,"source_material":{"id":11,"file_name":"A.JPG","thumbnail_url":""},"input_material_ids":[11],"output_type":"refined","template":{"id":21,"name":"清透"},"status":"failed","result_asset":null,"error":{"code":"FACE_NOT_FOUND","message":"未识别到清晰人脸，请更换照片","retryable":true},"created_at":"2026-08-14T06:00:00Z","started_at":null,"finished_at":"2026-08-14T06:01:00Z"}],"estimated_finish_at":null,"created_at":"2026-08-14T06:00:00Z","started_at":null,"finished_at":"2026-08-14T06:01:00Z","duration_seconds":60}"#
|
|
)
|
|
let session = MockURLSession(responses: [data])
|
|
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
|
|
|
|
let detail = try await api.aiRetouchJobInfo(batchId: 91)
|
|
|
|
XCTAssertEqual(detail.status, .partiallySucceeded)
|
|
XCTAssertEqual(detail.targets.first?.displayFailureMessage, "未识别到清晰人脸，请更换照片")
|
|
XCTAssertEqual(detail.targets.first?.error?.retryable, true)
|
|
let request = try XCTUnwrap(session.requests.first)
|
|
XCTAssertEqual(request.url?.path, "/api/yf-handset-app/photog/travel-album/ai-retouch-job-info")
|
|
let query = URLComponents(url: try XCTUnwrap(request.url), resolvingAgainstBaseURL: false)?.queryItems
|
|
XCTAssertEqual(query?.first { $0.name == "ai_retouch_batch_id" }?.value, "91")
|
|
}
|
|
|
|
private func envelopeJSON(_ dataJSON: String) -> Data {
|
|
"""
|
|
{"code":100000,"msg":"success","data":\(dataJSON)}
|
|
""".data(using: .utf8)!
|
|
}
|
|
|
|
private func jobSubmissionJSON(batchId: Int, albumId: Int) -> Data {
|
|
envelopeJSON(
|
|
#"{"ai_retouch_batch_id":\#(batchId),"user_equity_travel_id":\#(albumId),"status":"queued","progress":{"total":3,"queued":3,"processing":0,"succeeded":0,"failed":0,"canceled":0},"created_at":"2026-08-14T06:00:00Z"}"#
|
|
)
|
|
}
|
|
}
|