Axum HTTP tracker: unit tests for announce service

src/http/axum_implementation/services/announce.rs

@@ -22,3 +22,194 @@ pub async fn invoke(tracker: Arc<Tracker>, info_hash: InfoHash, peer: &mut Peer)
 
     announce_data
 }
+
+#[cfg(test)]
+mod tests {
+    use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr};
+    use std::sync::Arc;
+
+    use aquatic_udp_protocol::{AnnounceEvent, NumberOfBytes};
+
+    use crate::config::{ephemeral_configuration, Configuration};
+    use crate::protocol::clock::DurationSinceUnixEpoch;
+    use crate::protocol::info_hash::InfoHash;
+    use crate::tracker::mode::Mode;
+    use crate::tracker::statistics::Keeper;
+    use crate::tracker::{peer, Tracker};
+
+    fn public_tracker() -> Tracker {
+        let mut configuration = ephemeral_configuration();
+        configuration.mode = Mode::Public;
+        tracker_factory(configuration)
+    }
+
+    fn tracker_factory(configuration: Configuration) -> Tracker {
+        // code-review: the tracker initialization is duplicated in many places. Consider make this function public.
+
+        // Initialize stats tracker
+        let (stats_event_sender, stats_repository) = Keeper::new_active_instance();
+
+        // Initialize Torrust tracker
+        match Tracker::new(&Arc::new(configuration), Some(stats_event_sender), stats_repository) {
+            Ok(tracker) => tracker,
+            Err(error) => {
+                panic!("{}", error)
+            }
+        }
+    }
+
+    fn sample_info_hash() -> InfoHash {
+        "3b245504cf5f11bbdbe1201cea6a6bf45aee1bc0".parse::<InfoHash>().unwrap()
+    }
+
+    fn sample_peer_using_ipv4() -> peer::Peer {
+        sample_peer()
+    }
+
+    fn sample_peer_using_ipv6() -> peer::Peer {
+        let mut peer = sample_peer();
+        peer.peer_addr = SocketAddr::new(
+            IpAddr::V6(Ipv6Addr::new(0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969)),
+            8080,
+        );
+        peer
+    }
+
+    fn sample_peer() -> peer::Peer {
+        peer::Peer {
+            peer_id: peer::Id(*b"-qB00000000000000000"),
+            peer_addr: SocketAddr::new(IpAddr::V4(Ipv4Addr::new(126, 0, 0, 1)), 8080),
+            updated: DurationSinceUnixEpoch::new(1_669_397_478_934, 0),
+            uploaded: NumberOfBytes(0),
+            downloaded: NumberOfBytes(0),
+            left: NumberOfBytes(0),
+            event: AnnounceEvent::Started,
+        }
+    }
+
+    mod with_tracker_in_any_mode {
+        use std::future;
+        use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr};
+        use std::sync::Arc;
+
+        use mockall::predicate::eq;
+
+        use super::{sample_peer_using_ipv4, sample_peer_using_ipv6};
+        use crate::config::ephemeral_configuration;
+        use crate::http::axum_implementation::services::announce::invoke;
+        use crate::http::axum_implementation::services::announce::tests::{public_tracker, sample_info_hash, sample_peer};
+        use crate::tracker::peer::Peer;
+        use crate::tracker::torrent::SwarmStats;
+        use crate::tracker::{statistics, AnnounceData, Tracker};
+
+        #[tokio::test]
+        async fn it_should_return_the_announce_data() {
+            let tracker = Arc::new(public_tracker());
+
+            let mut peer = sample_peer();
+
+            let announce_data = invoke(tracker.clone(), sample_info_hash(), &mut peer).await;
+
+            let expected_announce_data = AnnounceData {
+                peers: vec![],
+                swarm_stats: SwarmStats {
+                    completed: 0,
+                    seeders: 1,
+                    leechers: 0,
+                },
+                interval: tracker.config.announce_interval,
+                interval_min: tracker.config.min_announce_interval,
+            };
+
+            assert_eq!(announce_data, expected_announce_data);
+        }
+
+        #[tokio::test]
+        async fn it_should_send_the_tcp_4_announce_event_when_the_peer_uses_ipv4() {
+            let mut stats_event_sender_mock = statistics::MockEventSender::new();
+            stats_event_sender_mock
+                .expect_send_event()
+                .with(eq(statistics::Event::Tcp4Announce))
+                .times(1)
+                .returning(|_| Box::pin(future::ready(Some(Ok(())))));
+            let stats_event_sender = Box::new(stats_event_sender_mock);
+
+            let tracker = Arc::new(
+                Tracker::new(
+                    &Arc::new(ephemeral_configuration()),
+                    Some(stats_event_sender),
+                    statistics::Repo::new(),
+                )
+                .unwrap(),
+            );
+
+            let mut peer = sample_peer_using_ipv4();
+
+            let _announce_data = invoke(tracker, sample_info_hash(), &mut peer).await;
+        }
+
+        fn tracker_with_an_ipv6_external_ip(stats_event_sender: Box<dyn statistics::EventSender>) -> Tracker {
+            let mut configuration = ephemeral_configuration();
+            configuration.external_ip =
+                Some(IpAddr::V6(Ipv6Addr::new(0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969, 0x6969)).to_string());
+
+            Tracker::new(&Arc::new(configuration), Some(stats_event_sender), statistics::Repo::new()).unwrap()
+        }
+
+        fn peer_with_the_ipv4_loopback_ip() -> Peer {
+            let loopback_ip = IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1));
+            let mut peer = sample_peer();
+            peer.peer_addr = SocketAddr::new(loopback_ip, 8080);
+            peer
+        }
+
+        #[tokio::test]
+        async fn it_should_send_the_tcp_4_announce_event_when_the_peer_uses_ipv4_even_if_the_tracker_changes_the_peer_ip_to_ipv6()
+        {
+            // Tracker changes the peer IP to the tracker external IP when the peer is using the loopback IP.
+
+            // Assert that the event sent is a TCP4 event
+            let mut stats_event_sender_mock = statistics::MockEventSender::new();
+            stats_event_sender_mock
+                .expect_send_event()
+                .with(eq(statistics::Event::Tcp4Announce))
+                .times(1)
+                .returning(|_| Box::pin(future::ready(Some(Ok(())))));
+            let stats_event_sender = Box::new(stats_event_sender_mock);
+
+            let mut peer = peer_with_the_ipv4_loopback_ip();
+
+            let _announce_data = invoke(
+                tracker_with_an_ipv6_external_ip(stats_event_sender).into(),
+                sample_info_hash(),
+                &mut peer,
+            )
+            .await;
+        }
+
+        #[tokio::test]
+        async fn it_should_send_the_tcp_6_announce_event_when_the_peer_uses_ipv6_even_if_the_tracker_changes_the_peer_ip_to_ipv4()
+        {
+            let mut stats_event_sender_mock = statistics::MockEventSender::new();
+            stats_event_sender_mock
+                .expect_send_event()
+                .with(eq(statistics::Event::Tcp6Announce))
+                .times(1)
+                .returning(|_| Box::pin(future::ready(Some(Ok(())))));
+            let stats_event_sender = Box::new(stats_event_sender_mock);
+
+            let tracker = Arc::new(
+                Tracker::new(
+                    &Arc::new(ephemeral_configuration()),
+                    Some(stats_event_sender),
+                    statistics::Repo::new(),
+                )
+                .unwrap(),
+            );
+
+            let mut peer = sample_peer_using_ipv6();
+
+            let _announce_data = invoke(tracker, sample_info_hash(), &mut peer).await;
+        }
+    }
+}