rocketmq-model 1.0.0

// Copyright 2023 The RocketMQ Rust Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

const FLAG_UNIT: u32 = 0x1 << 0;
const FLAG_UNIT_SUB: u32 = 0x1 << 1;

pub fn build_sys_flag(unit: bool, has_unit_sub: bool) -> u32 {
    let mut sys_flag = 0;
    if unit {
        sys_flag |= FLAG_UNIT;
    }
    if has_unit_sub {
        sys_flag |= FLAG_UNIT_SUB;
    }
    sys_flag
}

pub fn set_unit_flag(sys_flag: u32) -> u32 {
    sys_flag | FLAG_UNIT
}

pub fn clear_unit_flag(sys_flag: u32) -> u32 {
    sys_flag & !FLAG_UNIT
}

pub fn has_unit_flag(sys_flag: u32) -> bool {
    (sys_flag & FLAG_UNIT) == FLAG_UNIT
}

pub fn set_unit_sub_flag(sys_flag: u32) -> u32 {
    sys_flag | FLAG_UNIT_SUB
}

pub fn clear_unit_sub_flag(sys_flag: u32) -> u32 {
    sys_flag & !FLAG_UNIT_SUB
}

pub fn has_unit_sub_flag(sys_flag: u32) -> bool {
    (sys_flag & FLAG_UNIT_SUB) == FLAG_UNIT_SUB
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn build_sys_flag_sets_correct_flags() {
        assert_eq!(build_sys_flag(true, false), FLAG_UNIT);
        assert_eq!(build_sys_flag(false, true), FLAG_UNIT_SUB);
        assert_eq!(build_sys_flag(true, true), FLAG_UNIT | FLAG_UNIT_SUB);
    }

    #[test]
    fn set_unit_flag_sets_correct_flag() {
        let sys_flag = 0;
        assert_eq!(set_unit_flag(sys_flag), FLAG_UNIT);
    }

    #[test]
    fn clear_unit_flag_clears_correct_flag() {
        let sys_flag = FLAG_UNIT;
        assert_eq!(clear_unit_flag(sys_flag), 0);
    }

    #[test]
    fn has_unit_flag_returns_correct_value() {
        assert!(has_unit_flag(FLAG_UNIT));
        assert!(!has_unit_flag(FLAG_UNIT_SUB));
    }

    #[test]
    fn set_unit_sub_flag_sets_correct_flag() {
        let sys_flag = 0;
        assert_eq!(set_unit_sub_flag(sys_flag), FLAG_UNIT_SUB);
    }

    #[test]
    fn clear_unit_sub_flag_clears_correct_flag() {
        let sys_flag = FLAG_UNIT_SUB;
        assert_eq!(clear_unit_sub_flag(sys_flag), 0);
    }

    #[test]
    fn has_unit_sub_flag_returns_correct_value() {
        assert!(has_unit_sub_flag(FLAG_UNIT_SUB));
        assert!(!has_unit_sub_flag(FLAG_UNIT));
    }
}