searchengines-web/MeiliSearch
searchengines-web/MeiliSearch
1
0
Fork 0
mirror of https://github.com/meilisearch/MeiliSearch synced 2025-07-02 03:18:30 +02:00
Code Issues Releases Wiki Activity

Make test more reproducible

Browse Source
This commit is contained in:
Mubelotix 2025-06-24 11:08:06 +02:00
parent 2f82d94502
commit 59a1c5d9a7
No known key found for this signature in database
GPG Key ID: 89F391DBCC8CE7F0
1 changed files with 25 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
crates/meilisearch/tests/vector/rest.rs
View File

@ -1,4 +1,5 @@
use std::collections::{BTreeMap, BTreeSet};
use std::sync::atomic::AtomicUsize;
use meili_snap::{json_string, snapshot};
use reqwest::IntoUrl;
@ -338,36 +339,43 @@ async fn create_mock_raw() -> (MockServer, Value) {
(mock_server, embedder_settings)
}
/// A mock server that returns 500 errors, and sends a message once 5 requests are received
async fn create_faulty_mock_raw(mut sender: mpsc::Sender<()>) -> (MockServer, Value) {
async fn create_faulty_mock_raw(sender: mpsc::Sender<()>) -> (MockServer, Value) {
let mock_server = MockServer::start().await;
let count = AtomicUsize::new(0);
Mock::given(method("POST"))
.and(path("/"))
.respond_with(move |req: &Request| {
let req: String = match req.body_json() {
Ok(req) => req,
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}")
"error": format!("Invalid request: {error}")
}));
}
};
let sender = sender.clone();
spawn(async move {
sender.send(()).await;
});
if count >= 5 {
let _ = sender.try_send(());
ResponseTemplate::new(500)
.set_delay(Duration::from_secs(u64::MAX))
.set_body_json(json!({
"error": "Service Unavailable",
"text": req_body
}))
} else {
ResponseTemplate::new(500)
.set_delay(Duration::from_millis(500))
.set_body_json(json!({
ResponseTemplate::new(500).set_body_json(json!({
"error": "Service Unavailable",
"text": req
"text": req_body
}))
}
})
.mount(&mock_server)
.await;
let url = mock_server.uri();
let embedder_settings = json!({
@ -2187,12 +2195,9 @@ async fn observability() {
snapshot!(code, @"202 Accepted");
// The task will eventually fail, so let's not wait for it.
// Let's just wait for 5 errors from the mock server.
for _errors in 0..5 {
receiver.recv().await;
}
// Let's just wait for the server to block
receiver.recv().await;
let batches = index.filtered_batches(&[], &[], &[]).await;
println!("Batches: {batches:?}");
snapshot!(task, @r###""###);
}