|
|
@@ -0,0 +1,183 @@
|
|
|
+//
|
|
|
+// LivePhotoSourceTests.swift
|
|
|
+// Kingfisher
|
|
|
+//
|
|
|
+// Created by onevcat on 2024/10/01.
|
|
|
+//
|
|
|
+// Copyright (c) 2024 Wei Wang <onevcat@gmail.com>
|
|
|
+//
|
|
|
+// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
+// of this software and associated documentation files (the "Software"), to deal
|
|
|
+// in the Software without restriction, including without limitation the rights
|
|
|
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
+// copies of the Software, and to permit persons to whom the Software is
|
|
|
+// furnished to do so, subject to the following conditions:
|
|
|
+//
|
|
|
+// The above copyright notice and this permission notice shall be included in
|
|
|
+// all copies or substantial portions of the Software.
|
|
|
+//
|
|
|
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
+// THE SOFTWARE.
|
|
|
+
|
|
|
+import XCTest
|
|
|
+@testable import Kingfisher
|
|
|
+
|
|
|
+class LivePhotoSourceTests: XCTestCase {
|
|
|
+
|
|
|
+ func testLivePhotoResourceInitialization() {
|
|
|
+ let url = URL(string: "https://example.com/photo.heic")!
|
|
|
+ let resource = LivePhotoResource(downloadURL: url)
|
|
|
+
|
|
|
+ XCTAssertEqual(resource.downloadURL, url)
|
|
|
+ XCTAssertEqual(resource.referenceFileType, .heic)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceInitializationWithResource() {
|
|
|
+ let url = URL(string: "https://example.com/photo.mov")!
|
|
|
+ let imageResource = KF.ImageResource(downloadURL: url)
|
|
|
+ let resource = LivePhotoResource(resource: imageResource)
|
|
|
+
|
|
|
+ XCTAssertEqual(resource.downloadURL, url)
|
|
|
+ XCTAssertEqual(resource.referenceFileType, .mov)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileExtensionByType() {
|
|
|
+ let mov = LivePhotoResource.FileType.mov
|
|
|
+ XCTAssertEqual(mov.determinedFileExtension(Data()), "mov")
|
|
|
+ XCTAssertEqual(mov.fileExtension, "mov")
|
|
|
+
|
|
|
+ let heic = LivePhotoResource.FileType.heic
|
|
|
+ XCTAssertEqual(heic.determinedFileExtension(Data()), "heic")
|
|
|
+ XCTAssertEqual(heic.fileExtension, "heic")
|
|
|
+
|
|
|
+ let other = LivePhotoResource.FileType.other("exe")
|
|
|
+ XCTAssertEqual(other.fileExtension, "exe")
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForHEIC() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00, 0x66, 0x74, 0x79, 0x70, 0x68, 0x65, 0x69, 0x63])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, "heic")
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForQT() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00, 0x66, 0x74, 0x79, 0x70, 0x71, 0x74, 0x20, 0x20])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, "mov")
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForExplicitFileType() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00, 0x66, 0x74, 0x79, 0x70, 0x71, 0x74, 0x20, 0x20])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("ext")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, "ext")
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForUnknown() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00, 0x66, 0x74, 0x79, 0x70, 0x71, 0x74, 0x20, 0x22])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, nil)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForNonFYTP() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00, 0x12, 0x34, 0x56, 0x78, 0x71, 0x74, 0x20, 0x20])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, nil)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceFileTypeDeterminationForNotEnoughData() {
|
|
|
+ let data = Data([0x00, 0x00, 0x00, 0x00])
|
|
|
+ let fileType = LivePhotoResource.FileType.other("")
|
|
|
+ let determinedExtension = fileType.determinedFileExtension(data)
|
|
|
+
|
|
|
+ XCTAssertEqual(determinedExtension, nil)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoSourceInitializationWithResources() {
|
|
|
+ let url1 = URL(string: "https://example.com/photo1.heic")!
|
|
|
+ let url2 = URL(string: "https://example.com/photo2.mov")!
|
|
|
+ let resources = [KF.ImageResource(downloadURL: url1), KF.ImageResource(downloadURL: url2)]
|
|
|
+ let livePhotoSource = LivePhotoSource(resources: resources)
|
|
|
+
|
|
|
+ XCTAssertEqual(livePhotoSource.resources.count, 2)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[0].downloadURL, url1)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[1].downloadURL, url2)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoSourceInitializationWithURLs() {
|
|
|
+ let url1 = URL(string: "https://example.com/photo1.heic")!
|
|
|
+ let url2 = URL(string: "https://example.com/photo2.mov")!
|
|
|
+ let livePhotoSource = LivePhotoSource(urls: [url1, url2])
|
|
|
+
|
|
|
+ XCTAssertEqual(livePhotoSource.resources.count, 2)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[0].downloadURL, url1)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[1].downloadURL, url2)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceInitializationWithCacheKey() {
|
|
|
+ let url = URL(string: "https://example.com/photo.heic")!
|
|
|
+ let cacheKey = "customCacheKey"
|
|
|
+ let resource = LivePhotoResource(downloadURL: url, cacheKey: cacheKey)
|
|
|
+
|
|
|
+ XCTAssertEqual(resource.downloadURL, url)
|
|
|
+ XCTAssertEqual(resource.cacheKey, cacheKey)
|
|
|
+ XCTAssertEqual(resource.referenceFileType, .heic)
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceInitializationWithFileType() {
|
|
|
+ let url = URL(string: "https://example.com/photo.unknown")!
|
|
|
+ let resource = LivePhotoResource(downloadURL: url, fileType: .other("unknown"))
|
|
|
+
|
|
|
+ XCTAssertEqual(resource.downloadURL, url)
|
|
|
+ XCTAssertEqual(resource.referenceFileType, .other("unknown"))
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoResourceGuessedFileType() {
|
|
|
+ let url1 = URL(string: "https://example.com/photo.heic")!
|
|
|
+ let url2 = URL(string: "https://example.com/photo.mov")!
|
|
|
+ let url3 = URL(string: "https://example.com/photo.unknown")!
|
|
|
+
|
|
|
+ let resource1 = KF.ImageResource(downloadURL: url1)
|
|
|
+ let resource2 = KF.ImageResource(downloadURL: url2)
|
|
|
+ let resource3 = KF.ImageResource(downloadURL: url3)
|
|
|
+
|
|
|
+ XCTAssertEqual(resource1.guessedFileType, .heic)
|
|
|
+ XCTAssertEqual(resource2.guessedFileType, .mov)
|
|
|
+ XCTAssertEqual(resource3.guessedFileType, .other("unknown"))
|
|
|
+ }
|
|
|
+
|
|
|
+ func testLivePhotoSourceInitializationWithMixedResources() {
|
|
|
+ let url1 = URL(string: "https://example.com/photo1.heic")!
|
|
|
+ let url2 = URL(string: "https://example.com/photo2.mov")!
|
|
|
+ let url3 = URL(string: "https://example.com/photo3.unknown")!
|
|
|
+ let resources = [
|
|
|
+ KF.ImageResource(downloadURL: url1),
|
|
|
+ KF.ImageResource(downloadURL: url2),
|
|
|
+ KF.ImageResource(downloadURL: url3)
|
|
|
+ ]
|
|
|
+ let livePhotoSource = LivePhotoSource(resources: resources)
|
|
|
+
|
|
|
+ XCTAssertEqual(livePhotoSource.resources.count, 3)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[0].downloadURL, url1)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[1].downloadURL, url2)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[2].downloadURL, url3)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[0].referenceFileType, .heic)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[1].referenceFileType, .mov)
|
|
|
+ XCTAssertEqual(livePhotoSource.resources[2].referenceFileType, .other("unknown"))
|
|
|
+ }
|
|
|
+
|
|
|
+}
|
onevcat