import java.util.concurrent.TimeUnit;
import java.util.stream.IntStream;

import org.infinispan.Cache;
import org.infinispan.configuration.cache.Configuration;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.manager.DefaultCacheManager;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.persistence.sifs.configuration.SoftIndexFileStoreConfigurationBuilder;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.Timeout;

public class SifsLargeNodeTest {

    private EmbeddedCacheManager cacheManager;

    @Rule
    public Timeout timeout = new Timeout(15, TimeUnit.SECONDS);

    private static Configuration createConfig(String location, int maxNodeSizeInBytes) {
        return new ConfigurationBuilder()
                .persistence().addStore(SoftIndexFileStoreConfigurationBuilder.class)
                .dataLocation(location + "/data").indexLocation(location + "/index")
                .maxNodeSize(maxNodeSizeInBytes)
                .preload(false).purgeOnStartup(false)
                .build();
    }

    @Before
    public void setUp() {
        cacheManager = new DefaultCacheManager();
    }

    @After
    public void tearDown() {
        cacheManager.stop();
    }

    @Test
    public void testLoad() {
        int maxNodeSizeInBytes = 32780; // Note: Anything up to 32777 works fine
        String cacheName = "sifslargenodetestcache";
        Configuration cfg = createConfig(cacheName, maxNodeSizeInBytes);

        int numberOfRuns = 10;
        int valuesToPutPerRun = 10000;
        for (int run = 0; run < numberOfRuns; run++) {
            System.out.println("RUN " + run);
            cacheManager.defineConfiguration(cacheName, cfg);
            Cache<Integer, String> testCacheBeforeReduction = cacheManager.getCache(cacheName);
            IntStream.range(run * valuesToPutPerRun, (run + 1) * valuesToPutPerRun)
                    .forEach(i -> testCacheBeforeReduction.put(i, Integer.toString(i)));
            cacheManager.stop();
            cacheManager = new DefaultCacheManager();
        }
    }
}