starred/yaak
1
0
Fork 0
mirror of https://github.com/mountain-loop/yaak.git synced 2026-05-08 00:43:57 +02:00
Code Issues 8 Packages Projects Releases 10 Wiki Activity

Harden CLI test HTTP server

Browse Source
This commit is contained in:
Gregory Schier
2026-05-07 15:04:37 -07:00
parent f9ee53bc65
commit e504a2e83c
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
crates-cli/yaak-cli/tests/common/http_server.rs
View File

@@ -28,8 +28,30 @@ impl TestHttpServer {
match listener.accept() {
Ok((mut stream, _)) => {
let _ = stream.set_read_timeout(Some(Duration::from_secs(1)));
let mut request = Vec::new();
let mut request_buf = [0u8; 4096];
let _ = stream.read(&mut request_buf);
loop {
match stream.read(&mut request_buf) {
Ok(0) => break,
Ok(n) => {
request.extend_from_slice(&request_buf[..n]);
if request.windows(4).any(|window| window == b"\r\n\r\n") {
break;
}
}
Err(err)
if err.kind() == std::io::ErrorKind::WouldBlock
|| err.kind() == std::io::ErrorKind::TimedOut =>
{
break;
}
Err(_) => break,
}
}
if request.is_empty() {
continue;
}
let response = format!(
"HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\nContent-Length: {}\r\nConnection: close\r\n\r\n",
@@ -38,7 +60,6 @@ impl TestHttpServer {
let _ = stream.write_all(response.as_bytes());
let _ = stream.write_all(&body_bytes);
let _ = stream.flush();
break;
}
Err(err) if err.kind() == std::io::ErrorKind::WouldBlock => {
thread::sleep(Duration::from_millis(10));