add: generic vacuum test for all storages

geofmureithi · Jan 11, 2025 · 9277b79 · 9277b79
1 parent 3fc8d77
commit 9277b79
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/packages/apalis-core/src/lib.rs b/packages/apalis-core/src/lib.rs
@@ -354,6 +354,25 @@ pub mod test_utils {
                 // assert_eq!(res, 0);
                 t.vacuum().await.unwrap();
             }
+            #[tokio::test]
+            async fn integration_test_storage_vacuum() {
+                let backend = $setup().await;
+                let service = apalis_test_service_fn(|request: Request<u32, _>| async move {
+                    Ok::<_, io::Error>(request.args)
+                });
+                let (mut t, poller) = TestWrapper::new_with_service(backend, service);
+                tokio::spawn(poller);
+                let res = t.len().await.unwrap();
+                assert_eq!(res, 0); // No jobs
+                t.push(1).await.unwrap();
+                let res = t.len().await.unwrap();
+                assert_eq!(res, 1); // A job exists
+                let res = t.execute_next().await;
+                assert_eq!(res.1, Ok("1".to_owned()));                
+                t.vacuum().await.unwrap();
+                let res = t.len().await.unwrap();
+                assert_eq!(res, 0); // After vacuuming, there should be nothing
+            }
         };
     }
 }