From af7d1fdce62a5ddb109550569a291d3e286a2fba Mon Sep 17 00:00:00 2001
From: cospectrum <severinalexeyv@gmail.com>
Date: Sat, 6 Jan 2024 21:44:49 +0300
Subject: [PATCH] refactor

---
 memcrab-protocol/src/lib.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/memcrab-protocol/src/lib.rs b/memcrab-protocol/src/lib.rs
index c113c0d..2a205e2 100644
--- a/memcrab-protocol/src/lib.rs
+++ b/memcrab-protocol/src/lib.rs
@@ -4,8 +4,8 @@ use tokio::net::tcp::OwnedReadHalf;
 
 type Bytes = Vec<u8>;
 
-fn four_bytes_to_usize(bytes: &Bytes) -> usize {
-    debug_assert_eq!(bytes.len(), 4, "Are you stupid?");
+fn four_bytes_to_usize(bytes: &[u8]) -> usize {
+    assert_eq!(bytes.len(), 4);
 
     (bytes[0] as usize) << 24
         | (bytes[1] as usize) << 16
@@ -102,7 +102,7 @@ mod tests {
     #[tokio::test]
     async fn test_get() {
         let mock_reader = MockReader::new( 
-            vec![vec![0, 2, 1, 1].into()],
+            vec![vec![0, 2, 1, 1]],
         );
         let mut producer = Producer::new(mock_reader);
 
@@ -112,7 +112,7 @@ mod tests {
 
     #[tokio::test]
     async fn test_get_with_partitions() {
-        let mock_reader = MockReader::new(vec![vec![0].into(), vec![3, 1].into(), vec![2, 3].into()]);
+        let mock_reader = MockReader::new(vec![vec![0], vec![3, 1], vec![2, 3]]);
         let mut producer = Producer::new(mock_reader);
 
         let msg = producer.next_msg().await;
@@ -122,7 +122,7 @@ mod tests {
     #[tokio::test]
     async fn test_set() {
         let mock_reader = MockReader::new(
-            vec![vec![1, 1, 1, 0, 0, 0, 3, 8, 8, 8]].into(),
+            vec![vec![1, 1, 1, 0, 0, 0, 3, 8, 8, 8]],
         );
         let mut producer = Producer::new(mock_reader);