run swift-format and generate_linux_tests.rb

Author: dnadoba

Diff for: Tests/AsyncHTTPClientTests/HTTPConnectionPool+HTTP2StateMachineTests+XCTest.swift

@@ -30,6 +30,10 @@ extension HTTPConnectionPool_HTTP2StateMachineTests {
             ("testCancelRequestWorks", testCancelRequestWorks),
             ("testExecuteOnShuttingDownPool", testExecuteOnShuttingDownPool),
             ("testHTTP1ToHTTP2MigrationAndShutdownIfFirstConnectionIsHTTP1", testHTTP1ToHTTP2MigrationAndShutdownIfFirstConnectionIsHTTP1),
+            ("testSchedulingAndCancelingOfIdleTimeout", testSchedulingAndCancelingOfIdleTimeout),
+            ("testConnectionTimeout", testConnectionTimeout),
+            ("testConnectionEstablishmentFailure", testConnectionEstablishmentFailure),
+            ("testGoAway", testGoAway),
         ]
     }
 }

Diff for: Tests/AsyncHTTPClientTests/HTTPConnectionPool+HTTP2StateMachineTests.swift

@@ -96,7 +96,7 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
             XCTAssertEqual(action.request, .none)
             XCTAssertEqual(action.connection, .none)
         }
-        
+
         /// 4 streams are available and therefore request should be executed immediately
         for _ in 0..<4 {
             let mockRequest = MockHTTPRequest(eventLoop: el1, requiresEventLoopForChannel: true)
@@ -106,7 +106,7 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
             XCTAssertEqual(action.connection, .none)
             XCTAssertEqual(action.request, .executeRequest(request, conn, cancelTimeout: false))
         }
-        
+
         /// closing streams without any queued requests shouldn't do anything if it's *not* the last stream
         for _ in 0..<4 {
             let action = state.http2ConnectionStreamClosed(connID)
@@ -341,11 +341,11 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
         XCTAssertEqual(releaseAction.request, .none)
         XCTAssertEqual(releaseAction.connection, .closeConnection(conn1, isShutdown: .yes(unclean: true)))
     }
-    
+
     func testSchedulingAndCancelingOfIdleTimeout() {
         let elg = EmbeddedEventLoopGroup(loops: 1)
         let el1 = elg.next()
-        
+
         // establish one idle http2 connection
         let idGenerator = HTTPConnectionPool.Connection.ID.Generator()
         var http1Conns = HTTPConnectionPool.HTTP1Connections(maximumConcurrentConnections: 8, generator: idGenerator)
@@ -360,36 +360,36 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
         let connectAction = state.newHTTP2ConnectionEstablished(conn1, maxConcurrentStreams: 100)
         XCTAssertEqual(connectAction.request, .none)
         XCTAssertEqual(connectAction.connection, .scheduleTimeoutTimer(conn1ID, on: el1))
-        
+
         // execute request on idle connection
         let mockRequest1 = MockHTTPRequest(eventLoop: el1)
         let request1 = HTTPConnectionPool.Request(mockRequest1)
         let request1Action = state.executeRequest(request1)
         XCTAssertEqual(request1Action.request, .executeRequest(request1, conn1, cancelTimeout: false))
         XCTAssertEqual(request1Action.connection, .cancelTimeoutTimer(conn1ID))
-        
+
         // close stream
         let closeStream1Action = state.http2ConnectionStreamClosed(conn1ID)
         XCTAssertEqual(closeStream1Action.request, .none)
         XCTAssertEqual(closeStream1Action.connection, .scheduleTimeoutTimer(conn1ID, on: el1))
-        
+
         // execute request on idle connection with required event loop
         let mockRequest2 = MockHTTPRequest(eventLoop: el1, requiresEventLoopForChannel: true)
         let request2 = HTTPConnectionPool.Request(mockRequest2)
         let request2Action = state.executeRequest(request2)
         XCTAssertEqual(request2Action.request, .executeRequest(request2, conn1, cancelTimeout: false))
         XCTAssertEqual(request2Action.connection, .cancelTimeoutTimer(conn1ID))
-        
+
         // close stream
         let closeStream2Action = state.http2ConnectionStreamClosed(conn1ID)
         XCTAssertEqual(closeStream2Action.request, .none)
         XCTAssertEqual(closeStream2Action.connection, .scheduleTimeoutTimer(conn1ID, on: el1))
     }
