From df1ad5829ca99f4d6a6ce969afdd107f52760768 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20M=C3=BCller?= <deso@posteo.net>
Date: Sun, 29 Dec 2024 09:32:22 -0800
Subject: [PATCH] Introduce random_str() helper function

We already have a random_buf() function that created a buffer of random
bytes and yet we defined similar logic for creating a random string
in-line. Factor out a dedicated function, random_str(), for this
purpose.
---
 src/wrap.rs | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/wrap.rs b/src/wrap.rs
index 7560ac0..1de375e 100644
--- a/src/wrap.rs
+++ b/src/wrap.rs
@@ -754,6 +754,13 @@ mod tests {
         vec
       }
 
+      fn random_str() -> String {
+        let len = (0..32).choose(&mut thread_rng()).unwrap();
+        let mut string = String::new();
+        string.extend((0..len).map(|_| thread_rng().gen::<char>()));
+        string
+      }
+
       for _ in 0..50000 {
         let message = match (0..5).choose(&mut thread_rng()).unwrap() {
           0 => WebSocketMessage::Pong(random_buf()),
@@ -766,11 +773,7 @@ mod tests {
           // the connection.
           i => {
             if i & 0x1 == 0 {
-              let len = (0..32).choose(&mut thread_rng()).unwrap();
-              let mut string = String::new();
-              string.extend((0..len).map(|_| thread_rng().gen::<char>()));
-
-              WebSocketMessage::Text(string)
+              WebSocketMessage::Text(random_str())
             } else {
               WebSocketMessage::Binary(random_buf())
             }