Add test case to ShouldBlockRequestTest

Author: bitterpanda63

agent_api/src/test/java/ShouldBlockRequestTest.java

@@ -5,6 +5,7 @@
 import dev.aikido.agent_api.context.ContextObject;
 import dev.aikido.agent_api.context.User;
 import dev.aikido.agent_api.storage.ServiceConfigStore;
+import dev.aikido.agent_api.storage.statistics.StatisticsStore;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
@@ -38,6 +39,7 @@ public SampleContextObject() {
     public static void clean() {
         Context.set(null);
         ServiceConfigStore.updateFromAPIResponse(emptyAPIResponse);
+        StatisticsStore.clear();
     };
 
     @AfterEach
@@ -133,7 +135,7 @@ public void testEndpointsExistButNoMatch() throws SQLException {
 
     @Test
     public void testEndpointsExistWithMatch() throws SQLException {
-        Context.set(null);
+        Context.set(new SampleContextObject());
         setEmptyConfigWithEndpointList(List.of(
                 new Endpoint("GET", "/api/*", 1, 1000, Collections.emptyList(), false, false, false)
         ));
@@ -142,14 +144,23 @@ public void testEndpointsExistWithMatch() throws SQLException {
         var res1 = ShouldBlockRequest.shouldBlockRequest();
         assertFalse(res1.block());
 
-        Context.set(null);
         setEmptyConfigWithEndpointList(List.of(
                 new Endpoint("GET", "/api/*", 1, 1000, Collections.emptyList(), false, false, true)
         ));
 
         // Test with match & rate-limiting enabled :
         var res2 = ShouldBlockRequest.shouldBlockRequest();
         assertFalse(res2.block());
+        assertEquals(0, StatisticsStore.getStatsRecord().requests().rateLimited());
+
+        var res3 = ShouldBlockRequest.shouldBlockRequest();
+        var res4 = ShouldBlockRequest.shouldBlockRequest();
+        assertTrue(res3.block());
+        assertTrue(res4.block());
+        assertEquals("ip", res3.data().trigger());
+        assertEquals("192.168.1.1", res3.data().ip());
+        assertEquals("ratelimited", res3.data().type());
+        assertEquals(2, StatisticsStore.getStatsRecord().requests().rateLimited());
     }
 
     @Test