leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix flaky test

Browse source
This commit is contained in:
Igor Katson 2024-08-14 15:11:41 +01:00
parent 2618f8b389
commit 98038a2db4
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
2 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crates/librqbit/src/tests/test_util.rs
View file

@ -1,7 +1,7 @@
use std::{io::Write, path::Path};
use librqbit_core::Id20;
use rand::{RngCore, SeedableRng};
use rand::{thread_rng, Rng, RngCore, SeedableRng};
use tempfile::TempDir;
pub fn create_new_file_with_random_content(path: &Path, mut size: usize) {
@ -43,8 +43,16 @@ pub struct TestPeerMetadata {
}
impl TestPeerMetadata {
pub fn good() -> Self {
Self {
server_id: 0,
max_random_sleep_ms: 0,
}
}
pub fn as_peer_id(&self) -> Id20 {
let mut peer_id = Id20::default();
thread_rng().fill(&mut peer_id.0);
peer_id.0[0] = self.server_id;
peer_id.0[1] = self.max_random_sleep_ms;
peer_id