Add retry on rate limit in unit testing

Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
This commit is contained in:
Jiahao XU 2024-06-10 23:42:18 +10:00
parent f92cd84fd0
commit b5831b2137
No known key found for this signature in database
GPG key ID: 76D1E687CA3C4928

View file

@ -551,7 +551,16 @@ mod test {
tests.push((
None,
tokio::spawn(async move { client.get_repo_info(&repo).await }),
tokio::spawn(async move {
loop {
match client.get_repo_info(&repo).await {
Err(GhApiError::RateLimit { retry_after }) => {
sleep(retry_after.unwrap_or(DEFAULT_RETRY_AFTER)).await
}
res => break res,
}
}
}),
));
}