1 files changed, 6 insertions, 0 deletions
diff --git a/src/transport/traffic.rs b/src/transport/traffic.rs
index b427a14..a34bf4b 100644
--- a/src/transport/traffic.rs
+++ b/src/transport/traffic.rs
@@ -345,6 +345,12 @@ impl Traffic {
     pub(crate) fn retain(&self, mut keep: impl FnMut(&str) -> bool) {
         self.lock().peers.retain(|peer, _| keep(peer));
     }
+
+    /// How many peers (and their addresses) it counts for.
+    #[cfg(test)]
+    pub(crate) fn len(&self) -> usize {
+        self.lock().peers.len()
+    }
 }
 
 /// Counts what passes through `inner`.