amplify-swift/AmplifyPlugins/Storage/Tests/AWSS3StoragePluginTests/Operation/AWSS3StorageRemoveOperationTests.swift

//
// Copyright Amazon.com Inc. or its affiliates.
// All Rights Reserved.
//
// SPDX-License-Identifier: Apache-2.0
//

import XCTest
@testable import Amplify
@testable import AmplifyTestCommon
@testable import AWSPluginsCore
@testable import AWSS3StoragePlugin
@testable import AWSPluginsTestCommon

class AWSS3StorageRemoveOperationTests: AWSS3StorageOperationTestBase {

    func testRemoveOperationValidationError() {
        let options = StorageRemoveRequest.Options()
        let request = StorageRemoveRequest(key: "", options: options)

        let failedInvoked = expectation(description: "failed was invoked on operation")
        let operation = AWSS3StorageRemoveOperation(request,
                                                    storageConfiguration: testStorageConfiguration,
                                                    storageService: mockStorageService,
                                                    authService: mockAuthService) { result in
            switch result {
            case .failure(let error):
                guard case .validation = error else {
                    XCTFail("Should have failed with validation error")
                    return
                }
                failedInvoked.fulfill()
            default:
                XCTFail("Should have received failed event")
            }
        }

        operation.start()

        waitForOperationToFinish(operation)
        waitForExpectations(timeout: 1)
    }

    func testRemoveOperationGetIdentityIdError() async throws {
        mockAuthService.getIdentityIdError = AuthError.service("", "", "")

        let options = StorageRemoveRequest.Options()
        let request = StorageRemoveRequest(key: testKey, options: options)
        let failedInvoked = expectation(description: "failed was invoked on operation")
        let operation = AWSS3StorageRemoveOperation(request,
                                                    storageConfiguration: testStorageConfiguration,
                                                    storageService: mockStorageService,
                                                    authService: mockAuthService) { result in
            switch result {
            case .failure(let error):
                guard case .authError = error else {
                    XCTFail("Should have failed with authError")
                    return
                }
                failedInvoked.fulfill()
            default:
                XCTFail("Should have received failed event")
            }
        }

        operation.start()
        await waitForExpectations(timeout: 1)
        waitForOperationToFinish(operation)
    }

    func testRemoveOperationDeleteSuccess() async throws {
        mockStorageService.storageServiceDeleteEvents = [StorageEvent.completedVoid]
        let options = StorageRemoveRequest.Options()
        let request = StorageRemoveRequest(key: testKey, options: options)

        let expectedServiceKey = StorageAccessLevel.guest.serviceAccessPrefix + "/" + testKey
        let completeInvoked = expectation(description: "complete was invoked on operation")
        let operation = AWSS3StorageRemoveOperation(request,
                                                    storageConfiguration: testStorageConfiguration,
                                                    storageService: mockStorageService,
                                                    authService: mockAuthService) { result in
            switch result {
            case .success:
                completeInvoked.fulfill()
            default:
                XCTFail("Should have received completed event")
            }
        }

        operation.start()

        await waitForExpectations(timeout: 1)
        waitForOperationToFinish(operation)
        mockStorageService.verifyDelete(serviceKey: expectedServiceKey)
    }

    func testRemoveOperationDeleteFail() async throws {
        mockStorageService.storageServiceDeleteEvents = [StorageEvent.failed(StorageError.service("", ""))]
        let options = StorageRemoveRequest.Options()
        let request = StorageRemoveRequest(key: testKey, options: options)

        let expectedServiceKey = StorageAccessLevel.guest.serviceAccessPrefix + "/" + testKey
        let failedInvoked = expectation(description: "failed was invoked on operation")
        let operation = AWSS3StorageRemoveOperation(request,
                                                    storageConfiguration: testStorageConfiguration,
                                                    storageService: mockStorageService,
                                                    authService: mockAuthService) { result in
            switch result {
            case .failure:
                failedInvoked.fulfill()
            default:
                XCTFail("Should have received failed event")
            }
        }

        operation.start()

        await waitForExpectations(timeout: 1)
        waitForOperationToFinish(operation)
        mockStorageService.verifyDelete(serviceKey: expectedServiceKey)
    }

    func testRemoveOperationDeleteForPrivateAccessLevel() {
        mockAuthService.identityId = testIdentityId
        mockStorageService.storageServiceDeleteEvents = [StorageEvent.completedVoid]
        let options = StorageRemoveRequest.Options(accessLevel: .private)
        let request = StorageRemoveRequest(key: testKey, options: options)

        let expectedServiceKey = StorageAccessLevel.private.rawValue + "/" + testIdentityId + "/" + testKey
        let completeInvoked = expectation(description: "complete was invoked on operation")
        let operation = AWSS3StorageRemoveOperation(request,
                                                    storageConfiguration: testStorageConfiguration,
                                                    storageService: mockStorageService,
                                                    authService: mockAuthService) { result in
            switch result {
            case .success:
                completeInvoked.fulfill()
            default:
                XCTFail("Should have received completed event")
            }
        }

        operation.start()

        waitForExpectations(timeout: 1)
        waitForOperationToFinish(operation)
        mockStorageService.verifyDelete(serviceKey: expectedServiceKey)
    }
}