|
17 | 17 |
|
18 | 18 | import static com..benmanes.caffeine.cache.LocalCacheSubject.mapLocal;
|
19 | 19 | import static com..benmanes.caffeine.cache.testing.CacheSubject.assertThat;
|
| 20 | +import static com..benmanes.caffeine.testing.Awaits.await; |
20 | 21 | import static com..benmanes.caffeine.testing.MapSubject.assertThat;
|
21 | 22 | import static com.google.common.truth.Truth.assertAbout;
|
22 | 23 | import static com.google.common.truth.Truth.assertThat;
|
| 24 | +import static java.lang.Thread.State.BLOCKED; |
| 25 | +import static java.lang.Thread.State.WAITING; |
23 | 26 | import static org.junit.jupiter.api.Assertions.assertThrows;
|
24 | 27 |
|
25 | 28 | import java.lang.ref.WeakReference;
|
26 | 29 | import java.util.Arrays;
|
27 | 30 | import java.util.Collections;
|
| 31 | +import java.util.EnumSet; |
28 | 32 | import java.util.Set;
|
| 33 | +import java.util.concurrent.atomic.AtomicBoolean; |
| 34 | +import java.util.concurrent.atomic.AtomicReference; |
29 | 35 |
|
30 | 36 | import org.testng.Assert;
|
31 | 37 | import org.testng.annotations.DataProvider;
|
32 | 38 | import org.testng.annotations.Test;
|
33 | 39 |
|
34 | 40 | import com..benmanes.caffeine.cache.References.WeakKeyEqualsReference;
|
| 41 | +import com..benmanes.caffeine.testing.ConcurrentTestHarness; |
35 | 42 | import com..benmanes.caffeine.testing.Int;
|
36 | 43 | import com.google.common.collect.testing.SetTestSuiteBuilder;
|
37 | 44 | import com.google.common.collect.testing.TestStringSetGenerator;
|
@@ -138,6 +145,36 @@ public void intern_weak_cleanup() {
|
138 | 145 | assertThat(interner.cache.drainStatus).isEqualTo(BoundedLocalCache.IDLE);
|
139 | 146 | }
|
140 | 147 |
|
| 148 | +@Test |
| 149 | +public void intern_weak_retry() { |
| 150 | +var canonical = new Int(1); |
| 151 | +var other = new Int(1); |
| 152 | + |
| 153 | +var done = new AtomicBoolean(); |
| 154 | +var started = new AtomicBoolean(); |
| 155 | +var writer = new AtomicReference<Thread>(); |
| 156 | +var interner = (WeakInterner<Int>) Interner.<Int>newWeakInterner(); |
| 157 | + |
| 158 | +var result = interner.cache.compute(canonical, (k, v) -> { |
| 159 | +ConcurrentTestHarness.execute(() -> { |
| 160 | +writer.set(Thread.currentThread()); |
| 161 | +started.set(true); |
| 162 | +var value = interner.intern(other); |
| 163 | +assertThat(value).isSameInstanceAs(canonical); |
| 164 | +done.set(true); |
| 165 | +}); |
| 166 | +await().untilTrue(started); |
| 167 | +var threadState = EnumSet.of(BLOCKED, WAITING); |
| 168 | +await().until(() -> { |
| 169 | +var thread = writer.get(); |
| 170 | +return (thread != null) && threadState.contains(thread.getState()); |
| 171 | +}); |
| 172 | +return true; |
| 173 | +}); |
| 174 | +assertThat(result).isTrue(); |
| 175 | +assertThat(interner.intern(canonical)).isSameInstanceAs(canonical); |
| 176 | +} |
| 177 | + |
141 | 178 | @Test
|
142 | 179 | public void nullPointerExceptions() {
|
143 | 180 | new NullPointerTester().testAllPublicStaticMethods(Interner.class);
|
|
0 commit comments