@@ -15,7 +15,7 @@ use crate::common::torrent_peer_builder::{a_completed_peer, a_started_peer};
1515use 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]
104104async 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]
236236async 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