feat: Add concurrent integration test & utils

IntegrationTests/AWSIntegrationTestUtils/ConcurrentTestHelper.swift

@@ -0,0 +1,25 @@
+//
+// Copyright Amazon.com Inc. or its affiliates.
+// All Rights Reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+/// Runs a test concurrently for the number of times specified
+///
+/// The test is run repeatedly using a Swift task group; each test run is performed as a "child task".
+/// The function returns when all test runs have completed, and rethrows if a test run throws.
+/// - Parameters:
+///   - count: The number of test runs
+///   - test: A function pointer to the test to be run
+/// - Throws: Any error thrown by one of the test runs.
+public func repeatConcurrently(count: Int, test: @escaping () async throws -> Void) async throws {
+    try await withThrowingTaskGroup(of: Void.self) { taskGroup in
+        for _ in 0..<count {
+            taskGroup.addTask {
+                try await test()
+            }
+        }
+        try await taskGroup.waitForAll()
+    }
+}

IntegrationTests/Package.swift

@@ -13,6 +13,9 @@ import PackageDescription
 // MARK: - Target dependencies
 
 extension Target.Dependency {
+    // Test utility module
+    static var awsIntegrationTestUtils: Self { "AWSIntegrationTestUtils" }
+
     // AWS modules
     static var awsClientRuntime: Self { .product(name: "AWSClientRuntime", package: "aws-sdk-swift") }
     static var awsSDKCommon: Self { .product(name: "AWSSDKCommon", package: "aws-sdk-swift") }
@@ -36,6 +39,12 @@ let package = Package(
         .iOS(.v13),
         .tvOS(.v13),
         .watchOS(.v6)
+    ],
+    targets: [
+        .target(
+            name: "AWSIntegrationTestUtils",
+            path: "./AWSIntegrationTestUtils"
+        )
     ]
 )
 
@@ -119,6 +128,7 @@ func addIntegrationTestTarget(_ name: String) {
                 .awsSDKIdentity,
                 .smithyIdentity,
                 .awsSDKCommon,
+                .awsIntegrationTestUtils,
                 .product(name: name, package: "aws-sdk-swift")
             ] + additionalDependencies.map {
                 Target.Dependency.product(name: $0, package: "aws-sdk-swift", condition: nil)

IntegrationTests/Services/AWSS3IntegrationTests/S3ConcurrentTests.swift

@@ -0,0 +1,95 @@
+//
+//  File.swift
+//  
+//
+//  Created by Yoo, Chan on 6/27/24.
+//
+
+import Foundation
+import Smithy
+import XCTest
+import AWSS3
+import AWSIntegrationTestUtils
+
+class S3ConcurrentTests: S3XCTestCase {
+    let fileSize = 50_000_000
+
+    func test_50MB_putObject() async throws {
+        let fileURL = getFileURL()
+        generateDummyTextFile(fileURL: fileURL)
+        let file = ByteStream.from(
+            fileHandle: try FileHandle(
+                forReadingFrom: fileURL
+            )
+        )
+        let objectKey = UUID().uuidString.split(separator: "-").first!.lowercased()
+        let putObjectInput = PutObjectInput(body: file, bucket: bucketName, key: objectKey)
+        _ = try await client.putObject(input: putObjectInput)
+        let headObjectInput = HeadObjectInput(bucket: bucketName, key: objectKey)
+        let contentLength = try await client.headObject(input: headObjectInput).contentLength
+        XCTAssertEqual(contentLength, fileSize)
+        deleteDummyTextFile(fileURL: fileURL)
+    }
+
+    func test_50MB_getObject() async throws {
+        let fileURL = getFileURL()
+        generateDummyTextFile(fileURL: fileURL)
+        let file = ByteStream.from(
+            fileHandle: try FileHandle(
+                forReadingFrom: fileURL
+            )
+        )
+        let objectKey = UUID().uuidString.split(separator: "-").first!.lowercased()
+        let putObjectInput = PutObjectInput(body: file, bucket: bucketName, key: objectKey)
+        _ = try await client.putObject(input: putObjectInput)
+        let getObjectInput = GetObjectInput(bucket: bucketName, key: objectKey)
+        let retrievedData = try await client.getObject(input: getObjectInput).body.unsafelyUnwrapped.readData()
+        XCTAssertEqual(
+            try Data(contentsOf: fileURL),
+            retrievedData
+        )
+        deleteDummyTextFile(fileURL: fileURL)
+    }
+
+    func test_10x_50MB_putObject() async throws {
+        try await repeatConcurrently(count: 10, test: test_50MB_putObject)
+    }
+
+    func test_10x_50MB_getObject() async throws {
+        try await repeatConcurrently(count: 10, test: test_50MB_getObject)
+    }
+
+    // Helper functions
+
+    func getFileURL() -> URL {
+        let tempDirectoryURL = FileManager.default.temporaryDirectory
+        let uuid = UUID().uuidString.split(separator: "-").first!.lowercased()
+        return URL(fileURLWithPath: "50MiB-\(uuid)", relativeTo: tempDirectoryURL).appendingPathExtension("txt")
+    }
+
+    func generateDummyTextFile(fileURL: URL) {
+        let stringSegment = "1234567890abcdefghijklmnopqrstABCDEFGHIJKLMNOPQRST" // 50 char long string
+        guard let segmentData = stringSegment.data(using: .utf8) else {
+            XCTFail("Unable to convert string to data")
+            return
+        }
+        var wholeData = Data()
+        for _ in 1...1_000_000 {
+            wholeData.append(segmentData)
+        }
+        do {
+            try wholeData.write(to: fileURL)
+        } catch {
+            XCTFail(error.localizedDescription)
+        }
+    }
+
+    func deleteDummyTextFile(fileURL: URL) {
+        do {
+            // Delete the temporarily geenerated file
+            try FileManager.default.removeItem(at: fileURL)
+        } catch {
+            XCTFail(error.localizedDescription)
+        }
+    }
+}

IntegrationTests/Services/AWSS3IntegrationTests/S3EventStreamTests.swift

@@ -8,6 +8,7 @@
 import Foundation
 import AWSS3
 import XCTest
+import AWSIntegrationTestUtils
 
 class S3EventStreamTests: S3XCTestCase {
     private let objectKey = "integ-test-json-object"
@@ -59,4 +60,8 @@ class S3EventStreamTests: S3XCTestCase {
         let expectedOutput = "{\"id\":\"1\"}\n{\"id\":\"2\"}\n"
         XCTAssertEqual(expectedOutput, actualOutput)
     }
+
+    func test_100x_eventStreamOutput() async throws {
+        try await repeatConcurrently(count: 100, test: testEventStreamOutput)
+    }
 }

IntegrationTests/Services/AWSSTSIntegrationTests/STSPresignedRequestTests.swift

@@ -10,6 +10,7 @@ import AWSSTS
 import ClientRuntime
 import AWSClientRuntime
 import AWSSDKHTTPAuth
+import AWSIntegrationTestUtils
 
 /// Tests presigned request using STS::getCallerIdentity.
 class STSPresignedRequestTests: XCTestCase {
@@ -34,4 +35,8 @@ class STSPresignedRequestTests: XCTestCase {
         let httpResponse = try await stsConfig.httpClientEngine.send(request: presignedRequest)
         XCTAssertEqual(httpResponse.statusCode.rawValue, 200)
     }
+
+    func test_100x_getCallerIdentity() async throws {
+        try await repeatConcurrently(count: 100, test: testGetCallerIdentity)
+    }
 }