@@ -76,7 +76,7 @@ async fn smoke_test() {
7676 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
7777
7878 // send a request and make sure the peer is requested the corresponding download
79- let peer = SecretKey :: generate ( ) . public ( ) ;
79+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
8080 let kind: DownloadKind = HashAndFormat :: raw ( Hash :: new ( [ 0u8 ; 32 ] ) ) . into ( ) ;
8181 let req = DownloadRequest :: new ( kind, vec ! [ peer] ) ;
8282 let handle = downloader. queue ( req) . await ;
@@ -101,7 +101,7 @@ async fn deduplication() {
101101 let ( downloader, _lp) =
102102 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
103103
104- let peer = SecretKey :: generate ( ) . public ( ) ;
104+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
105105 let kind: DownloadKind = HashAndFormat :: raw ( Hash :: new ( [ 0u8 ; 32 ] ) ) . into ( ) ;
106106 let mut handles = Vec :: with_capacity ( 10 ) ;
107107 for _ in 0 ..10 {
@@ -134,7 +134,7 @@ async fn cancellation() {
134134 let ( downloader, _lp) =
135135 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
136136
137- let peer = SecretKey :: generate ( ) . public ( ) ;
137+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
138138 let kind_1: DownloadKind = HashAndFormat :: raw ( Hash :: new ( [ 0u8 ; 32 ] ) ) . into ( ) ;
139139 let req = DownloadRequest :: new ( kind_1, vec ! [ peer] ) ;
140140 let handle_a = downloader. queue ( req. clone ( ) ) . await ;
@@ -175,7 +175,7 @@ async fn max_concurrent_requests_total() {
175175 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
176176
177177 // send the downloads
178- let peer = SecretKey :: generate ( ) . public ( ) ;
178+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
179179 let mut handles = Vec :: with_capacity ( 5 ) ;
180180 let mut expected_history = Vec :: with_capacity ( 5 ) ;
181181 for i in 0 ..5 {
@@ -219,7 +219,7 @@ async fn max_concurrent_requests_per_peer() {
219219 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
220220
221221 // send the downloads
222- let peer = SecretKey :: generate ( ) . public ( ) ;
222+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
223223 let mut handles = Vec :: with_capacity ( 5 ) ;
224224 for i in 0 ..5 {
225225 let kind = HashAndFormat :: raw ( Hash :: new ( [ i; 32 ] ) ) ;
@@ -275,7 +275,7 @@ async fn concurrent_progress() {
275275 let ( downloader, _lp) =
276276 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , Default :: default ( ) ) ;
277277
278- let peer = SecretKey :: generate ( ) . public ( ) ;
278+ let peer = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
279279 let hash = Hash :: new ( [ 0u8 ; 32 ] ) ;
280280 let kind_1 = HashAndFormat :: raw ( hash) ;
281281
@@ -369,11 +369,12 @@ async fn long_queue() {
369369
370370 let ( downloader, _lp) =
371371 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
372+ let mut rng = rand:: thread_rng ( ) ;
372373 // send the downloads
373374 let nodes = [
374- SecretKey :: generate ( ) . public ( ) ,
375- SecretKey :: generate ( ) . public ( ) ,
376- SecretKey :: generate ( ) . public ( ) ,
375+ SecretKey :: generate ( & mut rng ) . public ( ) ,
376+ SecretKey :: generate ( & mut rng ) . public ( ) ,
377+ SecretKey :: generate ( & mut rng ) . public ( ) ,
377378 ] ;
378379 let mut handles = vec ! [ ] ;
379380 for i in 0 ..100usize {
@@ -417,7 +418,7 @@ async fn fail_while_running() {
417418 . boxed ( )
418419 } ) ) ;
419420
420- let node = SecretKey :: generate ( ) . public ( ) ;
421+ let node = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
421422 let req_success = DownloadRequest :: new ( blob_success, vec ! [ node] ) ;
422423 let req_fail = DownloadRequest :: new ( blob_fail, vec ! [ node] ) ;
423424 let handle_success = downloader. queue ( req_success) . await ;
@@ -437,7 +438,7 @@ async fn retry_nodes_simple() {
437438 let getter = getter:: TestingGetter :: default ( ) ;
438439 let ( downloader, _lp) =
439440 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , Default :: default ( ) ) ;
440- let node = SecretKey :: generate ( ) . public ( ) ;
441+ let node = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
441442 let dial_attempts = Arc :: new ( AtomicUsize :: new ( 0 ) ) ;
442443 let dial_attempts2 = dial_attempts. clone ( ) ;
443444 // fail on first dial, then succeed
@@ -467,7 +468,7 @@ async fn retry_nodes_fail() {
467468 Default :: default ( ) ,
468469 config,
469470 ) ;
470- let node = SecretKey :: generate ( ) . public ( ) ;
471+ let node = SecretKey :: generate ( rand :: thread_rng ( ) ) . public ( ) ;
471472 // fail always
472473 dialer. set_dial_outcome ( move |_node| false ) ;
473474
@@ -504,8 +505,9 @@ async fn retry_nodes_jump_queue() {
504505 let ( downloader, _lp) =
505506 Downloader :: spawn_for_test ( dialer. clone ( ) , getter. clone ( ) , concurrency_limits) ;
506507
507- let good_node = SecretKey :: generate ( ) . public ( ) ;
508- let bad_node = SecretKey :: generate ( ) . public ( ) ;
508+ let mut rng = rand:: thread_rng ( ) ;
509+ let good_node = SecretKey :: generate ( & mut rng) . public ( ) ;
510+ let bad_node = SecretKey :: generate ( & mut rng) . public ( ) ;
509511
510512 dialer. set_dial_outcome ( move |node| node == good_node) ;
511513 let kind1 = HashAndFormat :: raw ( Hash :: new ( [ 0u8 ; 32 ] ) ) ;
0 commit comments