test(api): fix flaky integration test cases (#2787)

Author: Di Wu

AmplifyPlugins/API/Tests/APIHostApp/APIHostApp.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -49,6 +49,7 @@ class GraphQLConnectionScenario2Tests: XCTestCase {
 
     override func tearDown() async throws {
         await Amplify.reset()
+        try await Task.sleep(seconds: 1)
     }
 
     // Create Project2 in different ways, then retrieve it

AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginFunctionalTests/GraphQLConnectionScenario2Tests.swift

@@ -159,14 +159,18 @@ extension GraphQLConnectionScenario3Tests {
             XCTFail("Could not create comment")
             return
         }
+
         let predicate = Comment3.keys.postID.eq(post.id)
-        let result = try await Amplify.API.query(request: .list(Comment3.self, where: predicate))
-        switch result {
-        case .success(let comments):
-            XCTAssertEqual(comments.count, 1)
-        case .failure(let response):
-            XCTFail("Failed with: \(response)")
+        guard case .success(var comments) = try await Amplify.API.query(request: .list(Comment3.self, where: predicate))
+        else {
+            XCTFail("Failed to retrieve comments")
+            return
+        }
+
+        while comments.count == 0 && comments.hasNextPage() {
+            comments = try await comments.getNextPage()
         }
+        XCTAssertEqual(comments.count, 1)
     }
 
     /// Test paginated list query returns a List containing pagination functionality. This test also aggregates page

AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginFunctionalTests/GraphQLConnectionScenario3Tests+List.swift

@@ -49,6 +49,7 @@ class GraphQLConnectionScenario3Tests: XCTestCase {
 
     override func tearDown() async throws {
         await Amplify.reset()
+        try await Task.sleep(seconds: 1)
     }
 
     func testQuerySinglePost() async throws {

AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginFunctionalTests/GraphQLConnectionScenario3Tests.swift

@@ -21,6 +21,7 @@ class GraphQLLazyLoadBaseTest: XCTestCase {
 
     override func tearDown() async throws {
         await Amplify.reset()
+        try await Task.sleep(seconds: 1)
     }
 
     func setupConfig() {