Update test to be more reproducible

This commit is contained in:
Mubelotix 2025-06-25 14:58:21 +02:00
parent d08e89ea3d
commit 1d3b18f774
No known key found for this signature in database
GPG Key ID: 0406DF6C3A69B942

View File

@ -343,31 +343,16 @@ async fn create_faulty_mock_raw(sender: mpsc::Sender<()>) -> (MockServer, Value)
Mock::given(method("POST")) Mock::given(method("POST"))
.and(path("/")) .and(path("/"))
.respond_with(move |req: &Request| { .respond_with(move |_req: &Request| {
let count = count.fetch_add(1, std::sync::atomic::Ordering::SeqCst); let count = count.fetch_add(1, std::sync::atomic::Ordering::SeqCst);
let req_body = match req.body_json::<Value>() {
Ok(body) => body,
Err(error) => {
return ResponseTemplate::new(400).set_body_json(json!({
"error": format!("Invalid request: {error}")
}));
}
};
if count >= 5 { if count >= 5 {
let _ = sender.try_send(()); let _ = sender.try_send(());
ResponseTemplate::new(500).set_delay(Duration::from_secs(u64::MAX)).set_body_json( ResponseTemplate::new(500)
json!({ .set_delay(Duration::from_secs(u64::MAX))
"error": "Service Unavailable", .set_body_string("Service Unavailable")
"text": req_body
}),
)
} else { } else {
ResponseTemplate::new(500).set_body_json(json!({ ResponseTemplate::new(500).set_body_string("Service Unavailable")
"error": "Service Unavailable",
"text": req_body
}))
} }
}) })
.mount(&mock_server) .mount(&mock_server)
@ -2195,7 +2180,11 @@ async fn last_error_stats() {
receiver.recv().await; receiver.recv().await;
let (response, _code) = index.filtered_batches(&[], &[], &[]).await; let (response, _code) = index.filtered_batches(&[], &[], &[]).await;
snapshot!(json_string!(response["results"][0], { ".progress" => "[ignored]", ".stats.embedder.totalCount" => "[ignored]", ".startedAt" => "[ignored]" }), @r#" snapshot!(json_string!(response["results"][0], {
".progress" => "[ignored]",
".stats.embedder.totalCount" => "[ignored]",
".startedAt" => "[ignored]"
}), @r#"
{ {
"uid": 1, "uid": 1,
"progress": "[ignored]", "progress": "[ignored]",
@ -2217,7 +2206,7 @@ async fn last_error_stats() {
"embedder": { "embedder": {
"totalCount": "[ignored]", "totalCount": "[ignored]",
"errorCount": 5, "errorCount": 5,
"lastError": "runtime error: received internal error HTTP 500 from embedding server\n - server replied with `{\"error\":\"Service Unavailable\",\"text\":\"will_error\"}`" "lastError": "runtime error: received internal error HTTP 500 from embedding server\n - server replied with `Service Unavailable`"
} }
}, },
"duration": null, "duration": null,