diff --git a/example_client_test.go b/example_client_test.go index 1090dd0..645bf57 100644 --- a/example_client_test.go +++ b/example_client_test.go @@ -278,6 +278,7 @@ func (session *Session) Close() error { if !session.isReady { return errAlreadyClosed } + close(session.done) err := session.channel.Close() if err != nil { return err @@ -286,7 +287,7 @@ func (session *Session) Close() error { if err != nil { return err } - close(session.done) + session.isReady = false return nil }