Fuzz test added

lukas-krecan · lukas-krecan · commit 8617b1f7df5d · 2016-12-18T10:30:02.000+01:00
diff --git a/shedlock-provider-mongo/pom.xml b/shedlock-provider-mongo/pom.xml
@@ -31,6 +31,13 @@
             <version>2.0.10</version>
             <scope>test</scope>
         </dependency>
+
+        <dependency>
+            <groupId>de.flapdoodle.embed</groupId>
+            <artifactId>de.flapdoodle.embed.mongo</artifactId>
+            <version>1.50.5</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
 </project>
diff --git a/shedlock-provider-mongo/src/test/java/net/javacrumbs/shedlock/provider/mongo/FuzzTest.java b/shedlock-provider-mongo/src/test/java/net/javacrumbs/shedlock/provider/mongo/FuzzTest.java
@@ -0,0 +1,109 @@
+/**
+ * Copyright 2009-2016 the original author or authors.
+ * <p>
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package net.javacrumbs.shedlock.provider.mongo;
+
+import de.flapdoodle.embed.mongo.tests.MongodForTestsFactory;
+import net.javacrumbs.shedlock.core.LockConfiguration;
+import net.javacrumbs.shedlock.core.LockProvider;
+import net.javacrumbs.shedlock.core.SimpleLock;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.net.UnknownHostException;
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
+import java.util.List;
+import java.util.Optional;
+import java.util.concurrent.Callable;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.Future;
+
+import static java.util.stream.Collectors.toList;
+import static java.util.stream.IntStream.range;
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class FuzzTest {
+    private static MongodForTestsFactory mongoFactory;
+
+
+    private static final int THREADS = 4;
+    private static final int ITERATIONS = 100;
+
+    private LockProvider lockProvider;
+
+    private int counter;
+    private final LockConfiguration config = new LockConfiguration("lock", Instant.now().plus(5, ChronoUnit.MINUTES));
+
+    @Before
+    public void createLockProvider() throws UnknownHostException {
+        lockProvider = new MongoLockProvider(mongoFactory.newMongo(), "test", "test");
+    }
+
+
+    @Test
+    public void fuzzTest() throws InterruptedException, ExecutionException {
+        ExecutorService executor = Executors.newFixedThreadPool(THREADS);
+
+        List<Callable<Void>> tasks = range(0, THREADS).mapToObj(i -> (Callable<Void>) this::task).collect(toList());
+        waitForIt(executor.invokeAll(tasks));
+
+        assertThat(counter).isEqualTo(THREADS * ITERATIONS);
+    }
+
+    private void waitForIt(List<Future<Void>> futures) throws InterruptedException, ExecutionException {
+        for (Future<Void> f : futures) {
+            f.get();
+        }
+    }
+
+    private Void task() {
+        for (int i = 0; i < ITERATIONS; ) {
+            Optional<SimpleLock> lock = lockProvider.lock(config);
+            if (lock.isPresent()) {
+                int n = counter;
+                sleep();
+                counter = n + 1;
+                i++;
+                lock.get().unlock();
+            }
+        }
+        return null;
+    }
+
+    @BeforeClass
+    public static void startMongo() throws IOException {
+        mongoFactory = new MongodForTestsFactory();
+    }
+
+    @AfterClass
+    public static void stopMongo() throws IOException {
+        mongoFactory.shutdown();
+    }
+
+    private void sleep() {
+        try {
+            Thread.sleep(1);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+}
