MockProductImageUploader.swift

import Combine
@testable import Yosemite
@testable import WooCommerce

final class MockProductImageUploader {
    let errors: AnyPublisher<ProductImageUploadErrorInfo, Never>

    var activeUploads: AnyPublisher<[ProductImageUploaderKey], Never> {
        $activeUploadsKeys.eraseToAnyPublisher()
    }

    @Published var activeUploadsKeys: [ProductImageUploaderKey] = []
    var replaceLocalIDWasCalled = false
    var saveProductImagesWhenNoneIsPendingUploadAnymoreWasCalled = false
    var startEmittingErrorsWasCalled = false
    var stopEmittingErrorsWasCalled = false
    var resetWasCalled = false
    var sendBackgroundUploadNoticeIfNeededWasCalled = false

    private var onProductSaveResult: Result<[ProductImage], Error>?
    private var hasUnsavedChangesOnImages = false

    init(errors: AnyPublisher<ProductImageUploadErrorInfo, Never> =
         Empty<ProductImageUploadErrorInfo, Never>().eraseToAnyPublisher()) {
        self.errors = errors
    }

    func whenProductIsSaved(thenReturn result: Result<[ProductImage], Error>) {
        onProductSaveResult = result
    }

    func whenHasUnsavedChangesOnImagesIsCalled(thenReturn hasUnsavedChangesOnImages: Bool) {
        self.hasUnsavedChangesOnImages = hasUnsavedChangesOnImages
    }
}

extension MockProductImageUploader: ProductImageUploaderProtocol {

    func replaceLocalID(siteID: Int64, localID: ProductOrVariationID, remoteID: Int64) {
        replaceLocalIDWasCalled = true
    }

    func saveProductImagesWhenNoneIsPendingUploadAnymore(key: ProductImageUploaderKey,
                                                         onProductSave: @escaping (Result<[ProductImage], Error>) -> Void) {
        saveProductImagesWhenNoneIsPendingUploadAnymoreWasCalled = true
        if let result = onProductSaveResult {
            onProductSave(result)
        }
    }

    func actionHandler(key: ProductImageUploaderKey, originalStatuses: [ProductImageStatus]) -> ProductImageActionHandler {
        ProductImageActionHandler(siteID: 0, productID: .product(id: 0), imageStatuses: [])
    }

    func startEmittingErrors(key: ProductImageUploaderKey) {
        startEmittingErrorsWasCalled = true
    }

    func stopEmittingErrors(key: ProductImageUploaderKey) {
        stopEmittingErrorsWasCalled = true
    }

    func hasUnsavedChangesOnImages(key: ProductImageUploaderKey, originalImages: [ProductImage]) -> Bool {
        hasUnsavedChangesOnImages
    }

    func sendBackgroundUploadNoticeIfNeeded(key: ProductImageUploaderKey, using noticePresenter: NoticePresenter) {
        sendBackgroundUploadNoticeIfNeededWasCalled = true

        // It actually sends a notification if there are active uploads for this key.
        if activeUploadsKeys.contains(where: { $0 == key }) {
            noticePresenter.enqueue(notice: Notice(title: "Test Notice"))
        }
    }

    func reset() {
        resetWasCalled = true
    }
}