@@ -41,3 +41,58 @@ impl MessageSysFlag {
41
41
( flag & expected_flag) != 0
42
42
}
43
43
}
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