Remove unnecessary use of Cursor in tests.

This commit is contained in:
KAMADA Ken'ichi 2020-12-02 21:48:40 +09:00
parent 386fb2cc8b
commit a406139859
3 changed files with 25 additions and 26 deletions

View File

@ -108,7 +108,6 @@ pub fn is_jpeg(buf: &[u8]) -> bool {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use std::io::Cursor;
use super::*; use super::*;
#[test] #[test]
@ -122,12 +121,12 @@ mod tests {
b"\xff\xd8\xff\xe1\x00\x08\x03\x04", b"\xff\xd8\xff\xe1\x00\x08\x03\x04",
]; ];
for &data in sets { for &data in sets {
assert_err_pat!(get_exif_attr(&mut Cursor::new(data)), assert_err_pat!(get_exif_attr(&mut &data[..]),
Error::InvalidFormat("Broken JPEG file")); Error::InvalidFormat("Broken JPEG file"));
} }
let mut data = b"\xff\xd8\xff\xe1\x00\x08Exif\0\0".to_vec(); let mut data = b"\xff\xd8\xff\xe1\x00\x08Exif\0\0".to_vec();
get_exif_attr(&mut Cursor::new(&data)).unwrap(); get_exif_attr(&mut &data[..]).unwrap();
while let Some(_) = data.pop() { while let Some(_) = data.pop() {
get_exif_attr(&mut &data[..]).unwrap_err(); get_exif_attr(&mut &data[..]).unwrap_err();
} }
@ -136,26 +135,26 @@ mod tests {
#[test] #[test]
fn no_exif() { fn no_exif() {
let data = b"\xff\xd8\xff\xd9"; let data = b"\xff\xd8\xff\xd9";
assert_err_pat!(get_exif_attr(&mut Cursor::new(data)), assert_err_pat!(get_exif_attr(&mut &data[..]),
Error::NotFound(_)); Error::NotFound(_));
} }
#[test] #[test]
fn out_of_sync() { fn out_of_sync() {
let data = b"\xff\xd8\x01\x02\x03\xff\x00\xff\xd9"; let data = b"\xff\xd8\x01\x02\x03\xff\x00\xff\xd9";
assert_err_pat!(get_exif_attr(&mut Cursor::new(data)), assert_err_pat!(get_exif_attr(&mut &data[..]),
Error::NotFound(_)); Error::NotFound(_));
} }
#[test] #[test]
fn empty() { fn empty() {
let data = b"\xff\xd8\xff\xe1\x00\x08Exif\0\0\xff\xd9"; let data = b"\xff\xd8\xff\xe1\x00\x08Exif\0\0\xff\xd9";
assert_ok!(get_exif_attr(&mut Cursor::new(data)), []); assert_ok!(get_exif_attr(&mut &data[..]), []);
} }
#[test] #[test]
fn non_empty() { fn non_empty() {
let data = b"\xff\xd8\xff\xe1\x00\x0aExif\0\0\xbe\xad\xff\xd9"; let data = b"\xff\xd8\xff\xe1\x00\x0aExif\0\0\xbe\xad\xff\xd9";
assert_ok!(get_exif_attr(&mut Cursor::new(data)), [0xbe, 0xad]); assert_ok!(get_exif_attr(&mut &data[..]), [0xbe, 0xad]);
} }
} }

View File

