From c273ada5026d36b368418065a6e58e2d0b8a1a6f Mon Sep 17 00:00:00 2001
From: Mel Henning <drawoc@darkrefraction.com>
Date: Mon, 30 Dec 2024 18:11:21 -0500
Subject: [PATCH] compiler/rust/bitset: Test next_unset()

Reviewed-by: Mary Guillemard <mary.guillemard@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/32812>
---
 src/compiler/rust/bitset.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/compiler/rust/bitset.rs b/src/compiler/rust/bitset.rs
index a07610c5871..a6955e4d9c5 100644
--- a/src/compiler/rust/bitset.rs
+++ b/src/compiler/rust/bitset.rs
@@ -427,6 +427,23 @@ mod tests {
         assert!(set2.is_empty());
     }
 
+    #[test]
+    fn test_next_unset() {
+        for test_range in
+            &[0..0, 42..1337, 1337..1337, 31..32, 32..33, 63..64, 64..65]
+        {
+            let mut set = BitSet::new();
+            for i in test_range.clone() {
+                set.insert(i);
+            }
+            for extra_bit in [17, 34, 39] {
+                assert!(test_range.end != extra_bit);
+                set.insert(extra_bit);
+            }
+            assert_eq!(set.next_unset(test_range.start), test_range.end);
+        }
+    }
+
     #[test]
     fn test_from_iter() {
         let vec = vec![0, 1, 99];