improve the count of the number of tasks in a batch

This commit is contained in:
Tamo 2024-11-20 17:48:26 +01:00
parent 71807cac6d
commit 1e694ae432
No known key found for this signature in database
GPG Key ID: 20CD8020AFA88D69
2 changed files with 8 additions and 3 deletions

View File

@ -61,9 +61,9 @@ impl ProcessingBatch {
/// Update itself with the content of the task and update the batch id in the task.
pub fn processing<'a>(&mut self, tasks: impl IntoIterator<Item = &'a mut Task>) {
self.stats.total_nb_tasks += 1;
for task in tasks.into_iter() {
self.stats.total_nb_tasks += 1;
task.batch_uid = Some(self.uid);
// We don't store the statuses in the map since they're all enqueued but we must
// still store them in the stats since that can be displayed.

View File

@ -46,7 +46,12 @@ async fn list_batches() {
.await;
let (response, code) = index.list_batches().await;
assert_eq!(code, 200);
assert_eq!(response["results"].as_array().unwrap().len(), 2);
assert_eq!(
response["results"].as_array().unwrap().len(),
2,
"{}",
serde_json::to_string_pretty(&response).unwrap()
);
}
#[actix_rt::test]