tests/mime.rs

   1 #[macro_use]
   2 mod utils;
   3
   4 use rand::{rngs::SmallRng, SeedableRng};
   5 use tempfile::NamedTempFile;
   6
   7 use crate::utils::write_random_content;
   8
   9 #[test]
  10 /// Makes sure that the files ouch produces are what they claim to be, checking their
  11 /// types through MIME sniffing.
  12 fn sanity_check_through_mime() {
  13     let temp_dir = tempfile::tempdir().expect("to build a temporary directory");
  14     let temp_dir_path = temp_dir.path();
  15
  16     let test_file = &mut NamedTempFile::new_in(temp_dir_path).expect("to be able to build a temporary file");
  17     write_random_content(test_file, &mut SmallRng::from_entropy());
  18
  19     let formats = [
  20         "tar", "zip", "tar.gz", "tgz", "tbz", "tbz2", "txz", "tlzma", "tzst", "tar.bz", "tar.bz2", "tar.lzma",
  21         "tar.xz", "tar.zst",
  22     ];
  23
  24     let expected_mimes = [
  25         "application/x-tar",
  26         "application/zip",
  27         "application/gzip",
  28         "application/gzip",
  29         "application/x-bzip2",
  30         "application/x-bzip2",
  31         "application/x-xz",
  32         "application/x-xz",
  33         "application/zstd",
  34         "application/x-bzip2",
  35         "application/x-bzip2",
  36         "application/x-xz",
  37         "application/x-xz",
  38         "application/zstd",
  39     ];
  40
  41     assert_eq!(formats.len(), expected_mimes.len());
  42
  43     for (format, expected_mime) in formats.iter().zip(expected_mimes.iter()) {
  44         let path_to_compress = test_file.path();
  45
  46         let compressed_file_path = &format!("{}.{}", path_to_compress.display(), format);
  47         ouch!("c", path_to_compress, compressed_file_path);
  48
  49         let sniffed = infer::get_from_path(compressed_file_path)
  50             .expect("the file to be read")
  51             .expect("the MIME to be found");
  52
  53         assert_eq!(&sniffed.mime_type(), expected_mime);
  54     }
  55 }