Add LockAssert.TestHelper.makeAllAssertsPass

lukas-krecan-lt · lukas-krecan-lt · commit 02a66fb9c971 · 2020-04-04T08:42:01.000+02:00
diff --git a/shedlock-core/src/main/java/net/javacrumbs/shedlock/core/LockAssert.java b/shedlock-core/src/main/java/net/javacrumbs/shedlock/core/LockAssert.java
@@ -47,4 +47,22 @@ public static void assertLocked() {
             throw new IllegalStateException("The task is not locked.");
         }
     }
+
+    static class TestHelper {
+        /**
+         * If pass is set to true, all LockAssert.assertLocked calls in current thread will pass.
+         * To be used in unit tests only
+         *
+         * <code>
+         * LockAssert.TestHelper.makeAllAssertsPass(true)
+         * </code>
+         */
+        public static void makeAllAssertsPass(boolean pass) {
+            if (pass) {
+                LockAssert.startLock("net.javacrumbs.shedlock.core.test-lock");
+            } else {
+                LockAssert.endLock();
+            }
+        }
+    }
 }
diff --git a/shedlock-core/src/test/java/net/javacrumbs/shedlock/core/LockAssertTest.java b/shedlock-core/src/test/java/net/javacrumbs/shedlock/core/LockAssertTest.java
@@ -2,7 +2,6 @@
 
 import org.junit.jupiter.api.Test;
 
-import java.time.Instant;
 import java.util.Optional;
 
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
@@ -29,4 +28,12 @@ void assertLockedShouldNotFailIfLockHeld() {
         );
     }
 
+    @Test
+    void assertShouldNotFailIfConfiguredForTests() {
+        LockAssert.TestHelper.makeAllAssertsPass(true);
+        LockAssert.assertLocked();
+
+        LockAssert.TestHelper.makeAllAssertsPass(false);
+        assertThatThrownBy(LockAssert::assertLocked).isInstanceOf(IllegalStateException.class);
+    }
 }
