From e58e338ffd39e165bab12666b5d43ccfb48fc2b7 Mon Sep 17 00:00:00 2001
From: Scott Mitchell <scott_mitchell@apple.com>
Date: Mon, 8 Jun 2015 07:59:36 -0700
Subject: [PATCH] HTTP/2 shutdown cleanup miss

Motiviation:
https://github.com/netty/netty/pull/3865 was merged from a machine with old code. A test case that was updates was not merged.

Modifications:
- Merge the missing test case updates

Result:
Test case no longer fails.
---
 .../netty/handler/codec/http2/Http2ConnectionHandlerTest.java  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/codec-http2/src/test/java/io/netty/handler/codec/http2/Http2ConnectionHandlerTest.java b/codec-http2/src/test/java/io/netty/handler/codec/http2/Http2ConnectionHandlerTest.java
index 89966361c8..7f44126aad 100644
--- a/codec-http2/src/test/java/io/netty/handler/codec/http2/Http2ConnectionHandlerTest.java
+++ b/codec-http2/src/test/java/io/netty/handler/codec/http2/Http2ConnectionHandlerTest.java
@@ -319,6 +319,9 @@ public class Http2ConnectionHandlerTest {
             }
         }).when(future).addListener(any(GenericFutureListener.class));
         handler.close(ctx, promise);
+        if (future.isDone()) {
+            when(connection.numActiveStreams()).thenReturn(0);
+        }
         handler.closeStream(stream, future);
         // Simulate another stream close call being made after the context should already be closed.
         handler.closeStream(stream, future);