feat: add compression engine test helpers and refactor tests
Co-authored-by: aider (openai/andrew/openrouter/qwen/qwen3-coder) <aider@aider.chat>
This commit is contained in:
@@ -1,57 +1,20 @@
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use tempfile::tempdir;
|
||||
use std::io::Write;
|
||||
use crate::compression_engine::none::CompressionEngineNone;
|
||||
use crate::compression_engine::CompressionEngine;
|
||||
use crate::tests::common::test_helpers::test_compression_engine;
|
||||
|
||||
#[test]
|
||||
fn test_compression_engine_none() {
|
||||
let test_data = b"test compression data";
|
||||
|
||||
// Create a temporary file
|
||||
let dir = tempdir().expect("Failed to create temporary directory");
|
||||
let file_path = dir.path().join("test.dat");
|
||||
|
||||
// Test compression engine
|
||||
let engine = CompressionEngineNone {};
|
||||
|
||||
// Create file
|
||||
{
|
||||
let mut writer = engine.create(file_path.clone()).expect("Failed to create writer");
|
||||
writer.write_all(test_data).expect("Failed to write data");
|
||||
}
|
||||
|
||||
// Read file
|
||||
let mut reader = engine.open(file_path).expect("Failed to open reader");
|
||||
let mut data = Vec::new();
|
||||
std::io::copy(&mut reader, &mut data).expect("Failed to read data");
|
||||
|
||||
assert_eq!(test_data, data.as_slice());
|
||||
test_compression_engine(&engine, test_data);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compression_engine_none_empty_data() {
|
||||
let test_data = b"";
|
||||
|
||||
// Create a temporary file
|
||||
let dir = tempdir().expect("Failed to create temporary directory");
|
||||
let file_path = dir.path().join("test_empty.dat");
|
||||
|
||||
// Test compression engine
|
||||
let engine = CompressionEngineNone {};
|
||||
|
||||
// Create file
|
||||
{
|
||||
let mut writer = engine.create(file_path.clone()).expect("Failed to create writer");
|
||||
writer.write_all(test_data).expect("Failed to write data");
|
||||
}
|
||||
|
||||
// Read file
|
||||
let mut reader = engine.open(file_path).expect("Failed to open reader");
|
||||
let mut data = Vec::new();
|
||||
std::io::copy(&mut reader, &mut data).expect("Failed to read data");
|
||||
|
||||
assert_eq!(test_data, data.as_slice());
|
||||
test_compression_engine(&engine, test_data);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user