Test Client::discard

tests/real/community.rs

@@ -889,7 +889,7 @@ async fn mutation_requeue_jobs() {
 }
 
 #[tokio::test(flavor = "multi_thread")]
-async fn mutation_kill_and_requeue() {
+async fn mutation_kill_and_requeue_and_discard() {
     skip_check!();
 
     // prepare a client and clean up the queue
@@ -965,6 +965,76 @@ async fn mutation_kill_and_requeue() {
         .map(|v| *v)
         .unwrap_or_default();
     assert_eq!(njobs, 2);
+
+    // prepare a worker that will fail the job unconditionally
+    let mut worker = Worker::builder::<io::Error>()
+        .register_fn(local, move |_job| async move {
+            panic!("force fail this job");
+        })
+        .connect()
+        .await
+        .unwrap();
+
+    // cosume them (and immediately fail them) and make
+    // sure the queue is drained
+    assert!(worker.run_one(0, &[local]).await.unwrap());
+    assert!(worker.run_one(0, &[local]).await.unwrap());
+    assert!(!worker.run_one(0, &[local]).await.unwrap());
+    let njobs = client
+        .current_info()
+        .await
+        .unwrap()
+        .data
+        .queues
+        .get(local)
+        .map(|v| *v)
+        .unwrap_or_default();
+    assert_eq!(njobs, 0); // sanity check
+
+    // so the jobs have transitioned from being enqueued
+    // to the `retries` set, and we can now completely discard them
+    client
+        .discard(
+            MutationTarget::Retries,
+            MutationFilter::builder().kind(local).build(),
+        )
+        .await
+        .unwrap();
+
+    // Double-check
+    client
+        .requeue(
+            MutationTarget::Retries,
+            MutationFilter::builder().kind(local).build(),
+        )
+        .await
+        .unwrap();
+    client
+        .requeue(
+            MutationTarget::Dead,
+            MutationFilter::builder().kind(local).build(),
+        )
+        .await
+        .unwrap();
+    client
+        .requeue(
+            MutationTarget::Scheduled,
+            MutationFilter::builder().kind(local).build(),
+        )
+        .await
+        .unwrap();
+
+    // Gone for good
+    let njobs = client
+        .current_info()
+        .await
+        .unwrap()
+        .data
+        .queues
+        .get(local)
+        .map(|v| *v)
+        .unwrap_or_default();
+    assert_eq!(njobs, 0);
 }
 
 #[tokio::test(flavor = "multi_thread")]
@@ -987,7 +1057,7 @@ async fn mutation_requeue_specific_jobs_only() {
     // prepare a worker that will fail the job unconditionally
     let mut worker = Worker::builder::<io::Error>()
         .register_fn(local, move |_job| async move {
-            panic!("Force fail this job");
+            panic!("force fail this job");
         })
         .connect()
         .await