PrivateMaxEntriesMapStressTest.java

package com.fasterxml.jackson.databind.util.internal;

import java.time.Duration;
import java.time.temporal.ChronoUnit;
import java.util.Map;
import java.util.Random;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.assertEquals;

public class PrivateMaxEntriesMapStressTest {

    //increase these to increase the stress
    private static int iterations = 100000;
    private static int threads = 20;
    private static int waitSeconds = 60;

    @Test
    public void testManyEntries() throws Exception {
        final int maxEntries = 30;
        final int maxKey = 100;
        final Random rnd = new Random();
        final PrivateMaxEntriesMap<Integer, UUID> clhm =
                new PrivateMaxEntriesMap.Builder<Integer, UUID>().maximumCapacity(maxEntries).build();
        final Map<Integer, UUID> map = new ConcurrentHashMap<>();
        final ExecutorService executor = Executors.newFixedThreadPool(threads);
        try {
            for (int i = 0; i < maxKey; i++) {
                final Integer key = Integer.valueOf(i);
                executor.submit(() -> {
                    UUID uuid = UUID.randomUUID();
                    synchronized (key) {
                        clhm.put(key, uuid);
                        map.put(key, uuid);
                    }
                });
            }
            for (int i = 0; i < iterations; i++) {
                executor.submit(() -> {
                    Integer key = Integer.valueOf(rnd.nextInt(maxKey));
                    UUID uuid = UUID.randomUUID();
                    synchronized (key) {
                        clhm.put(key, uuid);
                        map.put(key, uuid);
                    }
                });
            }
        } finally {
            executor.shutdown();
        }
        executor.awaitTermination(waitSeconds, TimeUnit.SECONDS);

        final long endTime = System.nanoTime() + Duration.of(waitSeconds, ChronoUnit.SECONDS).toNanos();
        boolean assertsFailing = true;
        while(assertsFailing) {
            clhm.drainBuffers();
            try {
                assertEquals(clhm.size(), maxEntries);
                for (int i = 0; i < maxKey; i++) {
                    UUID uuid = clhm.get(i);
                    if (uuid != null) {
                        assertEquals(map.get(i), clhm.get(i));
                    }
                }
                assertsFailing = false;
            } catch (Throwable t) {
                if (System.nanoTime() > endTime) {
                    throw t;
                }
                Thread.sleep(100);
            }
        }
    }
}