fix(Storage): increase testTask timeout and network timeout (#2446)

Author: thisisabhash

AmplifyAsyncTesting/Sources/AsyncTesting/XCTestCase+AsyncTesting.swift

@@ -10,7 +10,7 @@ import XCTest
 
 extension XCTestCase {
     public static let defaultTimeoutForAsyncExpectations = TimeInterval(60)
-    public static let defaultNetworkTimeoutForAsyncExpectations = TimeInterval(10)
+    public static let defaultNetworkTimeoutForAsyncExpectations = TimeInterval(300)
 
     /// Creates a new async expectation with an associated description.
     ///

AmplifyPlugins/Storage/Tests/StorageHostApp/AWSS3StoragePluginIntegrationTests/Helpers/TestConfigHelper.swift

@@ -39,5 +39,5 @@ class TestConfigHelper {
 }
 
 class TestCommonConstants {
-    static let networkTimeout = TimeInterval(180)
+    static let networkTimeout = TimeInterval(300)
 }

AmplifyPlugins/Storage/Tests/StorageHostApp/AWSS3StoragePluginIntegrationTests/ResumabilityTests/AWSS3StoragePluginGetDataResumabilityTests.swift

@@ -18,7 +18,7 @@ class AWSS3StoragePluginDownloadDataResumabilityTests: AWSS3StoragePluginTestBas
     /// When: Call the get API then pause
     /// Then: The operation is stalled (no progress, completed, or failed event)
     func testDownloadDataAndPause() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             let data = AWSS3StoragePluginTestBase.smallDataObject
             let uploadKey = try await Amplify.Storage.uploadData(key: key, data: data).value
@@ -44,7 +44,7 @@ class AWSS3StoragePluginDownloadDataResumabilityTests: AWSS3StoragePluginTestBas
                 }
             }
             await waitForExpectations([didPause], timeout: TestCommonConstants.networkTimeout)
-            await waitForExpectations([didContinue])
+            await waitForExpectations([didContinue], timeout: 5)
 
             let completeInvoked = asyncExpectation(description: "Download is completed", isInverted: true)
             let downloadTask = Task {
@@ -70,7 +70,7 @@ class AWSS3StoragePluginDownloadDataResumabilityTests: AWSS3StoragePluginTestBas
     /// When: Call the downloadData API, pause, and then resume the operation
     /// Then: The operation should complete successfully
     func testDownloadDataAndPauseThenResume() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             let data = AWSS3StoragePluginTestBase.smallDataObject
             let uploadKey = try await Amplify.Storage.uploadData(key: key, data: data).value
@@ -123,7 +123,7 @@ class AWSS3StoragePluginDownloadDataResumabilityTests: AWSS3StoragePluginTestBas
     /// When: Call the get API then cancel the operation,
     /// Then: The operation should not complete or fail.
     func testDownloadDataAndCancel() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             let data = AWSS3StoragePluginTestBase.smallDataObject
             let uploadKey = try await Amplify.Storage.uploadData(key: key, data: data).value
@@ -149,7 +149,7 @@ class AWSS3StoragePluginDownloadDataResumabilityTests: AWSS3StoragePluginTestBas
                 }
             }
             await waitForExpectations([didCancel], timeout: TestCommonConstants.networkTimeout)
-            await waitForExpectations([didContinue])
+            await waitForExpectations([didContinue], timeout: 5)
 
             let completeInvoked = asyncExpectation(description: "Download is completed", isInverted: true)
             let downloadTask = Task {

AmplifyPlugins/Storage/Tests/StorageHostApp/AWSS3StoragePluginIntegrationTests/ResumabilityTests/AWSS3StoragePluginPutDataResumabilityTests.swift

@@ -17,7 +17,7 @@ class AWSS3StoragePluginUploadDataResumabilityTests: AWSS3StoragePluginTestBase
     /// When: Call the put API and pause the operation
     /// Then: The operation is stalled (no progress, completed, or failed event)
     func testUploadLargeDataThenPause() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             Self.logger.debug("Uploading data")
             let task = try await Amplify.Storage.uploadData(key: key, data: AWSS3StoragePluginTestBase.largeDataObject)
@@ -39,7 +39,7 @@ class AWSS3StoragePluginUploadDataResumabilityTests: AWSS3StoragePluginTestBase
                 }
             }
             await waitForExpectations([didPause], timeout: TestCommonConstants.networkTimeout)
-            await waitForExpectations([didContinue])
+            await waitForExpectations([didContinue], timeout: 5)
 
             let completeInvoked = asyncExpectation(description: "Upload is completed", isInverted: true)
             let uploadTask = Task {
@@ -65,7 +65,7 @@ class AWSS3StoragePluginUploadDataResumabilityTests: AWSS3StoragePluginTestBase
     /// When: Call the put API, pause, and then resume the operation,
     /// Then: The operation should complete successfully
     func testUploadLargeDataAndPauseThenResume() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             Self.logger.debug("Uploading data")
             let task = try await Amplify.Storage.uploadData(key: key, data: AWSS3StoragePluginTestBase.largeDataObject)
@@ -112,7 +112,7 @@ class AWSS3StoragePluginUploadDataResumabilityTests: AWSS3StoragePluginTestBase
     /// When: Call the put API, pause, and then resume tthe operation,
     /// Then: The operation should complete successfully
     func testUploadLargeDataAndCancel() async throws {
-        try await testTask {
+        try await testTask(timeout: 300) {
             let key = UUID().uuidString
             Self.logger.debug("Uploading data")
             let task = try await Amplify.Storage.uploadData(key: key, data: AWSS3StoragePluginTestBase.largeDataObject)
@@ -134,7 +134,7 @@ class AWSS3StoragePluginUploadDataResumabilityTests: AWSS3StoragePluginTestBase
                 }
             }
             await waitForExpectations([didCancel], timeout: TestCommonConstants.networkTimeout)
-            await waitForExpectations([didContinue])
+            await waitForExpectations([didContinue], timeout: 5)
 
             let completeInvoked = asyncExpectation(description: "Upload is completed", isInverted: true)
             let uploadTask = Task {