Skip to content

Commit 5413e59

Browse files
committed
refactor: [torrust#1495] renamings to follow latest changes in torrent-repository pkg
1 parent 6f5cb27 commit 5413e59

File tree

4 files changed

+130
-127
lines changed

4 files changed

+130
-127
lines changed

packages/torrent-repository/tests/integration.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
use torrust_tracker_clock::clock;
88

99
pub mod common;
10-
mod entry;
11-
mod repository;
10+
mod swarm;
11+
mod swarms;
1212

1313
/// This code needs to be copied into each crate.
1414
/// Working version, for production.

packages/torrent-repository/tests/entry/mod.rs renamed to packages/torrent-repository/tests/swarm/mod.rs

Lines changed: 64 additions & 64 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ use crate::common::torrent_peer_builder::{a_completed_peer, a_started_peer};
1515
use crate::CurrentClock;
1616

1717
#[fixture]
18-
fn single() -> Swarm {
18+
fn swarm() -> Swarm {
1919
Swarm::default()
2020
}
2121

@@ -47,39 +47,39 @@ pub enum Makes {
4747
Three,
4848
}
4949

50-
fn make(torrent: &mut Swarm, makes: &Makes) -> Vec<Peer> {
50+
fn make(swarm: &mut Swarm, makes: &Makes) -> Vec<Peer> {
5151
match makes {
5252
Makes::Empty => vec![],
5353
Makes::Started => {
5454
let peer = a_started_peer(1);
55-
torrent.handle_announcement(&peer);
55+
swarm.handle_announcement(&peer);
5656
vec![peer]
5757
}
5858
Makes::Completed => {
5959
let peer = a_completed_peer(2);
60-
torrent.handle_announcement(&peer);
60+
swarm.handle_announcement(&peer);
6161
vec![peer]
6262
}
6363
Makes::Downloaded => {
6464
let mut peer = a_started_peer(3);
65-
torrent.handle_announcement(&peer);
65+
swarm.handle_announcement(&peer);
6666
peer.event = AnnounceEvent::Completed;
6767
peer.left = NumberOfBytes::new(0);
68-
torrent.handle_announcement(&peer);
68+
swarm.handle_announcement(&peer);
6969
vec![peer]
7070
}
7171
Makes::Three => {
7272
let peer_1 = a_started_peer(1);
73-
torrent.handle_announcement(&peer_1);
73+
swarm.handle_announcement(&peer_1);
7474

7575
let peer_2 = a_completed_peer(2);
76-
torrent.handle_announcement(&peer_2);
76+
swarm.handle_announcement(&peer_2);
7777

7878
let mut peer_3 = a_started_peer(3);
79-
torrent.handle_announcement(&peer_3);
79+
swarm.handle_announcement(&peer_3);
8080
peer_3.event = AnnounceEvent::Completed;
8181
peer_3.left = NumberOfBytes::new(0);
82-
torrent.handle_announcement(&peer_3);
82+
swarm.handle_announcement(&peer_3);
8383
vec![peer_1, peer_2, peer_3]
8484
}
8585
}
@@ -88,10 +88,10 @@ fn make(torrent: &mut Swarm, makes: &Makes) -> Vec<Peer> {
8888
#[rstest]
8989
#[case::empty(&Makes::Empty)]
9090
#[tokio::test]
91-
async fn it_should_be_empty_by_default(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
92-
make(&mut torrent, makes);
91+
async fn it_should_be_empty_by_default(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
92+
make(&mut swarm, makes);
9393

94-
assert_eq!(torrent.len(), 0);
94+
assert_eq!(swarm.len(), 0);
9595
}
9696

9797
#[rstest]
@@ -102,33 +102,33 @@ async fn it_should_be_empty_by_default(#[values(single())] mut torrent: Swarm, #
102102
#[case::three(&Makes::Three)]
103103
#[tokio::test]
104104
async fn it_should_check_if_entry_should_be_retained_based_on_the_tracker_policy(
105-
#[values(single())] mut torrent: Swarm,
105+
#[values(swarm())] mut swarm: Swarm,
106106
#[case] makes: &Makes,
107107
#[values(policy_none(), policy_persist(), policy_remove(), policy_remove_persist())] policy: TrackerPolicy,
108108
) {
109-
make(&mut torrent, makes);
109+
make(&mut swarm, makes);
110110

111-
let has_peers = !torrent.is_empty();
112-
let has_downloads = torrent.metadata().downloaded != 0;
111+
let has_peers = !swarm.is_empty();
112+
let has_downloads = swarm.metadata().downloaded != 0;
113113

114114
match (policy.remove_peerless_torrents, policy.persistent_torrent_completed_stat) {
115115
// remove torrents without peers, and keep completed download stats
116116
(true, true) => match (has_peers, has_downloads) {
117117
// no peers, but has downloads
118118
// peers, with or without downloads
119-
(false, true) | (true, true | false) => assert!(torrent.meets_retaining_policy(&policy)),
119+
(false, true) | (true, true | false) => assert!(swarm.meets_retaining_policy(&policy)),
120120
// no peers and no downloads
121-
(false, false) => assert!(!torrent.meets_retaining_policy(&policy)),
121+
(false, false) => assert!(!swarm.meets_retaining_policy(&policy)),
122122
},
123123
// remove torrents without peers and drop completed download stats
124124
(true, false) => match (has_peers, has_downloads) {
125125
// peers, with or without downloads
126-
(true, true | false) => assert!(torrent.meets_retaining_policy(&policy)),
126+
(true, true | false) => assert!(swarm.meets_retaining_policy(&policy)),
127127
// no peers and with or without downloads
128-
(false, true | false) => assert!(!torrent.meets_retaining_policy(&policy)),
128+
(false, true | false) => assert!(!swarm.meets_retaining_policy(&policy)),
129129
},
130130
// keep torrents without peers, but keep or drop completed download stats
131-
(false, true | false) => assert!(torrent.meets_retaining_policy(&policy)),
131+
(false, true | false) => assert!(swarm.meets_retaining_policy(&policy)),
132132
}
133133
}
134134

@@ -139,10 +139,10 @@ async fn it_should_check_if_entry_should_be_retained_based_on_the_tracker_policy
139139
#[case::downloaded(&Makes::Downloaded)]
140140
#[case::three(&Makes::Three)]
141141
#[tokio::test]
142-
async fn it_should_get_peers_for_torrent_entry(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
143-
let peers = make(&mut torrent, makes);
142+
async fn it_should_get_peers_for_torrent_entry(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
143+
let peers = make(&mut swarm, makes);
144144

145-
let torrent_peers = torrent.peers(None);
145+
let torrent_peers = swarm.peers(None);
146146

147147
assert_eq!(torrent_peers.len(), peers.len());
148148

@@ -158,15 +158,15 @@ async fn it_should_get_peers_for_torrent_entry(#[values(single())] mut torrent:
158158
#[case::downloaded(&Makes::Downloaded)]
159159
#[case::three(&Makes::Three)]
160160
#[tokio::test]
161-
async fn it_should_update_a_peer(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
162-
make(&mut torrent, makes);
161+
async fn it_should_update_a_peer(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
162+
make(&mut swarm, makes);
163163

164164
// Make and insert a new peer.
165165
let mut peer = a_started_peer(-1);
166-
torrent.handle_announcement(&peer);
166+
swarm.handle_announcement(&peer);
167167

168168
// Get the Inserted Peer by Id.
169-
let peers = torrent.peers(None);
169+
let peers = swarm.peers(None);
170170
let original = peers
171171
.iter()
172172
.find(|p| peer::ReadInfo::get_id(*p) == peer::ReadInfo::get_id(&peer))
@@ -176,10 +176,10 @@ async fn it_should_update_a_peer(#[values(single())] mut torrent: Swarm, #[case]
176176

177177
// Announce "Completed" torrent download event.
178178
peer.event = AnnounceEvent::Completed;
179-
torrent.handle_announcement(&peer);
179+
swarm.handle_announcement(&peer);
180180

181181
// Get the Updated Peer by Id.
182-
let peers = torrent.peers(None);
182+
let peers = swarm.peers(None);
183183
let updated = peers
184184
.iter()
185185
.find(|p| peer::ReadInfo::get_id(*p) == peer::ReadInfo::get_id(&peer))
@@ -195,17 +195,17 @@ async fn it_should_update_a_peer(#[values(single())] mut torrent: Swarm, #[case]
195195
#[case::downloaded(&Makes::Downloaded)]
196196
#[case::three(&Makes::Three)]
197197
#[tokio::test]
198-
async fn it_should_remove_a_peer_upon_stopped_announcement(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
198+
async fn it_should_remove_a_peer_upon_stopped_announcement(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
199199
use torrust_tracker_primitives::peer::ReadInfo as _;
200200

201-
make(&mut torrent, makes);
201+
make(&mut swarm, makes);
202202

203203
let mut peer = a_started_peer(-1);
204204

205-
torrent.handle_announcement(&peer);
205+
swarm.handle_announcement(&peer);
206206

207207
// The started peer should be inserted.
208-
let peers = torrent.peers(None);
208+
let peers = swarm.peers(None);
209209
let original = peers
210210
.iter()
211211
.find(|p| p.get_id() == peer.get_id())
@@ -215,10 +215,10 @@ async fn it_should_remove_a_peer_upon_stopped_announcement(#[values(single())] m
215215

216216
// Change peer to "Stopped" and insert.
217217
peer.event = AnnounceEvent::Stopped;
218-
torrent.handle_announcement(&peer);
218+
swarm.handle_announcement(&peer);
219219

220220
// It should be removed now.
221-
let peers = torrent.peers(None);
221+
let peers = swarm.peers(None);
222222

223223
assert_eq!(
224224
peers.iter().find(|p| p.get_id() == peer.get_id()),
@@ -234,7 +234,7 @@ async fn it_should_remove_a_peer_upon_stopped_announcement(#[values(single())] m
234234
#[case::three(&Makes::Three)]
235235
#[tokio::test]
236236
async fn it_should_handle_a_peer_completed_announcement_and_update_the_downloaded_statistic(
237-
#[values(single())] mut torrent: Swarm,
237+
#[values(swarm())] mut torrent: Swarm,
238238
#[case] makes: &Makes,
239239
) {
240240
make(&mut torrent, makes);
@@ -264,19 +264,19 @@ async fn it_should_handle_a_peer_completed_announcement_and_update_the_downloade
264264
#[case::downloaded(&Makes::Downloaded)]
265265
#[case::three(&Makes::Three)]
266266
#[tokio::test]
267-
async fn it_should_update_a_peer_as_a_seeder(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
268-
let peers = make(&mut torrent, makes);
267+
async fn it_should_update_a_peer_as_a_seeder(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
268+
let peers = make(&mut swarm, makes);
269269
let completed = u32::try_from(peers.iter().filter(|p| p.is_seeder()).count()).expect("it_should_not_be_so_many");
270270

271-
let peers = torrent.peers(None);
271+
let peers = swarm.peers(None);
272272
let mut peer = **peers.first().expect("there should be a peer");
273273

274274
let is_already_non_left = peer.left == NumberOfBytes::new(0);
275275

276276
// Set Bytes Left to Zero
277277
peer.left = NumberOfBytes::new(0);
278-
torrent.handle_announcement(&peer);
279-
let stats = torrent.metadata();
278+
swarm.handle_announcement(&peer);
279+
let stats = swarm.metadata();
280280

281281
if is_already_non_left {
282282
// it was already complete
@@ -293,19 +293,19 @@ async fn it_should_update_a_peer_as_a_seeder(#[values(single())] mut torrent: Sw
293293
#[case::downloaded(&Makes::Downloaded)]
294294
#[case::three(&Makes::Three)]
295295
#[tokio::test]
296-
async fn it_should_update_a_peer_as_incomplete(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
297-
let peers = make(&mut torrent, makes);
296+
async fn it_should_update_a_peer_as_incomplete(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
297+
let peers = make(&mut swarm, makes);
298298
let incomplete = u32::try_from(peers.iter().filter(|p| !p.is_seeder()).count()).expect("it should not be so many");
299299

300-
let peers = torrent.peers(None);
300+
let peers = swarm.peers(None);
301301
let mut peer = **peers.first().expect("there should be a peer");
302302

303303
let completed_already = peer.left == NumberOfBytes::new(0);
304304

305305
// Set Bytes Left to no Zero
306306
peer.left = NumberOfBytes::new(1);
307-
torrent.handle_announcement(&peer);
308-
let stats = torrent.metadata();
307+
swarm.handle_announcement(&peer);
308+
let stats = swarm.metadata();
309309

310310
if completed_already {
311311
// now it is incomplete
@@ -322,10 +322,10 @@ async fn it_should_update_a_peer_as_incomplete(#[values(single())] mut torrent:
322322
#[case::downloaded(&Makes::Downloaded)]
323323
#[case::three(&Makes::Three)]
324324
#[tokio::test]
325-
async fn it_should_get_peers_excluding_the_client_socket(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
326-
make(&mut torrent, makes);
325+
async fn it_should_get_peers_excluding_the_client_socket(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
326+
make(&mut swarm, makes);
327327

328-
let peers = torrent.peers(None);
328+
let peers = swarm.peers(None);
329329
let mut peer = **peers.first().expect("there should be a peer");
330330

331331
let socket = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 8081);
@@ -334,14 +334,14 @@ async fn it_should_get_peers_excluding_the_client_socket(#[values(single())] mut
334334
assert_ne!(peer.peer_addr, socket);
335335

336336
// it should get the peer as it dose not share the socket.
337-
assert!(torrent.peers_excluding(&socket, None).contains(&peer.into()));
337+
assert!(swarm.peers_excluding(&socket, None).contains(&peer.into()));
338338

339339
// set the address to the socket.
340340
peer.peer_addr = socket;
341-
torrent.handle_announcement(&peer); // Add peer
341+
swarm.handle_announcement(&peer); // Add peer
342342

343343
// It should not include the peer that has the same socket.
344-
assert!(!torrent.peers_excluding(&socket, None).contains(&peer.into()));
344+
assert!(!swarm.peers_excluding(&socket, None).contains(&peer.into()));
345345
}
346346

347347
#[rstest]
@@ -351,16 +351,16 @@ async fn it_should_get_peers_excluding_the_client_socket(#[values(single())] mut
351351
#[case::downloaded(&Makes::Downloaded)]
352352
#[case::three(&Makes::Three)]
353353
#[tokio::test]
354-
async fn it_should_limit_the_number_of_peers_returned(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
355-
make(&mut torrent, makes);
354+
async fn it_should_limit_the_number_of_peers_returned(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
355+
make(&mut swarm, makes);
356356

357357
// We add one more peer than the scrape limit
358358
for peer_number in 1..=74 + 1 {
359359
let peer = a_started_peer(peer_number);
360-
torrent.handle_announcement(&peer);
360+
swarm.handle_announcement(&peer);
361361
}
362362

363-
let peers = torrent.peers(Some(TORRENT_PEERS_LIMIT));
363+
let peers = swarm.peers(Some(TORRENT_PEERS_LIMIT));
364364

365365
assert_eq!(peers.len(), 74);
366366
}
@@ -372,11 +372,11 @@ async fn it_should_limit_the_number_of_peers_returned(#[values(single())] mut to
372372
#[case::downloaded(&Makes::Downloaded)]
373373
#[case::three(&Makes::Three)]
374374
#[tokio::test]
375-
async fn it_should_remove_inactive_peers_beyond_cutoff(#[values(single())] mut torrent: Swarm, #[case] makes: &Makes) {
375+
async fn it_should_remove_inactive_peers_beyond_cutoff(#[values(swarm())] mut swarm: Swarm, #[case] makes: &Makes) {
376376
const TIMEOUT: Duration = Duration::from_secs(120);
377377
const EXPIRE: Duration = Duration::from_secs(121);
378378

379-
let peers = make(&mut torrent, makes);
379+
let peers = make(&mut swarm, makes);
380380

381381
let mut peer = a_completed_peer(-1);
382382

@@ -385,12 +385,12 @@ async fn it_should_remove_inactive_peers_beyond_cutoff(#[values(single())] mut t
385385

386386
peer.updated = now.sub(EXPIRE);
387387

388-
torrent.handle_announcement(&peer);
388+
swarm.handle_announcement(&peer);
389389

390-
assert_eq!(torrent.len(), peers.len() + 1);
390+
assert_eq!(swarm.len(), peers.len() + 1);
391391

392392
let current_cutoff = CurrentClock::now_sub(&TIMEOUT).unwrap_or_default();
393-
torrent.remove_inactive(current_cutoff);
393+
swarm.remove_inactive(current_cutoff);
394394

395-
assert_eq!(torrent.len(), peers.len());
395+
assert_eq!(swarm.len(), peers.len());
396396
}

0 commit comments

Comments
 (0)