From 3ba4f56221c0e5cb23fc3bc5addf2d77a58133f7 Mon Sep 17 00:00:00 2001 From: Nils Andreas Svee Date: Tue, 28 Apr 2026 23:34:28 +0700 Subject: [PATCH] feat(client): add run status and packet count helpers --- crates/irtt-client/src/client.rs | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/crates/irtt-client/src/client.rs b/crates/irtt-client/src/client.rs index 5ea3455..671be77 100644 --- a/crates/irtt-client/src/client.rs +++ b/crates/irtt-client/src/client.rs @@ -322,6 +322,22 @@ impl Client { Ok(events) } + pub(crate) fn is_run_complete(&self) -> bool { + let Some(session) = self.session.as_ref() else { + return matches!( + self.phase, + ClientPhase::Closed | ClientPhase::NoTestCompleted + ); + }; + session.sending_done && session.pending.len() == 0 + } + + pub(crate) fn packets_sent(&self) -> u64 { + self.session + .as_ref() + .map_or(0, |session| session.packets_sent) + } + fn process_received_packet( &mut self, packet: &[u8],