From d1adfd46d2f3acf7a41d1718adbd5513084cd5dd Mon Sep 17 00:00:00 2001
From: Paul Campbell <pcampbell@kemitix.net>
Date: Mon, 1 Oct 2018 21:59:52 +0100
Subject: [PATCH] [lazy] add test to verify re-entrant behaviour of value()

---
 .../kemitix/mon/lazy/LazySupplierTest.java    | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/src/test/java/net/kemitix/mon/lazy/LazySupplierTest.java b/src/test/java/net/kemitix/mon/lazy/LazySupplierTest.java
index e2ccf5c..78d7def 100644
--- a/src/test/java/net/kemitix/mon/lazy/LazySupplierTest.java
+++ b/src/test/java/net/kemitix/mon/lazy/LazySupplierTest.java
@@ -1,15 +1,21 @@
 package net.kemitix.mon.lazy;
 
 import org.assertj.core.api.WithAssertions;
+import org.junit.Rule;
 import org.junit.Test;
+import org.junit.rules.Timeout;
 
 import java.util.UUID;
+import java.util.concurrent.*;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Supplier;
 
 public class LazySupplierTest implements WithAssertions {
 
+    @Rule
+    public final Timeout timeout = Timeout.seconds(1);
+
     @Test
     public void whenCreateLazyThenSupplierIsNotCalled() {
         //given
@@ -138,4 +144,33 @@ public class LazySupplierTest implements WithAssertions {
         assertThat(value).isEqualTo(uuid.toString());
     }
 
+    @Test
+    public void whenLazyValueCalledOnTwoThreadsThenSupplierIsOnlyCalledOnce() throws ExecutionException, InterruptedException {
+        //given
+        final AtomicInteger supplierCalledCounter = new AtomicInteger(0);
+        final CountDownLatch latch = new CountDownLatch(1);
+        final UUID uuid = UUID.randomUUID();
+        final Supplier<UUID> supplier = () -> {
+            supplierCalledCounter.incrementAndGet();
+            for (int i = 0; i < 10000; i++) {
+                // hum
+            };
+            return uuid;
+        };
+        final Lazy<UUID> lazy = Lazy.of(supplier);
+        final Callable<UUID> callable = () -> {
+            latch.await();
+            return lazy.value();
+        };
+        final ExecutorService executorService = Executors.newFixedThreadPool(2);
+        //when
+        final Future<UUID> call1 = executorService.submit(callable);
+        final Future<UUID> call2 = executorService.submit(callable);
+        latch.countDown();
+        //then
+        assertThat(call1.get()).isEqualTo(uuid);
+        assertThat(call2.get()).isEqualTo(uuid);
+        assertThat(supplierCalledCounter).hasValue(1);
+    }
+
 }
\ No newline at end of file