From f6074c2e5d8be0511e91aea483a7ad024c77dea2 Mon Sep 17 00:00:00 2001
From: DreamPiggy <lizhuoli1126@126.com>
Date: Wed, 10 Mar 2021 17:09:37 +0800
Subject: [PATCH] Update the test case for the CGImage based

---
 Tests/WebImageTests.swift | 29 +++++++++++++----------------
 1 file changed, 13 insertions(+), 16 deletions(-)

diff --git a/Tests/WebImageTests.swift b/Tests/WebImageTests.swift
index bb098674..1ba29b2a 100644
--- a/Tests/WebImageTests.swift
+++ b/Tests/WebImageTests.swift
@@ -22,10 +22,10 @@ class WebImageTests: XCTestCase {
         let imageUrl = URL(string: "https://nr-platform.s3.amazonaws.com/uploads/platform/published_extension/branding_icon/275/AmazonS3.png")
         let imageView = WebImage(url: imageUrl)
         let introspectView = imageView.onSuccess { image, data, cacheType in
-            #if os(iOS) || os(tvOS)
-            let displayImage = try? imageView.inspect().group().image(0).uiImage()
-            #else
+            #if os(macOS)
             let displayImage = try? imageView.inspect().group().image(0).nsImage()
+            #else
+            let displayImage = try? imageView.inspect().group().image(0).cgImage()
             #endif
             XCTAssertNotNil(displayImage)
             expectation.fulfill()
@@ -46,15 +46,17 @@ class WebImageTests: XCTestCase {
         let introspectView = imageView.onSuccess { image, data, cacheType in
             if let animatedImage = image as? SDAnimatedImage {
                 XCTAssertTrue(imageView.isAnimating)
-                #if os(iOS) || os(tvOS)
-                let displayImage = try? imageView.inspect().group().image(0).uiImage()
-                #else
+                #if os(macOS)
                 let displayImage = try? imageView.inspect().group().image(0).nsImage()
+                let size = displayImage?.size
+                #else
+                let displayImage = try? imageView.inspect().group().image(0).cgImage()
+                let size = CGSize(width: displayImage?.width ?? 0, height: displayImage?.height ?? 0)
                 #endif
                 XCTAssertNotNil(displayImage)
                 // Check display image should match the animated poster frame
                 let posterImage = animatedImage.animatedImageFrame(at: 0)
-                XCTAssertEqual(displayImage?.size, posterImage?.size)
+                XCTAssertEqual(size, posterImage?.size)
                 expectation.fulfill()
             } else {
                 XCTFail("WebImage animated image invalid")
@@ -162,15 +164,10 @@ class WebImageTests: XCTestCase {
             let displayImage = try? imageView.inspect().group().image(0).nsImage()
             XCTAssertNotNil(displayImage)
             #else
-            if #available(iOS 14.0, watchOS 7.0, tvOS 14.0, *) {
-                let displayImage = try? imageView.inspect().group().image(0).uiImage()
-                XCTAssertEqual(displayImage, image)
-            } else {
-                let displayImage = try? imageView.inspect().group().image(0).cgImage()
-                let orientation = try? imageView.inspect().group().image(0).orientation()
-                XCTAssertNotNil(displayImage)
-                XCTAssertEqual(orientation, .leftMirrored)
-            }
+            let displayImage = try? imageView.inspect().group().image(0).cgImage()
+            let orientation = try? imageView.inspect().group().image(0).orientation()
+            XCTAssertNotNil(displayImage)
+            XCTAssertEqual(orientation, .leftMirrored)
             #endif
             expectation.fulfill()
         }.onFailure { error in