@ -87,7 +87,6 @@ pub fn is_png(buf: &[u8]) -> bool {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use std::io::Cursor;
use super::*; use super::*;
#[test] #[test]
@ -98,12 +97,12 @@ mod tests {
b"\x89PNG\x0d\x0a\x1a", b"\x89PNG\x0d\x0a\x1a",
]; ];
for &data in sets { for &data in sets {
assert_err_pat!(get_exif_attr(&mut Cursor::new(data)), assert_err_pat!(get_exif_attr(&mut &data[..]),
Error::InvalidFormat("Broken PNG file")); Error::InvalidFormat("Broken PNG file"));
} }
let mut data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\x04eXIfExif".to_vec(); let mut data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\x04eXIfExif".to_vec();
get_exif_attr(&mut Cursor::new(&data)).unwrap(); get_exif_attr(&mut &data[..]).unwrap();
while let Some(_) = data.pop() { while let Some(_) = data.pop() {
get_exif_attr(&mut &data[..]).unwrap_err(); get_exif_attr(&mut &data[..]).unwrap_err();
} }
@ -112,19 +111,19 @@ mod tests {
#[test] #[test]
fn no_exif() { fn no_exif() {
let data = b"\x89PNG\x0d\x0a\x1a\x0a"; let data = b"\x89PNG\x0d\x0a\x1a\x0a";
assert_err_pat!(get_exif_attr(&mut Cursor::new(data)), assert_err_pat!(get_exif_attr(&mut &data[..]),
Error::NotFound(_)); Error::NotFound(_));
} }
#[test] #[test]
fn empty() { fn empty() {
let data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\0eXIfCRC_"; let data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\0eXIfCRC_";
assert_ok!(get_exif_attr(&mut Cursor::new(data)), []); assert_ok!(get_exif_attr(&mut &data[..]), []);
} }
#[test] #[test]
fn non_empty() { fn non_empty() {
let data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\x02eXIf\xbe\xadCRC_"; let data = b"\x89PNG\x0d\x0a\x1a\x0a\0\0\0\x02eXIf\xbe\xadCRC_";
assert_ok!(get_exif_attr(&mut Cursor::new(data)), [0xbe, 0xad]); assert_ok!(get_exif_attr(&mut &data[..]), [0xbe, 0xad]);
} }
} }

View File

@ -94,18 +94,18 @@ pub fn ctou32(c: u8) -> Result<u32, Error> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use std::io::Cursor;
use std::io::ErrorKind; use std::io::ErrorKind;
use std::io::Read; use std::io::Read;
use super::*; use super::*;
#[test] #[test]
fn read8_len() { fn read8_len() {
let mut reader = Cursor::new([]); let data = [];
assert_err_kind!(read8(&mut reader), ErrorKind::UnexpectedEof); assert_err_kind!(read8(&mut &data[..]), ErrorKind::UnexpectedEof);
let mut reader = Cursor::new([0x01]); let data = [0x01];
assert_ok!(read8(&mut reader), 0x01); assert_ok!(read8(&mut &data[..]), 0x01);
let mut reader = Cursor::new([0x01, 0x02]); let data = [0x01, 0x02];
let mut reader = &data[..];
let mut buf = Vec::new(); let mut buf = Vec::new();
assert_ok!(read8(&mut reader), 0x01); assert_ok!(read8(&mut reader), 0x01);
assert_ok!(reader.read_to_end(&mut buf), 1); assert_ok!(reader.read_to_end(&mut buf), 1);
@ -114,13 +114,14 @@ mod tests {
#[test] #[test]
fn read16_len() { fn read16_len() {
let mut reader = Cursor::new([]); let data = [];
assert_err_kind!(read16(&mut reader), ErrorKind::UnexpectedEof); assert_err_kind!(read16(&mut &data[..]), ErrorKind::UnexpectedEof);
let mut reader = Cursor::new([0x01]); let data = [0x01];
assert_err_kind!(read16(&mut reader), ErrorKind::UnexpectedEof); assert_err_kind!(read16(&mut &data[..]), ErrorKind::UnexpectedEof);
let mut reader = Cursor::new([0x01, 0x02]); let data = [0x01, 0x02];
assert_ok!(read16(&mut reader), 0x0102); assert_ok!(read16(&mut &data[..]), 0x0102);
let mut reader = Cursor::new([0x01, 0x02, 0x03]); let data = [0x01, 0x02, 0x03];
let mut reader = &data[..];
let mut buf = Vec::new(); let mut buf = Vec::new();
assert_ok!(read16(&mut reader), 0x0102); assert_ok!(read16(&mut reader), 0x0102);
assert_ok!(reader.read_to_end(&mut buf), 1); assert_ok!(reader.read_to_end(&mut buf), 1);