1 files changed, 30 insertions, 0 deletions
diff --git a/vespajlib/src/test/java/com/yahoo/concurrent/maintenance/MaintainerTest.java b/vespajlib/src/test/java/com/yahoo/concurrent/maintenance/MaintainerTest.java
index 820d1fc3d1d..2bfaad894a5 100644
--- a/vespajlib/src/test/java/com/yahoo/concurrent/maintenance/MaintainerTest.java
+++ b/vespajlib/src/test/java/com/yahoo/concurrent/maintenance/MaintainerTest.java
@@ -6,6 +6,7 @@ import org.junit.Test;
 import java.time.Duration;
 import java.time.Instant;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicLong;
 
 import static org.junit.Assert.assertEquals;
 
@@ -36,4 +37,33 @@ public class MaintainerTest {
         assertEquals(300, Maintainer.staggeredDelay(interval, now, "cfg0", cluster).toMillis());
     }
 
+    @Test
+    public void success_metric() {
+        AtomicLong consecutiveFailures = new AtomicLong();
+        JobMetrics jobMetrics = new JobMetrics((job, count) -> consecutiveFailures.set(count));
+        TestMaintainer maintainer = new TestMaintainer(jobMetrics);
+
+        // Maintainer fails twice in a row
+        maintainer.successOnNextRun(false).run();
+        assertEquals(1, consecutiveFailures.get());
+        maintainer.successOnNextRun(false).run();
+        assertEquals(2, consecutiveFailures.get());
+
+        // Maintainer runs successfully
+        maintainer.successOnNextRun(true).run();
+        assertEquals(0, consecutiveFailures.get());
+
+        // Maintainer runs successfully again
+        maintainer.run();
+        assertEquals(0, consecutiveFailures.get());
+
+        // Maintainer throws
+        maintainer.throwOnNextRun(true).run();
+        assertEquals(1, consecutiveFailures.get());
+
+        // Maintainer recovers
+        maintainer.throwOnNextRun(false).run();
+        assertEquals(0, consecutiveFailures.get());
+    }
+
 }