From 31b059c0d393fe18dbf5dcd96d6481becc377812 Mon Sep 17 00:00:00 2001
From: benedettadavico <benedettadavico@users.noreply.github.com>
Date: Fri, 1 May 2026 08:42:48 +0200
Subject: [PATCH] unit test

---
 .../src/deposits_buffer/refill_task.rs        | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/common/credential-proxy/src/deposits_buffer/refill_task.rs b/common/credential-proxy/src/deposits_buffer/refill_task.rs
index ccb318c575..5534eaab80 100644
--- a/common/credential-proxy/src/deposits_buffer/refill_task.rs
+++ b/common/credential-proxy/src/deposits_buffer/refill_task.rs
@@ -76,3 +76,34 @@ impl RefillTask {
         self.join_handle.lock().expect("mutex got poisoned").take()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn in_progress_resets_after_guard_drop() {
+        let task = RefillTask::default();
+
+        let (guard, completion_guard) = task.try_get_new_task_guard().unwrap();
+        drop(guard);
+        assert!(task.try_get_new_task_guard().is_none());
+
+        drop(completion_guard);
+        assert!(task.try_get_new_task_guard().is_some());
+    }
+
+    #[test]
+    fn in_progress_resets_on_panic() {
+        let task = RefillTask::default();
+
+        let (guard, completion_guard) = task.try_get_new_task_guard().unwrap();
+        drop(guard);
+
+        let _ = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
+            let _g = completion_guard;
+            panic!("simulated refill task panic");
+        }));
+        assert!(task.try_get_new_task_guard().is_some());
+    }
+}