lssa/lee/state_machine/src/program_deployment_transaction/message.rs

use borsh::{BorshDeserialize, BorshSerialize};

#[derive(Clone, PartialEq, Eq, BorshSerialize, BorshDeserialize)]
pub struct Message {
    pub(crate) bytecode: Vec<u8>,
}

impl std::fmt::Debug for Message {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("Message")
            .field("bytecode", &format_args!("<{} bytes>", self.bytecode.len()))
            .finish()
    }
}

impl Message {
    #[must_use]
    pub const fn new(bytecode: Vec<u8>) -> Self {
        Self { bytecode }
    }

    #[must_use]
    pub fn into_bytecode(self) -> Vec<u8> {
        self.bytecode
    }
}

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

    #[test]
    fn bytecode_roundtrip() {
        // `Message::new(b).into_bytecode()` must return exactly `b`. Catches
        // mutations of `into_bytecode` returning `vec![]`, `vec![0]`, or `vec![1]`.
        let bytecode = vec![0x7F_u8, 0x45, 0x4C, 0x46]; // ELF magic
        assert_eq!(Message::new(bytecode.clone()).into_bytecode(), bytecode);
        assert!(Message::new(vec![]).into_bytecode().is_empty());
        assert_eq!(Message::new(vec![0xAB]).into_bytecode(), vec![0xAB_u8]);
    }
}