From e2166a7ec3f955998e7626966848f8e7ffe36361 Mon Sep 17 00:00:00 2001
From: Paul Gallagher <pgal@google.com>
Date: Fri, 7 Apr 2023 08:51:27 -0400
Subject: [PATCH] fix streaming shadow test flake

Signed-off-by: Paul Gallagher <pgal@google.com>
---
 test/integration/shadow_policy_integration_test.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/integration/shadow_policy_integration_test.cc b/test/integration/shadow_policy_integration_test.cc
index 94dba9ac5fd1..a3d4ede668c7 100644
--- a/test/integration/shadow_policy_integration_test.cc
+++ b/test/integration/shadow_policy_integration_test.cc
@@ -459,6 +459,8 @@ TEST_P(ShadowPolicyIntegrationTest, RequestMirrorPolicyWithShadowOnlyTimeout) {
   // Clean up.
   ASSERT_TRUE(fake_upstream_connection_main->close());
   ASSERT_TRUE(fake_upstream_connection_main->waitForDisconnect());
+  ASSERT_TRUE(fake_upstream_connection_shadow->close());
+  ASSERT_TRUE(fake_upstream_connection_shadow->waitForDisconnect());
 
   cleanupUpstreamAndDownstream();