YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_ws_load_test: send num_want and better offer fake type

Browse source 
This means wt-tracker starts sending offers
This commit is contained in:
Joakim Frostegård 2020-08-03 06:52:29 +02:00
parent aee97b77a1
commit 01fdf40a02
4 changed files with 15 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aquatic_ws_load_test/src/common.rs
View file

@ -5,9 +5,6 @@ use rand_distr::Pareto;
pub use aquatic_ws_protocol::*;
/// JSON string used as fake announce offer / offer response
pub const JSON_VALUE: &str = r#"{ "offer": "abcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefghabcdefgh" }"#;
#[derive(PartialEq, Eq, Clone)]
pub struct TorrentPeer {

4
aquatic_ws_load_test/src/network.rs
View file

@ -196,7 +196,7 @@ impl Connection {
if let Some((peer_id, offer_id)) = self.send_answer {
r.to_peer_id = Some(peer_id);
r.offer_id = Some(offer_id);
r.answer = Some(JsonValue(::serde_json::Value::from(JSON_VALUE)));
r.answer = Some(JsonValue(::serde_json::json!({"abc": "def"})));
r.offers = None;
}
@ -219,7 +219,7 @@ impl Connection {
false
}
Err(err) => {
eprintln!("send request error: {:?}", err);
// eprintln!("send request error: {:?}", err);
true
}

15
aquatic_ws_load_test/src/utils.rs
View file

@ -43,6 +43,7 @@ pub fn create_random_request(
}
#[inline]
fn create_announce_request(
config: &Config,
@ -59,13 +60,17 @@ fn create_announce_request(
};
let info_hash_index = select_info_hash_index(config, &state, rng);
let mut offers = Vec::with_capacity(10);
for _ in 0..10 {
const NUM_OFFERS: usize = 2;
let mut offers = Vec::with_capacity(NUM_OFFERS);
for _ in 0..NUM_OFFERS {
offers.push(AnnounceRequestOffer {
offer_id: OfferId(rng.gen()),
offer: JsonValue(::serde_json::Value::from(JSON_VALUE)),
offer: JsonValue(::serde_json::json!(
{"sdp": "abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-"}
)),
})
}
@ -79,7 +84,7 @@ fn create_announce_request(
peer_id,
bytes_left: Some(bytes_left),
event,
numwant: None,
numwant: Some(offers.len()),
offers: Some(offers),
answer: None,
to_peer_id: None,