vecna
/
lox
mirror of https://gitlab.torproject.org/vecna/lox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add endpoint to artificially advance days (for testing)

This commit is contained in:
Vecna 2023-09-08 15:39:50 -04:00
parent 4f67f033c7
commit 3941dbed35
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
crates/lox-distributor/src/lox_context.rs
View File

@ -113,7 +113,7 @@ impl LoxServerContext {
ba_obj.bridge_update(&bridgeline)
}
#[cfg(test)]
//#[cfg(test)]
/// For testing only: manually advance the day by the given number
/// of days.
pub fn advance_days_test(&self, num: u16) {
@ -388,6 +388,18 @@ impl LoxServerContext {
}
}
}
//#[cfg(test)]
/// For testing only: manually advance the day by the given number
/// of days and send back the current day.
pub fn advance_days_with_response_test(self, request: Bytes) -> Response<Body> {
let req: u16 = match serde_json::from_slice(&request) {
Ok(req) => req,
Err(e) => return prepare_error_header(e.to_string()),
};
self.advance_days_test(req);
self.send_today()
}
}
fn prepare_header(response: String) -> Response<Body> {

5
crates/lox-distributor/src/request_handler.rs
View File

@ -55,6 +55,11 @@ pub async fn handle(
let bytes = body::to_bytes(req.into_body()).await.unwrap();
cloned_context.verify_and_send_blockage_migration(bytes)
}),
//#[cfg(test)]
(&Method::POST, "/advancedays") => Ok::<_, Infallible>({
let bytes = body::to_bytes(req.into_body()).await.unwrap();
cloned_context.advance_days_with_response_test(bytes)
}),
_ => {
// Return 404 not found response.
Ok(Response::builder()