Bug fix

test/core/memory_usage/client.cc

@@ -29,6 +29,7 @@
 #include "src/core/lib/gpr/env.h"
 #include "src/core/lib/gpr/string.h"
 #include "src/core/lib/gpr/useful.h"
+#include "src/core/lib/surface/init.h"
 
 #include "test/core/util/cmdline.h"
 #include "test/core/util/memory_counters.h"
@@ -286,6 +287,7 @@ int main(int argc, char** argv) {
 
   grpc_completion_queue_destroy(cq);
   grpc_shutdown();
+  grpc_maybe_wait_for_async_shutdown();
 
   gpr_log(GPR_INFO, "---------client stats--------");
   gpr_log(

test/core/memory_usage/server.cc

@@ -34,6 +34,7 @@
 #include <grpc/support/time.h>
 
 #include "src/core/lib/gpr/host_port.h"
+#include "src/core/lib/surface/init.h"
 #include "test/core/end2end/data/ssl_test_data.h"
 #include "test/core/util/cmdline.h"
 #include "test/core/util/memory_counters.h"
@@ -319,6 +320,7 @@ int main(int argc, char** argv) {
   grpc_server_destroy(server);
   grpc_completion_queue_destroy(cq);
   grpc_shutdown();
+  grpc_maybe_wait_for_async_shutdown();
   grpc_memory_counters_destroy();
   return 0;
 }