LoginSystem/src/test/java/fr/xephi/authme/util/expiring/TimedCounterTest.java

package fr.xephi.authme.util.expiring;

import org.junit.Test;

import java.util.concurrent.TimeUnit;

import static org.hamcrest.Matchers.equalTo;
import static org.junit.Assert.assertThat;

/**
 * Test for {@link TimedCounter}.
 */
public class TimedCounterTest {

    @Test
    public void shouldReturnZeroForAnyKey() {
        // given
        TimedCounter<Double> counter = new TimedCounter<>(1, TimeUnit.DAYS);

        // when / then
        assertThat(counter.get(2.0), equalTo(0));
        assertThat(counter.get(-3.14159), equalTo(0));
    }

    @Test
    public void shouldIncrementCount() {
        // given
        TimedCounter<String> counter = new TimedCounter<>(10, TimeUnit.MINUTES);
        counter.put("moto", 12);

        // when
        counter.increment("hello");
        counter.increment("moto");

        // then
        assertThat(counter.get("hello"), equalTo(1));
        assertThat(counter.get("moto"), equalTo(13));
    }

    @Test
    public void shouldSumUpEntries() {
        // given
        TimedCounter<String> counter = new TimedCounter<>(90, TimeUnit.SECONDS);
        counter.entries.put("expired", new ExpiringMap.ExpiringEntry<>(800, 0));
        counter.entries.put("expired2", new ExpiringMap.ExpiringEntry<>(24, System.currentTimeMillis() - 100));
        counter.put("other", 10);
        counter.put("Another", 4);

        // when
        int totals = counter.total();

        // then
        assertThat(totals, equalTo(14));
    }
}