MiniDay/Velocity
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Add ratelimit tests

Browse Source
This commit is contained in:
MatrixTunnel
2018-08-10 18:57:10 -07:00
parent d752edc92d
commit 6e4f90dfae
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
proxy/src/test/java/com/velocitypowered/proxy/util/RatelimiterTest.java
View File

@@ -12,7 +12,14 @@ import static org.junit.jupiter.api.Assertions.*;
class RatelimiterTest { class RatelimiterTest {
@Test @Test
void attempt() { void attemptZero() {
Ratelimiter noRatelimiter = new Ratelimiter(0);
assertTrue(noRatelimiter.attempt(InetAddress.getLoopbackAddress()));
assertTrue(noRatelimiter.attempt(InetAddress.getLoopbackAddress()));
}
@Test
void attemptOne() {
long base = System.nanoTime(); long base = System.nanoTime();
AtomicLong extra = new AtomicLong(); AtomicLong extra = new AtomicLong();
Ticker testTicker = new Ticker() { Ticker testTicker = new Ticker() {
@@ -27,4 +34,5 @@ class RatelimiterTest {
extra.addAndGet(TimeUnit.SECONDS.toNanos(2)); extra.addAndGet(TimeUnit.SECONDS.toNanos(2));
assertTrue(ratelimiter.attempt(InetAddress.getLoopbackAddress())); assertTrue(ratelimiter.attempt(InetAddress.getLoopbackAddress()));
} }
} }