Skip to content

Commit ef6b319

Browse files
authored
fix coverage (#5893)
1 parent e61459b commit ef6b319

File tree

1 file changed

+10
-5
lines changed
  • quickwit/quickwit-indexing/src/source/queue_sources

1 file changed

+10
-5
lines changed

quickwit/quickwit-indexing/src/source/queue_sources/sqs_queue.rs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -321,12 +321,17 @@ pub mod test_helpers {
321321
pub async fn start_mock_sqs_get_queue_attributes_endpoint() -> (String, oneshot::Sender<()>) {
322322
let hello = warp::path!().map(|| "{}");
323323
let (tx, rx) = oneshot::channel();
324-
let server = warp::serve(hello).bind(([127, 0, 0, 1], 0)).await;
325-
let signal_future = async {
324+
let listener = tokio::net::TcpListener::bind("127.0.0.1:0")
325+
.await
326+
.expect("listener should bind");
327+
let addr = listener.local_addr().unwrap();
328+
329+
let server = warp::serve(hello).incoming(listener).graceful(async {
326330
rx.await.ok();
327-
};
328-
server.graceful(signal_future);
329-
let queue_url = "http://127.0.0.1:0/".to_string();
331+
});
332+
tokio::spawn(server.run());
333+
334+
let queue_url = format!("http://{}:{}/", addr.ip(), addr.port());
330335
(queue_url, tx)
331336
}
332337

0 commit comments

Comments
 (0)