/******************************************************************************
 * Struct
 ******************************************************************************/

#[derive(Debug, PartialEq)]
pub struct FiveTuple {
    src_ip: String,
    src_port: u16,
    dst_ip: String,
    dst_port: u16,
    protocol: u8,
}

/******************************************************************************
 * API
 ******************************************************************************/

impl FiveTuple {
    pub fn new(
        src_ip: String,
        src_port: u16,
        dst_ip: String,
        dst_port: u16,
        protocol: u8,
    ) -> FiveTuple {
        FiveTuple {
            src_ip,
            src_port,
            dst_ip,
            dst_port,
            protocol,
        }
    }

    pub fn to_string(&self) -> String {
        let mut session_id = String::new();
        session_id.push_str(&self.src_ip.to_string());
        session_id.push_str(":");
        session_id.push_str(&self.src_port.to_string());
        session_id.push_str("-");
        session_id.push_str(&self.dst_ip.to_string());
        session_id.push_str(":");
        session_id.push_str(&self.dst_port.to_string());
        session_id.push_str("-");
        session_id.push_str(&self.protocol.to_string());
        session_id
    }
}

/******************************************************************************
 * TEST
 ******************************************************************************/

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

    #[test]
    fn test_five_tuple() {
        let five_tuple_4 = FiveTuple::new(
            "192.168.0.1".to_string(),
            2345,
            "192.168.0.2".to_string(),
            80,
            1,
        );
        let session_id_4 = five_tuple_4.to_string();
        assert_eq!(session_id_4, "192.168.0.1:2345-192.168.0.2:80-1");

        let five_tuple_6 = FiveTuple::new(
            "11:22:33:44:55:66:77:88".to_string(),
            2345,
            "88:99:aa:bb:cc:dd:ee:ff".to_string(),
            80,
            1,
        );
        let session_id_6 = five_tuple_6.to_string();
        assert_eq!(
            session_id_6,
            "11:22:33:44:55:66:77:88:2345-88:99:aa:bb:cc:dd:ee:ff:80-1"
        );
    }
}