third_party/rust_crates/vendor/podio/tests/io.rs

extern crate podio;

use std::io::{self, Read};
use podio::ReadPodExt;

struct TestReader {
    state: Option<io::Result<usize>>,
}

impl TestReader {
    fn new(first_return: io::Result<usize>) -> TestReader {
        TestReader { state: Some(first_return) }
    }

    fn get(&mut self) -> io::Result<u32> {
        self.read_u32::<podio::LittleEndian>()
    }

    fn test(first_return: io::Result<usize>) -> io::Result<u32> {
        TestReader::new(first_return).get()
    }
}

impl Read for TestReader {
    fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
        match self.state.take().unwrap_or(Ok(1)) {
            Ok(n) => {
                for i in buf.iter_mut().take(n) {
                    *i = 0xA5;
                }
                Ok(n)
            },
            e @ Err(..) => e,
        }
    }
}

#[test]
fn interrupted() {
    // Getting an io::ErrorKind::Interrupted should be retried, and thus succeed
    let first_read = Err(io::Error::new(io::ErrorKind::Interrupted, "Interrupted"));
    assert_eq!(TestReader::test(first_read).unwrap(), 0xA5A5A5A5);
}

#[test]
fn eof() {
    // Getting a Ok(0) implies an unexpected EOF
    let first_read = Ok(0);
    assert!(TestReader::test(first_read).is_err());
}

#[test]
fn err() {
    // Getting any other error is still an error
    let first_read = Err(io::Error::new(io::ErrorKind::Other, "Other"));
    assert!(TestReader::test(first_read).is_err());
}