Skip to content

Commit 1ef34f9

Browse files
[ISSUE #443✅]Add test case for BloomFilterData💚 (#444)
1 parent ccbcfcf commit 1ef34f9

File tree

1 file changed

+51
-0
lines changed

1 file changed

+51
-0
lines changed

rocketmq-filter/src/utils/bloom_filter_data.rs

+51
Original file line numberDiff line numberDiff line change
@@ -42,3 +42,54 @@ impl BloomFilterData {
4242
self.bit_num
4343
}
4444
}
45+
46+
#[cfg(test)]
47+
mod tests {
48+
use super::*;
49+
50+
#[test]
51+
fn new_creates_bloom_filter_data_with_given_values() {
52+
let bit_pos = vec![1, 2, 3];
53+
let bit_num = 10;
54+
let bloom_filter_data = BloomFilterData::new(bit_pos.clone(), bit_num);
55+
56+
assert_eq!(bloom_filter_data.bit_pos(), &bit_pos);
57+
assert_eq!(bloom_filter_data.bit_num(), bit_num);
58+
}
59+
60+
#[test]
61+
fn set_bit_pos_updates_bit_pos() {
62+
let mut bloom_filter_data = BloomFilterData::new(vec![1, 2, 3], 10);
63+
let new_bit_pos = vec![4, 5, 6];
64+
65+
bloom_filter_data.set_bit_pos(new_bit_pos.clone());
66+
67+
assert_eq!(bloom_filter_data.bit_pos(), &new_bit_pos);
68+
}
69+
70+
#[test]
71+
fn set_bit_num_updates_bit_num() {
72+
let mut bloom_filter_data = BloomFilterData::new(vec![1, 2, 3], 10);
73+
let new_bit_num = 20;
74+
75+
bloom_filter_data.set_bit_num(new_bit_num);
76+
77+
assert_eq!(bloom_filter_data.bit_num(), new_bit_num);
78+
}
79+
80+
#[test]
81+
fn bit_pos_returns_bit_pos() {
82+
let bit_pos = vec![1, 2, 3];
83+
let bloom_filter_data = BloomFilterData::new(bit_pos.clone(), 10);
84+
85+
assert_eq!(bloom_filter_data.bit_pos(), &bit_pos);
86+
}
87+
88+
#[test]
89+
fn bit_num_returns_bit_num() {
90+
let bit_num = 10;
91+
let bloom_filter_data = BloomFilterData::new(vec![1, 2, 3], bit_num);
92+
93+
assert_eq!(bloom_filter_data.bit_num(), bit_num);
94+
}
95+
}

0 commit comments

Comments
 (0)