Fix failing integration test

Author: thisisabhash

Tests/IntegrationTestApp/IntegrationTestAppTests/APIKeyTests.swift

@@ -138,6 +138,9 @@ final class APIKeyTests: IntegrationTestBase {
     }
 
     func testMaxSubscriptionReached() async throws {
+        let subscriptionLimit = 200
+        let failedSubscriptionCount = 5
+        
         let configuration = try AWSAppSyncConfiguration(with: .amplifyOutputs)
         let store = ApolloStore(cache: InMemoryNormalizedCache())
         let authorizer = APIKeyAuthorizer(apiKey: configuration.apiKey ?? "")
@@ -148,13 +151,12 @@ final class APIKeyTests: IntegrationTestBase {
         let websocket = AppSyncWebSocketClient(endpointURL: configuration.endpoint,
                                                authorizer: authorizer)
         let receivedConnection = expectation(description: "received connection")
-        receivedConnection.expectedFulfillmentCount = 200
+        receivedConnection.expectedFulfillmentCount = subscriptionLimit
 
         let receivedMaxSubscriptionsReachedError = expectation(description: "received MaxSubscriptionsReachedError")
-        receivedMaxSubscriptionsReachedError.expectedFulfillmentCount = 5
+        receivedMaxSubscriptionsReachedError.expectedFulfillmentCount = failedSubscriptionCount
 
         let sink = websocket.publisher.sink { event in
-            print("Received event: \(event)")
             if case .string(let message) = event {
                 if message.contains("start_ack") {
                     receivedConnection.fulfill()
@@ -171,16 +173,18 @@ final class APIKeyTests: IntegrationTestBase {
             webSocketNetworkTransport: webSocketTransport
         )
         let client = ApolloClient(networkTransport: splitTransport, store: store)
-
+        
+        try await Task.sleep(nanoseconds: 5 * 1_000_000_000) // 5 seconds
+        
         var cancellables = [Cancellable]()
-        for _ in 1...205 {
+        for _ in 1...subscriptionLimit + failedSubscriptionCount {
             cancellables.append(client.subscribe(subscription: OnCreateSubscription()) { _ in })
         }
-
-        await fulfillment(of: [receivedConnection, receivedMaxSubscriptionsReachedError], timeout: 30)
+        
+        await fulfillment(of: [receivedConnection, receivedMaxSubscriptionsReachedError], timeout: 10)
+        
         for cancellable in cancellables {
             cancellable.cancel()
         }
     }
-
 }