rust-gamedig/src/utils.rs

use crate::{
    GDError::{PacketOverflow, PacketUnderflow},
    GDResult,
};

use std::cmp::Ordering;

pub fn error_by_expected_size(expected: usize, size: usize) -> GDResult<()> {
    match size.cmp(&expected) {
        Ordering::Greater => Err(PacketOverflow),
        Ordering::Less => Err(PacketUnderflow),
        Ordering::Equal => Ok(()),
    }
}

pub fn address_and_port_as_string(address: &str, port: u16) -> String { format!("{}:{}", address, port) }

pub fn u8_lower_upper(n: u8) -> (u8, u8) { (n & 15, n >> 4) }

#[cfg(test)]
mod tests {
    #[test]
    fn address_and_port_as_string() {
        assert_eq!(
            super::address_and_port_as_string("192.168.0.1", 27015),
            "192.168.0.1:27015"
        );
    }

    #[test]
    fn u8_lower_upper() {
        assert_eq!(super::u8_lower_upper(171), (11, 10));
    }

    #[test]
    fn error_by_expected_size() {
        assert!(super::error_by_expected_size(69, 69).is_ok());
        assert!(super::error_by_expected_size(69, 68).is_err());
        assert!(super::error_by_expected_size(69, 70).is_err());
    }
}