Signal-iOS/SignalServiceKit/tests/Util/NSData+ImageTest.swift
2024-01-08 15:43:31 -06:00

53 lines
1.6 KiB
Swift

//
// Copyright 2020 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
import Foundation
import XCTest
@testable import SignalServiceKit
class NSDataImageTests: XCTestCase {
func testIsAnimatedPngData_png() {
let image = UIImage(color: .red, size: CGSize(width: 1, height: 1))
let data = image.pngData()!
let isApng = (data as NSData).isAnimatedPngData()
XCTAssertEqual(isApng?.boolValue, false)
}
func testIsAnimatedPngData_apng() {
let data: Data = {
let testBundle = Bundle(for: Self.self)
let url = testBundle.url(forResource: "test-apng", withExtension: "png")!
return try! Data(contentsOf: url)
}()
let isApng = (data as NSData).isAnimatedPngData()
XCTAssertEqual(isApng?.boolValue, true)
}
func testIsAnimatedPngData_invalid() {
do {
let data = Randomness.generateRandomBytes(0)
let isApng: NSNumber? = (data as NSData).isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(1)
let isApng: NSNumber? = (data as NSData).isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(64)
let isApng: NSNumber? = (data as NSData).isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(1024)
let isApng: NSNumber? = (data as NSData).isAnimatedPngData()
XCTAssertNil(isApng)
}
}
}