Add tests for fallback image size

Author: RafaelKayumov

WooCommerce/WooCommerceTests/Mocks/MockKingfisherImageDownloader.swift

@@ -5,6 +5,8 @@ final class MockKingfisherImageDownloader: Kingfisher.ImageDownloader {
     // Mocks in-memory cache.
     private let imagesByKey: [String: UIImage]
 
+    private(set) var capturedProcessor: ImageProcessor?
+
     init(imagesByKey: [String: UIImage]) {
         self.imagesByKey = imagesByKey
         super.init(name: "Mock!")
@@ -14,6 +16,14 @@ final class MockKingfisherImageDownloader: Kingfisher.ImageDownloader {
                                 options: KingfisherOptionsInfo? = nil,
                                 progressBlock: DownloadProgressBlock?,
                                 completionHandler: ((Result<ImageLoadingResult, KingfisherError>) -> Void)? = nil) -> DownloadTask? {
+        if let options = options {
+            for option in options {
+                if case .processor(let processor) = option {
+                    capturedProcessor = processor
+                    break
+                }
+            }
+        }
         if let image = imagesByKey[url.absoluteString] {
             completionHandler?(.success(.init(image: image, url: url, originalData: Data())))
         } else {
@@ -25,6 +35,7 @@ final class MockKingfisherImageDownloader: Kingfisher.ImageDownloader {
     override func downloadImage(with url: URL,
                                 options: KingfisherParsedOptionsInfo,
                                 completionHandler: ((Result<ImageLoadingResult, KingfisherError>) -> Void)? = nil) -> DownloadTask? {
+        capturedProcessor = options.processor
         if let image = imagesByKey[url.absoluteString] {
             completionHandler?(.success(.init(image: image, url: url, originalData: Data())))
         } else {

WooCommerce/WooCommerceTests/Tools/DefaultImageServiceTests.swift

@@ -265,4 +265,33 @@ final class DefaultImageServiceTests: XCTestCase {
 
         waitForExpectations(timeout: Constants.expectationTimeout, handler: nil)
     }
+
+    func testDownloadAndCacheImageForImageView_withEmptyBounds_usesDefaultThumbnailSize() {
+        // Given
+        let featureFlagService = MockFeatureFlagService()
+        featureFlagService.isProductImageOptimizedHandlingEnabled = true
+        ServiceLocator.setFeatureFlagService(featureFlagService)
+
+        let mockImageView = UIImageView(frame: .zero)
+        let mockCache = MockImageCache(name: "Testing")
+        let mockDownloader = MockKingfisherImageDownloader(imagesByKey: [url.absoluteString: testImage])
+        imageService = DefaultImageService(imageCache: mockCache, imageDownloader: mockDownloader)
+
+        // When
+        imageService.downloadAndCacheImageForImageView(
+            mockImageView,
+            with: url.absoluteString,
+            placeholder: nil,
+            progressBlock: nil,
+            completion: nil
+        )
+
+        // Then
+        guard let downsamplingProcessor = mockDownloader.capturedProcessor as? DownsamplingImageProcessor else {
+            XCTFail("DownsamplingImageProcessor not found or not the correct type")
+            return
+        }
+
+        XCTAssertEqual(downsamplingProcessor.size, CGSize(width: 800, height: 800))
+    }
 }