8372147: ConnectionFlowControlTest should use HttpResponse.connectionLabel()

Reviewed-by: mdoerr
Backport-of: 78b155b

Author: GoeLin

test/jdk/java/net/httpclient/http2/ConnectionFlowControlTest.java

@@ -27,7 +27,7 @@
  * @summary checks connection flow control
  * @library /test/lib /test/jdk/java/net/httpclient/lib
  * @build jdk.httpclient.test.lib.http2.Http2TestServer jdk.test.lib.net.SimpleSSLContext
- * @run junit/othervm   -Djdk.internal.httpclient.debug=true
+ * @run junit/othervm   -Djdk.internal.httpclient.debug=err
  *                      -Djdk.httpclient.connectionWindowSize=65535
  *                      -Djdk.httpclient.windowsize=16384
  *                      ConnectionFlowControlTest
@@ -122,7 +122,7 @@ void test(String uri) throws Exception {
                     final HttpClient cc = client;
                     var response = cc.send(request, BodyHandlers.ofInputStream());
                     responses.put(query, response);
-                    String ckey = response.headers().firstValue("X-Connection-Key").get();
+                    String ckey = response.connectionLabel().get();
                     if (label == null) label = ckey;
                     try {
                         if (i < max - 1) {
@@ -149,7 +149,7 @@ void test(String uri) throws Exception {
                     try {
                         String query = keys[i];
                         var response = responses.get(keys[i]);
-                        String ckey = response.headers().firstValue("X-Connection-Key").get();
+                        String ckey = response.connectionLabel().get();
                         if (label == null) label = ckey;
                         if (i < max - 1) {
                             // the connection window might be exceeded at i == max - 2, which
@@ -206,7 +206,7 @@ void test(String uri) throws Exception {
             System.out.println("\nSending last request:" + uriWithQuery);
             var response = client.send(request, BodyHandlers.ofString());
             if (label != null) {
-                String ckey = response.headers().firstValue("X-Connection-Key").get();
+                String ckey = response.connectionLabel().get();
                 assertNotEquals(label, ckey);
                 System.out.printf("last request %s sent on different connection as expected:" +
                         "\n\tlast: %s\n\tprevious: %s%n", query, ckey, label);
@@ -281,7 +281,6 @@ public void handle(Http2TestExchange t) throws IOException {
                 byte[] bytes = is.readAllBytes();
                 System.out.println("Server " + t.getLocalAddress() + " received:\n"
                         + t.getRequestURI() + ": " + new String(bytes, StandardCharsets.UTF_8));
-                t.getResponseHeaders().setHeader("X-Connection-Key", t.getConnectionKey());
 
                 if (bytes.length == 0) bytes = "no request body!".getBytes(StandardCharsets.UTF_8);
                 int window = Math.max(16384, Integer.getInteger("jdk.httpclient.windowsize", 2*16*1024));