Statistics class add new test cases for rateLimited count

Author: bitterpanda63

agent_api/src/main/java/dev/aikido/agent_api/storage/statistics/Statistics.java

@@ -150,6 +150,7 @@ public void clear() {
         this.attacksDetected = 0;
         this.attackWavesBlocked = 0;
         this.attackWavesDetected = 0;
+        this.rateLimited = 0;
         this.startedAt = UnixTimeMS.getUnixTimeMS();
         this.operations.clear();
         this.ipAddressMatches.clear();

agent_api/src/test/java/storage/StatisticsTest.java

@@ -34,11 +34,14 @@ public void testClear() {
         stats.incrementAttacksDetected("test2");
         stats.incrementAttacksDetected("test1");
         stats.incrementAttacksDetected("test1");
+        stats.incrementRateLimitStats();
+        stats.incrementRateLimitStats();
         assertEquals(3, stats.getAttacksDetected());
         assertEquals(2, stats.getAttacksBlocked());
         assertEquals(20, stats.getTotalHits());
         assertEquals(2, stats.getOperations().get("test1").getAttacksDetected().get("total"));
         assertEquals(1, stats.getOperations().get("test1").getAttacksDetected().get("blocked"));
+        assertEquals(2, stats.getRateLimitedStats());
 
         assertFalse(stats.getOperations().containsKey("test2"));
         // Reset :
@@ -47,6 +50,7 @@ public void testClear() {
         assertEquals(0, stats.getAttacksBlocked());
         assertEquals(0, stats.getAttacksDetected());
         assertEquals(0, stats.getTotalHits());
+        assertEquals(0, stats.getRateLimitedStats());
 
     }
 
@@ -56,6 +60,7 @@ public void testConstructor() {
         assertEquals(100, stats2.getTotalHits());
         assertEquals(5, stats2.getAttacksDetected());
         assertEquals(1, stats2.getAttacksBlocked());
+        assertEquals(0, stats.getRateLimitedStats());
     }
 
     @Test