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:
		
							parent
							
								
									e64acba454
								
							
						
					
					
						commit
						05bca22121
					
				
					 5 changed files with 122 additions and 0 deletions
				
			
		| 
						 | 
					@ -14,6 +14,8 @@ use std::io::{
 | 
				
			||||||
use crate::nar::wire;
 | 
					use crate::nar::wire;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
mod read;
 | 
					mod read;
 | 
				
			||||||
 | 
					#[cfg(test)]
 | 
				
			||||||
 | 
					mod test;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
pub type Reader<'a> = dyn Read + Send + 'a;
 | 
					pub type Reader<'a> = dyn Read + Send + 'a;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
							
								
								
									
										120
									
								
								tvix/nix-compat/src/nar/reader/test.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										120
									
								
								tvix/nix-compat/src/nar/reader/test.rs
									
										
									
									
									
										Normal 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"),
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/complicated.nar
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/complicated.nar
									
										
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							
							
								
								
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/helloworld.nar
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/helloworld.nar
									
										
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							
							
								
								
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/symlink.nar
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								tvix/nix-compat/src/nar/tests/symlink.nar
									
										
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue