TM-SGNL-iOS/SignalServiceKit/tests/Util/NSData+ImageTest.swift
TeleMessage developers dde0620daf initial commit
2025-05-03 12:28:28 -07:00

52 lines
1.5 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.image(color: .red, size: CGSize(width: 1, height: 1))
let data = image.pngData()!
let isApng = data.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.isAnimatedPngData()
XCTAssertEqual(isApng?.boolValue, true)
}
func testIsAnimatedPngData_invalid() {
do {
let data = Randomness.generateRandomBytes(0)
let isApng: NSNumber? = data.isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(1)
let isApng: NSNumber? = data.isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(64)
let isApng: NSNumber? = data.isAnimatedPngData()
XCTAssertNil(isApng)
}
do {
let data = Randomness.generateRandomBytes(1024)
let isApng: NSNumber? = data.isAnimatedPngData()
XCTAssertNil(isApng)
}
}
}