From d6577e555e9777d7f3ba89c6f97b2abd5d6e1161 Mon Sep 17 00:00:00 2001
From: Naim A <227396+naim94a@users.noreply.github.com>
Date: Wed, 6 May 2020 03:23:29 +0300
Subject: [PATCH] fixed broken tests

---
 src/server.rs  |  2 +-
 src/tracker.rs | 10 +++++++---
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/server.rs b/src/server.rs
index e505d3a..7648ea6 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -407,7 +407,7 @@ mod tests {
         let mut payload = StackVec::from(&mut buffer);
 
         assert!(pack_into(&mut payload, &mystruct).is_ok());
-        assert_eq!(payload.len(), 16);
+        assert_eq!(payload.as_slice().len(), 16);
         assert_eq!(payload.as_slice(), &[0, 0, 0, 0, 0, 0, 0, 200u8, 0, 0, 0, 0, 0, 1, 47, 203]);
     }
 
diff --git a/src/tracker.rs b/src/tracker.rs
index 5ad33ca..de22a3c 100644
--- a/src/tracker.rs
+++ b/src/tracker.rs
@@ -507,12 +507,16 @@ mod tests {
     #[tokio::test]
     async fn test_save_db() {
         let tracker = TorrentTracker::new(TrackerMode::DynamicMode);
-        tracker.add_torrent(&[0u8, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0].into());
+        tracker
+            .add_torrent(&[0u8, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0].into())
+            .await
+            .expect("failed to add torrent");
 
         let mut out = Vec::new();
+        let mut cursor = std::io::Cursor::new(&mut out);
 
-        tracker.save_database(&mut out).await.expect("db save failed");
-        assert!(out.len() > 0);
+        tracker.save_database(&mut cursor).await.expect("db save failed");
+        assert!(cursor.position() > 0);
     }
 
     #[test]