fix BitsArray upper boundary validate bug, UT

apache · Sep 8, 2024 · abd35b5 · abd35b5
1 parent 41c0000
commit abd35b5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/filter/src/test/java/org/apache/rocketmq/filter/BitsArrayTest.java b/filter/src/test/java/org/apache/rocketmq/filter/BitsArrayTest.java
@@ -120,4 +120,18 @@ public void testOr() {
             assertThat(b1.getBit(i)).isTrue();
         }
     }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testBitBoundaryOverflow() {
+        BitsArray bitsArray = BitsArray.create(16);
+        assertThat(bitsArray.bitLength()).isEqualTo(16);
+        bitsArray.getBit(16);
+    }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testByteBoundaryOverflow() {
+        BitsArray bitsArray = BitsArray.create(16);
+        assertThat(bitsArray.byteLength() ).isEqualTo(2);
+        bitsArray.getByte(2);
+    }
 }