Skip to content

Commit 4362d45

Browse files
authored
[ISSUE #397]✅Add test case for MessageSysFlag💡 (#398)
1 parent c748ed5 commit 4362d45

File tree

1 file changed

+55
-0
lines changed

1 file changed

+55
-0
lines changed

rocketmq-common/src/common/sys_flag/message_sys_flag.rs

+55
Original file line numberDiff line numberDiff line change
@@ -41,3 +41,58 @@ impl MessageSysFlag {
4141
(flag & expected_flag) != 0
4242
}
4343
}
44+
45+
#[cfg(test)]
46+
mod tests {
47+
use super::*;
48+
49+
#[test]
50+
fn get_transaction_value_returns_correct_value() {
51+
let flag = MessageSysFlag::TRANSACTION_COMMIT_TYPE;
52+
assert_eq!(
53+
MessageSysFlag::get_transaction_value(flag),
54+
MessageSysFlag::TRANSACTION_COMMIT_TYPE
55+
);
56+
}
57+
58+
#[test]
59+
fn reset_transaction_value_resets_correctly() {
60+
let flag = MessageSysFlag::TRANSACTION_COMMIT_TYPE;
61+
let new_flag = MessageSysFlag::reset_transaction_value(
62+
flag,
63+
MessageSysFlag::TRANSACTION_ROLLBACK_TYPE,
64+
);
65+
assert_eq!(new_flag, MessageSysFlag::TRANSACTION_ROLLBACK_TYPE);
66+
}
67+
68+
#[test]
69+
fn clear_compressed_flag_clears_correctly() {
70+
let flag = MessageSysFlag::COMPRESSED_FLAG;
71+
let new_flag = MessageSysFlag::clear_compressed_flag(flag);
72+
assert_eq!(new_flag, 0);
73+
}
74+
75+
#[test]
76+
fn get_compression_type_returns_correct_type() {
77+
let flag = MessageSysFlag::COMPRESSION_LZ4_TYPE;
78+
assert_eq!(
79+
MessageSysFlag::get_compression_type(flag),
80+
CompressionType::LZ4
81+
);
82+
}
83+
84+
#[test]
85+
fn check_returns_true_when_flag_is_set() {
86+
let flag = MessageSysFlag::COMPRESSED_FLAG;
87+
assert!(MessageSysFlag::check(flag, MessageSysFlag::COMPRESSED_FLAG));
88+
}
89+
90+
#[test]
91+
fn check_returns_false_when_flag_is_not_set() {
92+
let flag = 0;
93+
assert!(!MessageSysFlag::check(
94+
flag,
95+
MessageSysFlag::COMPRESSED_FLAG
96+
));
97+
}
98+
}

0 commit comments

Comments
 (0)