-    
+
     func testConnectionTimeout() {
         let elg = EmbeddedEventLoopGroup(loops: 1)
         let el1 = elg.next()
-        
+
         // establish one idle http2 connection
         let idGenerator = HTTPConnectionPool.Connection.ID.Generator()
         var http1Conns = HTTPConnectionPool.HTTP1Connections(maximumConcurrentConnections: 8, generator: idGenerator)
@@ -404,20 +404,19 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
         let connectAction = state.newHTTP2ConnectionEstablished(conn1, maxConcurrentStreams: 100)
         XCTAssertEqual(connectAction.request, .none)
         XCTAssertEqual(connectAction.connection, .scheduleTimeoutTimer(conn1ID, on: el1))
-        
-        
+
         // let the connection timeout
         let timeoutAction = state.connectionIdleTimeout(conn1ID)
         XCTAssertEqual(timeoutAction.request, .none)
         XCTAssertEqual(timeoutAction.connection, .closeConnection(conn1, isShutdown: .no))
     }
-    
+
     func testConnectionEstablishmentFailure() {
         struct SomeError: Error, Equatable {}
-        
+
         let elg = EmbeddedEventLoopGroup(loops: 1)
         let el1 = elg.next()
-        
+
         // establish one idle http2 connection
         let idGenerator = HTTPConnectionPool.Connection.ID.Generator()
         var http1Conns = HTTPConnectionPool.HTTP1Connections(maximumConcurrentConnections: 8, generator: idGenerator)
@@ -428,12 +427,48 @@ class HTTPConnectionPool_HTTP2StateMachineTests: XCTestCase {
             requests: HTTPConnectionPool.RequestQueue()
         )
         XCTAssertEqual(migrationAction, .none)
-        
+
         let action = state.failedToCreateNewConnection(SomeError(), connectionID: conn1ID)
         XCTAssertEqual(action.request, .none)
         guard case .scheduleBackoffTimer(conn1ID, _, let eventLoop) = action.connection else {
             return XCTFail("unexpected connection action \(action.connection)")
         }
         XCTAssertEqual(eventLoop.id, el1.id)
     }
+
+    func testGoAway() {
+        let elg = EmbeddedEventLoopGroup(loops: 1)
+        let el1 = elg.next()
+
+        // establish one idle http2 connection
+        let idGenerator = HTTPConnectionPool.Connection.ID.Generator()
+        var http1Conns = HTTPConnectionPool.HTTP1Connections(maximumConcurrentConnections: 8, generator: idGenerator)
+        let conn1ID = http1Conns.createNewConnection(on: el1)
+        var state = HTTPConnectionPool.HTTP2StateMachine(idGenerator: idGenerator)
+        let migrationAction = state.migrateConnectionsFromHTTP1(
+            connections: http1Conns,
+            requests: HTTPConnectionPool.RequestQueue()
+        )
+        XCTAssertEqual(migrationAction, .none)
+        let conn1 = HTTPConnectionPool.Connection.__testOnly_connection(id: conn1ID, eventLoop: el1)
+        let connectAction = state.newHTTP2ConnectionEstablished(conn1, maxConcurrentStreams: 100)
+        XCTAssertEqual(connectAction.request, .none)
+        XCTAssertEqual(connectAction.connection, .scheduleTimeoutTimer(conn1ID, on: el1))
+
+        // execute request on idle connection
+        let mockRequest1 = MockHTTPRequest(eventLoop: el1)
+        let request1 = HTTPConnectionPool.Request(mockRequest1)
+        let request1Action = state.executeRequest(request1)
+        XCTAssertEqual(request1Action.request, .executeRequest(request1, conn1, cancelTimeout: false))
+        XCTAssertEqual(request1Action.connection, .cancelTimeoutTimer(conn1ID))
+
+        let goAwayAction = state.http2ConnectionGoAwayReceived(conn1ID)
+        XCTAssertEqual(goAwayAction.request, .none)
+        XCTAssertEqual(goAwayAction.connection, .none)
+
+        // close stream
+        let closeStream1Action = state.http2ConnectionStreamClosed(conn1ID)
+        XCTAssertEqual(closeStream1Action.request, .none)
+        XCTAssertEqual(closeStream1Action.connection, .closeConnection(conn1, isShutdown: .no))
+    }
 }