Refactor StatsTracker

Cargo.toml

@@ -59,3 +59,6 @@ async-trait = "0.1"
 
 aquatic_udp_protocol = "0.2"
 uuid = { version = "1", features = ["v4"] }
+
+[dev-dependencies]
+mockall = "0.11"

src/lib.rs

@@ -14,6 +14,7 @@ pub mod jobs;
 pub mod logging;
 pub mod protocol;
 pub mod setup;
+pub mod stats;
 pub mod tracker;
 pub mod udp;
 

src/main.rs

@@ -1,7 +1,7 @@
 use std::sync::Arc;
 
 use log::info;
-use torrust_tracker::tracker::statistics::StatsTracker;
+use torrust_tracker::stats::setup_statistics;
 use torrust_tracker::tracker::tracker::TorrentTracker;
 use torrust_tracker::{ephemeral_instance_keys, logging, setup, static_time, Configuration};
 
@@ -23,11 +23,11 @@ async fn main() {
         }
     };
 
-    // Initialize stats tracker
-    let stats_tracker = StatsTracker::new_instance(config.tracker_usage_statistics);
+    // Initialize statistics
+    let (stats_event_sender, stats_repository) = setup_statistics(config.tracker_usage_statistics);
 
     // Initialize Torrust tracker
-    let tracker = match TorrentTracker::new(config.clone(), Box::new(stats_tracker)) {
+    let tracker = match TorrentTracker::new(config.clone(), stats_event_sender, stats_repository) {
         Ok(tracker) => Arc::new(tracker),
         Err(error) => {
             panic!("{}", error)

src/stats.rs

@@ -0,0 +1,36 @@
+use crate::statistics::{StatsRepository, StatsTracker, TrackerStatisticsEventSender};
+
+pub fn setup_statistics(tracker_usage_statistics: bool) -> (Option<Box<dyn TrackerStatisticsEventSender>>, StatsRepository) {
+    let mut stats_event_sender = None;
+
+    let mut stats_tracker = StatsTracker::new();
+
+    if tracker_usage_statistics {
+        stats_event_sender = Some(stats_tracker.run_event_listener());
+    }
+
+    (stats_event_sender, stats_tracker.stats_repository)
+}
+
+#[cfg(test)]
+mod test {
+    use crate::stats::setup_statistics;
+
+    #[tokio::test]
+    async fn should_not_send_any_event_when_statistics_are_disabled() {
+        let tracker_usage_statistics = false;
+
+        let (stats_event_sender, _stats_repository) = setup_statistics(tracker_usage_statistics);
+
+        assert!(stats_event_sender.is_none());
+    }
+
+    #[tokio::test]
+    async fn should_send_events_when_statistics_are_enabled() {
+        let tracker_usage_statistics = true;
+
+        let (stats_event_sender, _stats_repository) = setup_statistics(tracker_usage_statistics);
+
+        assert!(stats_event_sender.is_some());
+    }
+}