Fixes test after the changes for #960.

jetty-http2/http2-client/src/test/java/org/eclipse/jetty/http2/client/StreamCloseTest.java

@@ -23,6 +23,7 @@ import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicReference;
 
 import org.eclipse.jetty.http.HttpFields;
 import org.eclipse.jetty.http.HttpVersion;
@@ -308,6 +309,7 @@ public class StreamCloseTest extends AbstractTest
     @Test
     public void testFailedSessionClosesIdleStream() throws Exception
     {
+        AtomicReference<Session> sessionRef = new AtomicReference<>();
         final CountDownLatch latch = new CountDownLatch(1);
         final List<Stream> streams = new ArrayList<>();
         start(new ServerSessionListener.Adapter()
@@ -329,9 +331,7 @@ public class StreamCloseTest extends AbstractTest
             @Override
             public void onFailure(Session session, Throwable failure)
             {
-                Assert.assertEquals(0, session.getStreams().size());
-                for (Stream stream : streams)
-                    Assert.assertTrue(stream.isClosed());
+                sessionRef.set(session);
                 latch.countDown();
             }
         });
@@ -347,5 +347,13 @@ public class StreamCloseTest extends AbstractTest
         session.newStream(request2, new Promise.Adapter<>(), new Stream.Listener.Adapter());
 
         Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
+        Session serverSession = sessionRef.get();
+
+        // Wait for the server to finish the close activities.
+        Thread.sleep(1000);
+
+        Assert.assertEquals(0, serverSession.getStreams().size());
+        for (Stream stream : streams)
+            Assert.assertTrue(stream.isClosed());
     }
 }

jetty-http2/http2-common/src/main/java/org/eclipse/jetty/http2/HTTP2Session.java

@@ -462,8 +462,8 @@ public abstract class HTTP2Session extends ContainerLifeCycle implements ISessio
     @Override
     public void onConnectionFailure(int error, String reason)
     {
-        close(error, reason, Callback.NOOP);
         notifyFailure(this, new IOException(String.format("%d/%s", error, reason)));
+        close(error, reason, Callback.NOOP);
     }
 
     @Override