feat(tvix/nix-compat/nar/reader): add tests

Change-Id: If0abaf74a3b7d201af8972e54e3e34b20b8c0bc4
Reviewed-on: https://cl.tvl.fyi/c/depot/+/9764
Tested-by: BuildkiteCI
Reviewed-by: Connor Brewster <cbrewster@hey.com>
This commit is contained in:
Florian Klink 2023-10-18 13:27:37 +01:00 committed by flokli
parent e64acba454
commit 05bca22121
5 changed files with 122 additions and 0 deletions

View file

@ -14,6 +14,8 @@ use std::io::{
use crate::nar::wire;
mod read;
#[cfg(test)]
mod test;
pub type Reader<'a> = dyn Read + Send + 'a;

View file

@ -0,0 +1,120 @@
use std::io::Read;
use crate::nar;
#[test]
fn symlink() {
let mut f = std::io::Cursor::new(include_bytes!("../tests/symlink.nar"));
let node = nar::reader::open(&mut f).unwrap();
match node {
nar::reader::Node::Symlink { target } => {
assert_eq!(
&b"/nix/store/somewhereelse"[..],
&target,
"target must match"
);
}
_ => panic!("unexpected type"),
}
}
#[test]
fn file() {
let mut f = std::io::Cursor::new(include_bytes!("../tests/helloworld.nar"));
let node = nar::reader::open(&mut f).unwrap();
match node {
nar::reader::Node::File {
executable,
mut reader,
} => {
assert!(!executable);
let mut buf = vec![];
reader.read_to_end(&mut buf).expect("read must succeed");
assert_eq!(&b"Hello World!"[..], &buf);
}
_ => panic!("unexpected type"),
}
}
#[test]
fn complicated() {
let mut f = std::io::Cursor::new(include_bytes!("../tests/complicated.nar"));
let node = nar::reader::open(&mut f).unwrap();
match node {
nar::reader::Node::Directory(mut dir_reader) => {
// first entry is .keep, an empty regular file.
let entry = dir_reader
.next()
.expect("next must succeed")
.expect("must be some");
assert_eq!(&b".keep"[..], &entry.name);
match entry.node {
nar::reader::Node::File {
executable,
mut reader,
} => {
assert!(!executable);
assert_eq!(reader.read(&mut [0]).unwrap(), 0);
}
_ => panic!("unexpected type for .keep"),
}
// second entry is aa, a symlink to /nix/store/somewhereelse
let entry = dir_reader
.next()
.expect("next must be some")
.expect("must be some");
assert_eq!(&b"aa"[..], &entry.name);
match entry.node {
nar::reader::Node::Symlink { target } => {
assert_eq!(&b"/nix/store/somewhereelse"[..], &target);
}
_ => panic!("unexpected type for aa"),
}
// third entry is a directory called "keep"
let entry = dir_reader
.next()
.expect("next must be some")
.expect("must be some");
assert_eq!(&b"keep"[..], &entry.name);
match entry.node {
nar::reader::Node::Directory(mut subdir_reader) => {
// first entry is .keep, an empty regular file.
let entry = subdir_reader
.next()
.expect("next must succeed")
.expect("must be some");
// … it contains a single .keep, an empty regular file.
assert_eq!(&b".keep"[..], &entry.name);
match entry.node {
nar::reader::Node::File {
executable,
mut reader,
} => {
assert!(!executable);
assert_eq!(reader.read(&mut [0]).unwrap(), 0);
}
_ => panic!("unexpected type for keep/.keep"),
}
}
_ => panic!("unexpected type for keep/.keep"),
}
// reading more entries yields None (and we actually must read until this)
assert!(dir_reader.next().expect("must succeed").is_none());
}
_ => panic!("unexpected type"),
}
}

Binary file not shown.

Binary file not shown.

Binary file not shown.