fix(tests): resolve compilation error in the multiple OCR functionality
This commit is contained in:
perf3ct
parent
e0b0f49ba2
commit
8d1a886139
156
src/ocr/tests.rs
156
src/ocr/tests.rs
|
|
@ -8,8 +8,6 @@ mod tests {
|
||||||
use tempfile::TempDir;
|
use tempfile::TempDir;
|
||||||
use std::fs;
|
use std::fs;
|
||||||
|
|
||||||
// Include language validation tests
|
|
||||||
mod language_validation_tests;
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_ocr_error_types() {
|
fn test_ocr_error_types() {
|
||||||
|
|
@ -212,4 +210,158 @@ mod tests {
|
||||||
panic!("Expected InsufficientMemory error, got: {:?}", result);
|
panic!("Expected InsufficientMemory error, got: {:?}", result);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Language validation tests
|
||||||
|
fn create_test_health_checker_with_languages() -> (OcrHealthChecker, TempDir) {
|
||||||
|
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
||||||
|
let tessdata_path = temp_dir.path().join("tessdata");
|
||||||
|
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
||||||
|
|
||||||
|
// Create mock language files
|
||||||
|
let language_files = vec![
|
||||||
|
"eng.traineddata",
|
||||||
|
"spa.traineddata",
|
||||||
|
"fra.traineddata",
|
||||||
|
"deu.traineddata",
|
||||||
|
"chi_sim.traineddata",
|
||||||
|
];
|
||||||
|
|
||||||
|
for file in language_files {
|
||||||
|
fs::write(tessdata_path.join(file), "mock data")
|
||||||
|
.expect("Failed to create mock language file");
|
||||||
|
}
|
||||||
|
|
||||||
|
let health_checker = OcrHealthChecker::new(tessdata_path);
|
||||||
|
(health_checker, temp_dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_available_languages_success() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
let result = health_checker.get_available_languages();
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
let languages = result.unwrap();
|
||||||
|
assert_eq!(languages.len(), 5);
|
||||||
|
assert!(languages.contains(&"eng".to_string()));
|
||||||
|
assert!(languages.contains(&"spa".to_string()));
|
||||||
|
assert!(languages.contains(&"fra".to_string()));
|
||||||
|
assert!(languages.contains(&"deu".to_string()));
|
||||||
|
assert!(languages.contains(&"chi_sim".to_string()));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_validate_language_success() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
// Test valid languages
|
||||||
|
assert!(health_checker.validate_language("eng").is_ok());
|
||||||
|
assert!(health_checker.validate_language("spa").is_ok());
|
||||||
|
assert!(health_checker.validate_language("fra").is_ok());
|
||||||
|
assert!(health_checker.validate_language("deu").is_ok());
|
||||||
|
assert!(health_checker.validate_language("chi_sim").is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_validate_language_invalid() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
// Test invalid languages
|
||||||
|
let result = health_checker.validate_language("invalid");
|
||||||
|
assert!(result.is_err());
|
||||||
|
match result.unwrap_err() {
|
||||||
|
OcrError::LanguageDataNotFound { lang } => {
|
||||||
|
assert_eq!(lang, "invalid");
|
||||||
|
},
|
||||||
|
_ => panic!("Expected LanguageDataNotFound error"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_validate_language_case_sensitive() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
// Should be case sensitive
|
||||||
|
assert!(health_checker.validate_language("eng").is_ok());
|
||||||
|
|
||||||
|
let result = health_checker.validate_language("ENG");
|
||||||
|
assert!(result.is_err());
|
||||||
|
match result.unwrap_err() {
|
||||||
|
OcrError::LanguageDataNotFound { lang } => {
|
||||||
|
assert_eq!(lang, "ENG");
|
||||||
|
},
|
||||||
|
_ => panic!("Expected LanguageDataNotFound error"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_language_display_name() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
// Test known language codes
|
||||||
|
assert_eq!(health_checker.get_language_display_name("eng"), "English");
|
||||||
|
assert_eq!(health_checker.get_language_display_name("spa"), "Spanish");
|
||||||
|
assert_eq!(health_checker.get_language_display_name("fra"), "French");
|
||||||
|
assert_eq!(health_checker.get_language_display_name("deu"), "German");
|
||||||
|
assert_eq!(health_checker.get_language_display_name("chi_sim"), "Chinese (Simplified)");
|
||||||
|
|
||||||
|
// Test unknown language code (should return the code itself)
|
||||||
|
assert_eq!(health_checker.get_language_display_name("unknown"), "unknown");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_ignore_non_traineddata_files() {
|
||||||
|
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
||||||
|
let tessdata_path = temp_dir.path().join("tessdata");
|
||||||
|
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
||||||
|
|
||||||
|
// Create mix of valid and invalid files
|
||||||
|
let files = vec![
|
||||||
|
"eng.traineddata", // Valid
|
||||||
|
"readme.txt", // Invalid - not .traineddata
|
||||||
|
"spa.traineddata", // Valid
|
||||||
|
"config.json", // Invalid - not .traineddata
|
||||||
|
"fra.backup", // Invalid - not .traineddata
|
||||||
|
"deu.traineddata", // Valid
|
||||||
|
];
|
||||||
|
|
||||||
|
for file in files {
|
||||||
|
fs::write(tessdata_path.join(file), "mock data")
|
||||||
|
.expect("Failed to create mock file");
|
||||||
|
}
|
||||||
|
|
||||||
|
let health_checker = OcrHealthChecker::new(tessdata_path);
|
||||||
|
let languages = health_checker.get_available_languages().unwrap();
|
||||||
|
|
||||||
|
// Should only include .traineddata files
|
||||||
|
assert_eq!(languages.len(), 3);
|
||||||
|
assert!(languages.contains(&"eng".to_string()));
|
||||||
|
assert!(languages.contains(&"spa".to_string()));
|
||||||
|
assert!(languages.contains(&"deu".to_string()));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_validate_multiple_languages_batch() {
|
||||||
|
let (health_checker, _temp_dir) = create_test_health_checker_with_languages();
|
||||||
|
|
||||||
|
let languages_to_test = vec![
|
||||||
|
("eng", true),
|
||||||
|
("spa", true),
|
||||||
|
("fra", true),
|
||||||
|
("invalid", false),
|
||||||
|
("", false),
|
||||||
|
("ENG", false),
|
||||||
|
("chi_sim", true),
|
||||||
|
];
|
||||||
|
|
||||||
|
for (lang, should_be_valid) in languages_to_test {
|
||||||
|
let result = health_checker.validate_language(lang);
|
||||||
|
if should_be_valid {
|
||||||
|
assert!(result.is_ok(), "Language '{}' should be valid", lang);
|
||||||
|
} else {
|
||||||
|
assert!(result.is_err(), "Language '{}' should be invalid", lang);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -1,298 +0,0 @@
|
||||||
#[cfg(test)]
|
|
||||||
mod language_validation_tests {
|
|
||||||
use super::super::health::{OcrHealthChecker, OcrError};
|
|
||||||
use std::path::Path;
|
|
||||||
use tempfile::TempDir;
|
|
||||||
use std::fs;
|
|
||||||
|
|
||||||
fn create_test_health_checker() -> (OcrHealthChecker, TempDir) {
|
|
||||||
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
|
||||||
let tessdata_path = temp_dir.path().join("tessdata");
|
|
||||||
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
|
||||||
|
|
||||||
// Create mock language files
|
|
||||||
let language_files = vec![
|
|
||||||
"eng.traineddata",
|
|
||||||
"spa.traineddata",
|
|
||||||
"fra.traineddata",
|
|
||||||
"deu.traineddata",
|
|
||||||
"chi_sim.traineddata",
|
|
||||||
];
|
|
||||||
|
|
||||||
for file in language_files {
|
|
||||||
fs::write(tessdata_path.join(file), "mock data")
|
|
||||||
.expect("Failed to create mock language file");
|
|
||||||
}
|
|
||||||
|
|
||||||
let health_checker = OcrHealthChecker::new(tessdata_path);
|
|
||||||
(health_checker, temp_dir)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_get_available_languages_success() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
let result = health_checker.get_available_languages();
|
|
||||||
assert!(result.is_ok());
|
|
||||||
|
|
||||||
let languages = result.unwrap();
|
|
||||||
assert_eq!(languages.len(), 5);
|
|
||||||
assert!(languages.contains(&"eng".to_string()));
|
|
||||||
assert!(languages.contains(&"spa".to_string()));
|
|
||||||
assert!(languages.contains(&"fra".to_string()));
|
|
||||||
assert!(languages.contains(&"deu".to_string()));
|
|
||||||
assert!(languages.contains(&"chi_sim".to_string()));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_get_available_languages_empty_directory() {
|
|
||||||
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
|
||||||
let tessdata_path = temp_dir.path().join("tessdata");
|
|
||||||
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
|
||||||
|
|
||||||
let health_checker = OcrHealthChecker::new(tessdata_path);
|
|
||||||
let result = health_checker.get_available_languages();
|
|
||||||
|
|
||||||
assert!(result.is_ok());
|
|
||||||
let languages = result.unwrap();
|
|
||||||
assert!(languages.is_empty());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_get_available_languages_nonexistent_directory() {
|
|
||||||
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
|
||||||
let nonexistent_path = temp_dir.path().join("nonexistent");
|
|
||||||
|
|
||||||
let health_checker = OcrHealthChecker::new(nonexistent_path);
|
|
||||||
let result = health_checker.get_available_languages();
|
|
||||||
|
|
||||||
assert!(result.is_err());
|
|
||||||
match result.unwrap_err() {
|
|
||||||
OcrError::TessdataPathNotFound { .. } => {},
|
|
||||||
_ => panic!("Expected TessdataPathNotFound error"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_success() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// Test valid languages
|
|
||||||
assert!(health_checker.validate_language("eng").is_ok());
|
|
||||||
assert!(health_checker.validate_language("spa").is_ok());
|
|
||||||
assert!(health_checker.validate_language("fra").is_ok());
|
|
||||||
assert!(health_checker.validate_language("deu").is_ok());
|
|
||||||
assert!(health_checker.validate_language("chi_sim").is_ok());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_invalid() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// Test invalid languages
|
|
||||||
let result = health_checker.validate_language("invalid");
|
|
||||||
assert!(result.is_err());
|
|
||||||
match result.unwrap_err() {
|
|
||||||
OcrError::LanguageDataNotFound { lang } => {
|
|
||||||
assert_eq!(lang, "invalid");
|
|
||||||
},
|
|
||||||
_ => panic!("Expected LanguageDataNotFound error"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_empty_string() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
let result = health_checker.validate_language("");
|
|
||||||
assert!(result.is_err());
|
|
||||||
match result.unwrap_err() {
|
|
||||||
OcrError::LanguageDataNotFound { lang } => {
|
|
||||||
assert_eq!(lang, "");
|
|
||||||
},
|
|
||||||
_ => panic!("Expected LanguageDataNotFound error"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_case_sensitive() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// Should be case sensitive
|
|
||||||
assert!(health_checker.validate_language("eng").is_ok());
|
|
||||||
|
|
||||||
let result = health_checker.validate_language("ENG");
|
|
||||||
assert!(result.is_err());
|
|
||||||
match result.unwrap_err() {
|
|
||||||
OcrError::LanguageDataNotFound { lang } => {
|
|
||||||
assert_eq!(lang, "ENG");
|
|
||||||
},
|
|
||||||
_ => panic!("Expected LanguageDataNotFound error"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_with_special_characters() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// chi_sim contains underscore
|
|
||||||
assert!(health_checker.validate_language("chi_sim").is_ok());
|
|
||||||
|
|
||||||
// Test invalid special characters
|
|
||||||
let result = health_checker.validate_language("chi-sim");
|
|
||||||
assert!(result.is_err());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_language_whitespace() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// Test with leading/trailing whitespace
|
|
||||||
let result = health_checker.validate_language(" eng ");
|
|
||||||
assert!(result.is_err());
|
|
||||||
|
|
||||||
let result = health_checker.validate_language("eng ");
|
|
||||||
assert!(result.is_err());
|
|
||||||
|
|
||||||
let result = health_checker.validate_language(" eng");
|
|
||||||
assert!(result.is_err());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_get_language_display_name() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
// Test known language codes
|
|
||||||
assert_eq!(health_checker.get_language_display_name("eng"), "English");
|
|
||||||
assert_eq!(health_checker.get_language_display_name("spa"), "Spanish");
|
|
||||||
assert_eq!(health_checker.get_language_display_name("fra"), "French");
|
|
||||||
assert_eq!(health_checker.get_language_display_name("deu"), "German");
|
|
||||||
assert_eq!(health_checker.get_language_display_name("chi_sim"), "Chinese (Simplified)");
|
|
||||||
|
|
||||||
// Test unknown language code (should return the code itself)
|
|
||||||
assert_eq!(health_checker.get_language_display_name("unknown"), "unknown");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_concurrent_language_validation() {
|
|
||||||
use std::sync::Arc;
|
|
||||||
use std::thread;
|
|
||||||
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
let health_checker = Arc::new(health_checker);
|
|
||||||
|
|
||||||
let mut handles = vec![];
|
|
||||||
|
|
||||||
// Test concurrent validation of different languages
|
|
||||||
for lang in &["eng", "spa", "fra", "deu", "chi_sim"] {
|
|
||||||
let hc = Arc::clone(&health_checker);
|
|
||||||
let lang = lang.to_string();
|
|
||||||
let handle = thread::spawn(move || {
|
|
||||||
hc.validate_language(&lang)
|
|
||||||
});
|
|
||||||
handles.push(handle);
|
|
||||||
}
|
|
||||||
|
|
||||||
// All validations should succeed
|
|
||||||
for handle in handles {
|
|
||||||
let result = handle.join().expect("Thread panicked");
|
|
||||||
assert!(result.is_ok());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_languages_alphabetically_sorted() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
let languages = health_checker.get_available_languages().unwrap();
|
|
||||||
let mut sorted_languages = languages.clone();
|
|
||||||
sorted_languages.sort();
|
|
||||||
|
|
||||||
assert_eq!(languages, sorted_languages, "Languages should be sorted alphabetically");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_ignore_non_traineddata_files() {
|
|
||||||
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
|
||||||
let tessdata_path = temp_dir.path().join("tessdata");
|
|
||||||
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
|
||||||
|
|
||||||
// Create mix of valid and invalid files
|
|
||||||
let files = vec![
|
|
||||||
"eng.traineddata", // Valid
|
|
||||||
"readme.txt", // Invalid - not .traineddata
|
|
||||||
"spa.traineddata", // Valid
|
|
||||||
"config.json", // Invalid - not .traineddata
|
|
||||||
"fra.backup", // Invalid - not .traineddata
|
|
||||||
"deu.traineddata", // Valid
|
|
||||||
];
|
|
||||||
|
|
||||||
for file in files {
|
|
||||||
fs::write(tessdata_path.join(file), "mock data")
|
|
||||||
.expect("Failed to create mock file");
|
|
||||||
}
|
|
||||||
|
|
||||||
let health_checker = OcrHealthChecker::new(tessdata_path);
|
|
||||||
let languages = health_checker.get_available_languages().unwrap();
|
|
||||||
|
|
||||||
// Should only include .traineddata files
|
|
||||||
assert_eq!(languages.len(), 3);
|
|
||||||
assert!(languages.contains(&"eng".to_string()));
|
|
||||||
assert!(languages.contains(&"spa".to_string()));
|
|
||||||
assert!(languages.contains(&"deu".to_string()));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_handle_permission_errors() {
|
|
||||||
// This test simulates permission errors by using a non-readable directory
|
|
||||||
// Note: This may not work on all systems, particularly Windows
|
|
||||||
#[cfg(unix)]
|
|
||||||
{
|
|
||||||
use std::os::unix::fs::PermissionsExt;
|
|
||||||
|
|
||||||
let temp_dir = TempDir::new().expect("Failed to create temp directory");
|
|
||||||
let tessdata_path = temp_dir.path().join("tessdata");
|
|
||||||
fs::create_dir_all(&tessdata_path).expect("Failed to create tessdata directory");
|
|
||||||
|
|
||||||
// Remove read permissions
|
|
||||||
let mut perms = fs::metadata(&tessdata_path).unwrap().permissions();
|
|
||||||
perms.set_mode(0o000);
|
|
||||||
fs::set_permissions(&tessdata_path, perms).unwrap();
|
|
||||||
|
|
||||||
let health_checker = OcrHealthChecker::new(&tessdata_path);
|
|
||||||
let result = health_checker.get_available_languages();
|
|
||||||
|
|
||||||
// Should handle permission error gracefully
|
|
||||||
assert!(result.is_err());
|
|
||||||
|
|
||||||
// Restore permissions for cleanup
|
|
||||||
let mut perms = fs::metadata(&tessdata_path).unwrap().permissions();
|
|
||||||
perms.set_mode(0o755);
|
|
||||||
fs::set_permissions(&tessdata_path, perms).unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_validate_multiple_languages_batch() {
|
|
||||||
let (health_checker, _temp_dir) = create_test_health_checker();
|
|
||||||
|
|
||||||
let languages_to_test = vec![
|
|
||||||
("eng", true),
|
|
||||||
("spa", true),
|
|
||||||
("fra", true),
|
|
||||||
("invalid", false),
|
|
||||||
("", false),
|
|
||||||
("ENG", false),
|
|
||||||
("chi_sim", true),
|
|
||||||
];
|
|
||||||
|
|
||||||
for (lang, should_be_valid) in languages_to_test {
|
|
||||||
let result = health_checker.validate_language(lang);
|
|
||||||
if should_be_valid {
|
|
||||||
assert!(result.is_ok(), "Language '{}' should be valid", lang);
|
|
||||||
} else {
|
|
||||||
assert!(result.is_err(), "Language '{}' should be invalid", lang);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1 +0,0 @@
|
||||||
pub mod language_validation_tests;
|
|
||||||
Loading…
Reference in New Issue