Skip to content

Commit 3d0f4f8

Browse files
committed
refactor: [#1187] use AppContainer in test environments
1 parent c45a12b commit 3d0f4f8

File tree

2 files changed

+12
-28
lines changed

2 files changed

+12
-28
lines changed

tests/servers/http/environment.rs

+6-13
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,8 @@ use std::sync::Arc;
33
use bittorrent_primitives::info_hash::InfoHash;
44
use futures::executor::block_on;
55
use torrust_tracker_configuration::{Configuration, HttpTracker};
6-
use torrust_tracker_lib::bootstrap::app::initialize_global_services;
6+
use torrust_tracker_lib::bootstrap::app::{initialize_app_container, initialize_global_services};
77
use torrust_tracker_lib::bootstrap::jobs::make_rust_tls;
8-
use torrust_tracker_lib::core::services::{initialize_database, initialize_tracker, initialize_whitelist, statistics};
98
use torrust_tracker_lib::core::statistics::event::sender::Sender;
109
use torrust_tracker_lib::core::statistics::repository::Repository;
1110
use torrust_tracker_lib::core::whitelist::WhiteListManager;
@@ -34,15 +33,9 @@ impl<S> Environment<S> {
3433
impl Environment<Stopped> {
3534
#[allow(dead_code)]
3635
pub fn new(configuration: &Arc<Configuration>) -> Self {
37-
let (stats_event_sender, stats_repository) = statistics::setup::factory(configuration.core.tracker_usage_statistics);
38-
let stats_event_sender = Arc::new(stats_event_sender);
39-
let stats_repository = Arc::new(stats_repository);
40-
4136
initialize_global_services(configuration);
4237

43-
let database = initialize_database(configuration);
44-
let whitelist_manager = initialize_whitelist(database.clone());
45-
let tracker = Arc::new(initialize_tracker(configuration, &database, &whitelist_manager));
38+
let app_container = initialize_app_container(configuration);
4639

4740
let http_tracker = configuration
4841
.http_trackers
@@ -59,10 +52,10 @@ impl Environment<Stopped> {
5952

6053
Self {
6154
config,
62-
tracker,
63-
stats_event_sender,
64-
stats_repository,
65-
whitelist_manager,
55+
tracker: app_container.tracker.clone(),
56+
stats_event_sender: app_container.stats_event_sender.clone(),
57+
stats_repository: app_container.stats_repository.clone(),
58+
whitelist_manager: app_container.whitelist_manager.clone(),
6659
registar: Registar::default(),
6760
server,
6861
}

tests/servers/udp/environment.rs

+6-15
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,12 @@ use std::sync::Arc;
44
use bittorrent_primitives::info_hash::InfoHash;
55
use tokio::sync::RwLock;
66
use torrust_tracker_configuration::{Configuration, UdpTracker, DEFAULT_TIMEOUT};
7-
use torrust_tracker_lib::bootstrap::app::initialize_global_services;
8-
use torrust_tracker_lib::core::services::{initialize_database, initialize_tracker, initialize_whitelist, statistics};
7+
use torrust_tracker_lib::bootstrap::app::{initialize_app_container, initialize_global_services};
98
use torrust_tracker_lib::core::statistics::event::sender::Sender;
109
use torrust_tracker_lib::core::statistics::repository::Repository;
1110
use torrust_tracker_lib::core::Tracker;
1211
use torrust_tracker_lib::servers::registar::Registar;
1312
use torrust_tracker_lib::servers::udp::server::banning::BanService;
14-
use torrust_tracker_lib::servers::udp::server::launcher::MAX_CONNECTION_ID_ERRORS_PER_IP;
1513
use torrust_tracker_lib::servers::udp::server::spawner::Spawner;
1614
use torrust_tracker_lib::servers::udp::server::states::{Running, Stopped};
1715
use torrust_tracker_lib::servers::udp::server::Server;
@@ -44,16 +42,9 @@ where
4442
impl Environment<Stopped> {
4543
#[allow(dead_code)]
4644
pub fn new(configuration: &Arc<Configuration>) -> Self {
47-
let (stats_event_sender, stats_repository) = statistics::setup::factory(configuration.core.tracker_usage_statistics);
48-
let stats_event_sender = Arc::new(stats_event_sender);
49-
let stats_repository = Arc::new(stats_repository);
50-
let ban_service = Arc::new(RwLock::new(BanService::new(MAX_CONNECTION_ID_ERRORS_PER_IP)));
51-
5245
initialize_global_services(configuration);
5346

54-
let database = initialize_database(configuration);
55-
let whitelist_manager = initialize_whitelist(database.clone());
56-
let tracker = Arc::new(initialize_tracker(configuration, &database, &whitelist_manager));
47+
let app_container = initialize_app_container(configuration);
5748

5849
let udp_tracker = configuration.udp_trackers.clone().expect("missing UDP tracker configuration");
5950

@@ -65,10 +56,10 @@ impl Environment<Stopped> {
6556

6657
Self {
6758
config,
68-
tracker,
69-
stats_event_sender,
70-
stats_repository,
71-
ban_service,
59+
tracker: app_container.tracker.clone(),
60+
stats_event_sender: app_container.stats_event_sender.clone(),
61+
stats_repository: app_container.stats_repository.clone(),
62+
ban_service: app_container.ban_service.clone(),
7263
registar: Registar::default(),
7364
server,
7465
}

0 commit comments

Comments
 (0)