From 795d50d346761c458ab7149dc1e7de35524f5e28 Mon Sep 17 00:00:00 2001
From: phoenix <mail@kundeng.us>
Date: Thu, 11 Jun 2026 17:56:03 -0400
Subject: [PATCH] Cleanup

---
 tests/tests.rs | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/tests/tests.rs b/tests/tests.rs
index 83b5b96..bd2cba1 100644
--- a/tests/tests.rs
+++ b/tests/tests.rs
@@ -194,13 +194,16 @@ async fn register_user(
                     response.status()
                 )))
             } else {
-                match axum::body::to_bytes(response.into_body(), usize::MAX).await {
-                    Ok(body) => {
-                        let parsed_body: callers::register::response::Response =
-                            serde_json::from_slice(&body).unwrap();
-                        Ok(parsed_body.data[0].clone())
+                match convert_response::<callers::register::response::Response>(response).await {
+                    Ok(response) => {
+                        if response.data.len() > 0 {
+                            let user = response.data[0].clone();
+                            Ok(user)
+                        } else {
+                            Err(std::io::Error::other("No data returned"))
+                        }
                     }
-                    Err(err) => Err(std::io::Error::other(err.to_string())),
+                    Err(err) => Err(err),
                 }
             }
         }