test: refactor tests to use tls connection

Author: kezhuw

Cargo.toml

@@ -44,5 +44,4 @@ testcontainers = { git = "https://github.com/kezhuw/testcontainers-rs.git", bran
 assertor = "0.0.2"
 assert_matches = "1.5.0"
 tempfile = "3.6.0"
-maplit = "1.0.2"
 rcgen = { version = "0.12.1", features = ["default", "x509-parser"] }

src/session/mod.rs

@@ -443,7 +443,7 @@ impl Session {
         tick.set_missed_tick_behavior(time::MissedTickBehavior::Skip);
         let mut channel_closed = false;
         depot.start();
-        while !(channel_closed && depot.is_empty()) {
+        while !(channel_closed && depot.is_empty() && !conn.wants_write()) {
             select! {
                 _ = conn.readable() => {
                     self.read_connection(conn, buf)?;