From d578f354335a704f33df809092582c7b0d94a10c Mon Sep 17 00:00:00 2001
From: Vincent Jiang <vjiang@confluent.io>
Date: Thu, 3 Mar 2022 10:41:47 -0800
Subject: [PATCH] KAFKA-13706: remove closed connections from
 MockSelector.ready

---
 clients/src/test/java/org/apache/kafka/test/MockSelector.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/clients/src/test/java/org/apache/kafka/test/MockSelector.java b/clients/src/test/java/org/apache/kafka/test/MockSelector.java
index 2d7c1517d82c7..3560c3535aa73 100644
--- a/clients/src/test/java/org/apache/kafka/test/MockSelector.java
+++ b/clients/src/test/java/org/apache/kafka/test/MockSelector.java
@@ -83,6 +83,7 @@ public void close(String id) {
 
         removeSendsForNode(id, completedSends);
         removeSendsForNode(id, initiatedSends);
+        ready.remove(id);
 
         for (int i = 0; i < this.connected.size(); i++) {
             if (this.connected.get(i).equals(id)) {