187 lines
8.4 KiB
Swift
187 lines
8.4 KiB
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 AWSS3StoragePlugin
|
|
@testable import AWSPluginsCore
|
|
@testable import AWSPluginsTestCommon
|
|
|
|
import AWSS3
|
|
|
|
class AWSS3StorageDownloadFileOperationTests: AWSS3StorageOperationTestBase {
|
|
|
|
func testDownloadFileOperationValidationError() {
|
|
let request = StorageDownloadFileRequest(key: "",
|
|
local: testURL,
|
|
options: StorageDownloadFileRequest.Options())
|
|
let failedInvoked = expectation(description: "failed was invoked on operation")
|
|
let operation = AWSS3StorageDownloadFileOperation(request,
|
|
storageConfiguration: testStorageConfiguration,
|
|
storageService: mockStorageService,
|
|
authService: mockAuthService,
|
|
progressListener: nil) { 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()
|
|
waitForExpectations(timeout: 1)
|
|
waitForOperationToFinish(operation)
|
|
}
|
|
|
|
func testDownloadFileOperationGetIdentityIdError() async throws {
|
|
mockAuthService.getIdentityIdError = AuthError.service("", "", "")
|
|
let request = StorageDownloadFileRequest(key: testKey,
|
|
local: testURL,
|
|
options: StorageDownloadFileRequest.Options())
|
|
let failedInvoked = expectation(description: "failed was invoked on operation")
|
|
let operation = AWSS3StorageDownloadFileOperation(request,
|
|
storageConfiguration: testStorageConfiguration,
|
|
storageService: mockStorageService,
|
|
authService: mockAuthService,
|
|
progressListener: nil) { result in
|
|
switch result {
|
|
case .failure(let error):
|
|
guard case .authError = error else {
|
|
XCTFail("Should have failed with identity error")
|
|
return
|
|
}
|
|
failedInvoked.fulfill()
|
|
default:
|
|
XCTFail("Should have received failed event")
|
|
}
|
|
}
|
|
|
|
operation.start()
|
|
|
|
await waitForExpectations(timeout: 1)
|
|
waitForOperationToFinish(operation)
|
|
}
|
|
|
|
func testDownloadFileOperationDownloadLocal() {
|
|
let task = StorageTransferTask(transferType: .download(onEvent: { _ in }), bucket: "bucket", key: "key")
|
|
mockStorageService.storageServiceDownloadEvents = [
|
|
StorageEvent.initiated(StorageTaskReference(task)),
|
|
StorageEvent.inProcess(Progress()),
|
|
StorageEvent.completed(nil)]
|
|
let url = URL(fileURLWithPath: "path")
|
|
let request = StorageDownloadFileRequest(key: testKey,
|
|
local: testURL,
|
|
options: StorageDownloadFileRequest.Options())
|
|
let expectedServiceKey = StorageAccessLevel.guest.serviceAccessPrefix + "/" + testKey
|
|
let inProcessInvoked = expectation(description: "inProgress was invoked on operation")
|
|
let completeInvoked = expectation(description: "complete was invoked on operation")
|
|
let operation = AWSS3StorageDownloadFileOperation(
|
|
request,
|
|
storageConfiguration: testStorageConfiguration,
|
|
storageService: mockStorageService,
|
|
authService: mockAuthService,
|
|
progressListener: { _ in
|
|
inProcessInvoked.fulfill()
|
|
}, resultListener: { result in
|
|
switch result {
|
|
case .success:
|
|
completeInvoked.fulfill()
|
|
case .failure(let error):
|
|
XCTFail("Unexpected error on operation: \(error)")
|
|
}
|
|
})
|
|
|
|
operation.start()
|
|
|
|
waitForExpectations(timeout: 1)
|
|
waitForOperationToFinish(operation)
|
|
mockStorageService.verifyDownload(serviceKey: expectedServiceKey, fileURL: url)
|
|
}
|
|
|
|
func testDownloadFileOperationDownloadLocalFailed() {
|
|
let task = StorageTransferTask(transferType: .download(onEvent: { _ in }), bucket: "bucket", key: "key")
|
|
mockStorageService.storageServiceDownloadEvents = [
|
|
StorageEvent.initiated(StorageTaskReference(task)),
|
|
StorageEvent.inProcess(Progress()),
|
|
StorageEvent.failed(StorageError.service("", ""))]
|
|
let url = URL(fileURLWithPath: "path")
|
|
let request = StorageDownloadFileRequest(key: testKey,
|
|
local: testURL,
|
|
options: StorageDownloadFileRequest.Options())
|
|
let expectedServiceKey = StorageAccessLevel.guest.serviceAccessPrefix + "/" + testKey
|
|
let inProcessInvoked = expectation(description: "inProgress was invoked on operation")
|
|
let failedInvoked = expectation(description: "failed was invoked on operation")
|
|
let operation = AWSS3StorageDownloadFileOperation(
|
|
request,
|
|
storageConfiguration: testStorageConfiguration,
|
|
storageService: mockStorageService,
|
|
authService: mockAuthService,
|
|
progressListener: { _ in
|
|
inProcessInvoked.fulfill()
|
|
}, resultListener: { result in
|
|
switch result {
|
|
case .failure:
|
|
failedInvoked.fulfill()
|
|
case .success:
|
|
XCTFail("Unexpected event invoked on operation")
|
|
}
|
|
})
|
|
|
|
operation.start()
|
|
|
|
waitForExpectations(timeout: 1)
|
|
waitForOperationToFinish(operation)
|
|
mockStorageService.verifyDownload(serviceKey: expectedServiceKey, fileURL: url)
|
|
}
|
|
|
|
func testGetOperationDownloadLocalFromTargetIdentityId() async throws {
|
|
let task = StorageTransferTask(transferType: .download(onEvent: { _ in }), bucket: "bucket", key: "key")
|
|
mockStorageService.storageServiceDownloadEvents = [
|
|
StorageEvent.initiated(StorageTaskReference(task)),
|
|
StorageEvent.inProcess(Progress()),
|
|
StorageEvent.completed(nil)]
|
|
let url = URL(fileURLWithPath: "path")
|
|
let options = StorageDownloadFileRequest.Options(accessLevel: .protected,
|
|
targetIdentityId: testTargetIdentityId)
|
|
let request = StorageDownloadFileRequest(key: testKey,
|
|
local: testURL,
|
|
options: options)
|
|
let expectedServiceKey = StorageAccessLevel.protected.rawValue + "/" + testTargetIdentityId + "/" + testKey
|
|
let inProcessInvoked = expectation(description: "inProgress was invoked on operation")
|
|
let completeInvoked = expectation(description: "complete was invoked on operation")
|
|
let operation = AWSS3StorageDownloadFileOperation(
|
|
request,
|
|
storageConfiguration: testStorageConfiguration,
|
|
storageService: mockStorageService,
|
|
authService: mockAuthService,
|
|
progressListener: { _ in
|
|
inProcessInvoked.fulfill()
|
|
}, resultListener: { result in
|
|
switch result {
|
|
case .success:
|
|
completeInvoked.fulfill()
|
|
case .failure(let error):
|
|
XCTFail("Unexpected error on operation: \(error)")
|
|
}
|
|
})
|
|
|
|
operation.start()
|
|
|
|
await waitForExpectations(timeout: 1)
|
|
waitForOperationToFinish(operation)
|
|
mockStorageService.verifyDownload(serviceKey: expectedServiceKey, fileURL: url)
|
|
}
|
|
|
|
// TODO: missing unit tests for pause resume and cancel. do we create a mock of the StorageTaskReference?
|
|
}
|