From 73dfcc46c3250fa4f87518d744882cb178ea510c Mon Sep 17 00:00:00 2001
From: Dan Sosedoff <dan.sosedoff@gmail.com>
Date: Fri, 2 Dec 2022 13:59:22 -0600
Subject: [PATCH] Verify client is closed

---
 pkg/api/session_manager_test.go | 1 +
 pkg/client/client.go            | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/pkg/api/session_manager_test.go b/pkg/api/session_manager_test.go
index 9387592..cd7ef55 100644
--- a/pkg/api/session_manager_test.go
+++ b/pkg/api/session_manager_test.go
@@ -75,5 +75,6 @@ func TestSessionManager(t *testing.T) {
 		command.Opts.ConnectionIdleTimeout = 1
 		assert.Equal(t, 1, manager.Cleanup())
 		assert.Equal(t, 0, manager.Len())
+		assert.True(t, conn.IsClosed())
 	})
 }
diff --git a/pkg/client/client.go b/pkg/client/client.go
index d58365d..f24272c 100644
--- a/pkg/client/client.go
+++ b/pkg/client/client.go
@@ -442,6 +442,10 @@ func (client *Client) Close() error {
 	return nil
 }
 
+func (c *Client) IsClosed() bool {
+	return c.closed
+}
+
 func (client *Client) IsIdle() bool {
 	mins := int(time.Since(client.lastQueryTime).Minutes())