TimedHashSetTest.java

package com.jcwhatever.nucleus.collections.timed;

import static org.junit.Assert.assertEquals;

import com.jcwhatever.v1_8_R3.BukkitTester;
import com.jcwhatever.v1_8_R3.MockPlugin;
import com.jcwhatever.nucleus.NucleusTest;
import com.jcwhatever.nucleus.collections.java.SetRunnable;
import com.jcwhatever.nucleus.utils.TimeScale;

import org.junit.Test;

public class TimedHashSetTest {

    @Test
    public void testSetInterface() {

        NucleusTest.init();

        MockPlugin plugin = new MockPlugin("dummy");
        plugin.onEnable();

        TimedHashSet<String> set = new TimedHashSet<String>(plugin);

        SetRunnable<String> setTest = new SetRunnable<String>(set, "a", "b", "c");

        setTest.run();
    }


    /**
     * Make sure element is removed.
     * Tests the scheduled janitor task.
     */
    @Test
    public void testElementLifespan() throws Exception {

        NucleusTest.init();

        MockPlugin plugin = new MockPlugin("dummy");
        plugin.onEnable();

        TimedHashSet<String> set = new TimedHashSet<String>(plugin);

        set.add("a", 1000, TimeScale.MILLISECONDS);

        long expires = System.currentTimeMillis() + 1000;

        while (System.currentTimeMillis() < expires + 100) {

            assertEquals(set.contains("a"), System.currentTimeMillis() < expires);

            Thread.sleep(5);
        }
    }

    /**
     * Make sure element is removed without a direct check.
     */
    @Test
    public void testElementLifespan1() throws Exception {

        NucleusTest.init();

        MockPlugin plugin = new MockPlugin("dummy");
        plugin.onEnable();

        TimedHashSet<String> set = new TimedHashSet<String>(plugin);

        set.add("a", 20, TimeScale.TICKS);

        BukkitTester.pause(30);

        assertEquals(0, set.size());
    }


}