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())