test: refactor tests to use tls connection

kezhuw · Mar 20, 2024 · 309672d · 309672d
1 parent 0b225d9
commit 309672d
Show file tree

Hide file tree

Showing 3 changed files with 319 additions and 327 deletions.
diff --git a/Cargo.toml b/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"] }
diff --git a/src/session/mod.rs b/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)?;