test: [#1358] add tests for events in torrent-repository pkg

Author: josecelano

packages/primitives/src/peer.rs

@@ -22,20 +22,21 @@
 //! };
 //! ```
 
+use std::fmt;
 use std::net::{IpAddr, SocketAddr};
 use std::ops::{Deref, DerefMut};
+use std::str::FromStr;
 use std::sync::Arc;
 
 use aquatic_udp_protocol::{AnnounceEvent, NumberOfBytes, PeerId};
-use derive_more::Display;
 use serde::Serialize;
 use zerocopy::FromBytes as _;
 
 use crate::DurationSinceUnixEpoch;
 
 pub type PeerAnnouncement = Peer;
 
-#[derive(Debug, Display, Serialize, Copy, Clone, PartialEq, Eq, Hash)]
+#[derive(Debug, Serialize, Copy, Clone, PartialEq, Eq, Hash)]
 #[serde(rename_all_fields = "lowercase")]
 pub enum PeerRole {
     Seeder,
@@ -53,6 +54,39 @@ impl PeerRole {
     }
 }
 
+impl fmt::Display for PeerRole {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            PeerRole::Seeder => write!(f, "seeder"),
+            PeerRole::Leecher => write!(f, "leecher"),
+        }
+    }
+}
+
+impl FromStr for PeerRole {
+    type Err = ParsePeerRoleError;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        match s.to_lowercase().as_str() {
+            "seeder" => Ok(PeerRole::Seeder),
+            "leecher" => Ok(PeerRole::Leecher),
+            _ => Err(ParsePeerRoleError::InvalidPeerRole {
+                location: Location::caller(),
+                raw_param: s.to_string(),
+            }),
+        }
+    }
+}
+
+#[derive(Error, Debug)]
+pub enum ParsePeerRoleError {
+    #[error("invalid param {raw_param} in {location}")]
+    InvalidPeerRole {
+        location: &'static Location<'static>,
+        raw_param: String,
+    },
+}
+
 /// Peer struct used by the core `Tracker`.
 ///
 /// A sample peer:
@@ -264,6 +298,14 @@ impl Peer {
             ..self
         }
     }
+
+    #[must_use]
+    pub fn into_seeder(self) -> Self {
+        Self {
+            left: NumberOfBytes::new(0),
+            ..self
+        }
+    }
 }
 
 use std::panic::Location;

packages/torrent-repository/src/event.rs

@@ -83,3 +83,29 @@ pub mod bus {
 
     pub type EventBus = torrust_tracker_events::bus::EventBus<Event>;
 }
+
+#[cfg(test)]
+pub mod test {
+
+    use torrust_tracker_primitives::peer::Peer;
+
+    use super::Event;
+    use crate::tests::sample_info_hash;
+
+    #[test]
+    fn events_should_be_comparable() {
+        let info_hash = sample_info_hash();
+
+        let event1 = Event::TorrentAdded {
+            info_hash,
+            announcement: Peer::default(),
+        };
+
+        let event2 = Event::TorrentRemoved { info_hash };
+
+        let event1_clone = event1.clone();
+
+        assert!(event1 == event1_clone);
+        assert!(event1 != event2);
+    }
+}