75 lines
2.3 KiB
Rust
75 lines
2.3 KiB
Rust
|
use std::fs::File;
|
||
|
|
use std::sync::Once;
|
||
|
|
use crate::cleanup::remove_compressed_test_file;
|
||
|
|
|
||
|
|
mod cleanup;
|
||
|
|
|
||
|
|
static INIT: Once = Once::new();
|
||
|
|
|
||
|
|
pub fn initialize(file: &str) {
|
||
|
|
INIT.call_once(|| {
|
||
|
|
remove_compressed_test_file(file);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn compress_to_1_byte() {
|
||
|
|
let output = "tests/samples/output/compressed_1b.jpg";
|
||
|
|
initialize(output);
|
||
|
|
let mut pars = caesium::initialize_parameters();
|
||
|
|
caesium::compress_to_size(
|
||
|
|
String::from("tests/samples/uncompressed_드림캐쳐.jpg"),
|
||
|
|
String::from(output),
|
||
|
|
&mut pars,
|
||
|
|
1
|
||
|
|
).expect_err("Cannot compress to desired quality");
|
||
|
|
remove_compressed_test_file(output)
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn compress_to_10_mb() {
|
||
|
|
let output = "tests/samples/output/compressed_10mb.jpg";
|
||
|
|
let desired_output_size = 10_000_000;
|
||
|
|
initialize(output);
|
||
|
|
let mut pars = caesium::initialize_parameters();
|
||
|
|
caesium::compress_to_size(
|
||
|
|
String::from("tests/samples/uncompressed_드림캐쳐.jpg"),
|
||
|
|
String::from(output),
|
||
|
|
&mut pars,
|
||
|
|
desired_output_size
|
||
|
|
).unwrap();
|
||
|
|
|
||
|
|
assert_eq!(100, pars.jpeg.quality);
|
||
|
|
assert!(std::path::Path::new(output).exists());
|
||
|
|
assert!(File::open(output).unwrap().metadata().unwrap().len() < desired_output_size as u64);
|
||
|
|
let kind = infer::get_from_path(output).unwrap().unwrap();
|
||
|
|
assert_eq!(kind.mime_type(), "image/jpeg");
|
||
|
|
remove_compressed_test_file(output)
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn compress_to_range() {
|
||
|
|
let output = "tests/samples/output/compressed_range.jpg";
|
||
|
|
let max_desired_size = 2_000_000;
|
||
|
|
let mut desired_output_size = 500_000;
|
||
|
|
initialize(output);
|
||
|
|
|
||
|
|
while desired_output_size < max_desired_size {
|
||
|
|
let mut pars = caesium::initialize_parameters();
|
||
|
|
caesium::compress_to_size(
|
||
|
|
String::from("tests/samples/uncompressed_드림캐쳐.jpg"),
|
||
|
|
String::from(output),
|
||
|
|
&mut pars,
|
||
|
|
desired_output_size
|
||
|
|
).unwrap();
|
||
|
|
|
||
|
|
assert!(std::path::Path::new(output).exists());
|
||
|
|
assert!(File::open(output).unwrap().metadata().unwrap().len() < desired_output_size as u64);
|
||
|
|
let kind = infer::get_from_path(output).unwrap().unwrap();
|
||
|
|
assert_eq!(kind.mime_type(), "image/jpeg");
|
||
|
|
desired_output_size += 100_000;
|
||
|
|
remove_compressed_test_file(output);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|