rxprog 1.0.2

use super::command_impl_prelude::*;

/// Select a device
#[derive(Debug)]
pub struct DeviceSelection {
    /// The 4 character device code of the device to select
    pub device_code: String,
}

impl TransmitCommandData for DeviceSelection {
    fn command_data(&self) -> CommandData {
        assert_eq!(self.device_code.len(), 4);

        CommandData {
            opcode: 0x10,
            has_size_field: true,
            payload: self.device_code.bytes().collect(),
        }
    }
}

impl Receive for DeviceSelection {
    type Response = ();

    fn rx<T: io::Read>(&self, p: &mut T) -> Result<Self::Response> {
        let mut reader = ResponseReader::<_, SimpleResponse, WithError>::new(
            p,
            ResponseFirstByte::Byte(0x06),
            ErrorFirstByte(0x90),
        );

        reader
            .read_response()?
            .map(|_| ())
            .map_err(|error_code| match error_code {
                0x11 => CommandError::Checksum.into(),
                0x21 => CommandError::DeviceCode.into(),
                _ => panic!("Unknown error code"),
            })
    }
}

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

    #[test]
    fn test_tx() -> Result<()> {
        let cmd = DeviceSelection {
            device_code: "DEV1".to_string(),
        };
        let command_bytes = [0x10, 0x04, 0x44, 0x45, 0x56, 0x31, 0xDC];
        let mut p = mock_io::Builder::new().write(&command_bytes).build();

        cmd.tx(&mut p)?;

        assert!(is_script_complete(&mut p));

        Ok(())
    }

    #[test]
    fn test_rx_success() {
        let cmd = DeviceSelection {
            device_code: "DEV1".to_string(),
        };
        let response_bytes = [0x06];
        let mut p = mock_io::Builder::new().read(&response_bytes).build();

        let response = cmd.rx(&mut p);

        assert_eq!(response, Ok(()));
        assert!(is_script_complete(&mut p));
    }

    #[test]
    fn test_rx_fail() {
        let cmd = DeviceSelection {
            device_code: "DEV1".to_string(),
        };
        let response_bytes = [0x90, 0x21];
        let mut p = mock_io::Builder::new().read(&response_bytes).build();

        let response = cmd.rx(&mut p);

        assert_eq!(response, Err(CommandError::DeviceCode.into()));
        assert!(is_script_complete(&mut p));
    }
}