From e15aa801c6c5d698b830387ac8f39a243db2cc53 Mon Sep 17 00:00:00 2001 From: perf3ct Date: Wed, 2 Jul 2025 18:48:26 +0000 Subject: [PATCH] feat(tests): fix ocr_retry issues in tests --- src/tests/db_tests.rs | 2 ++ src/tests/document_response_serialization_tests.rs | 2 ++ src/tests/document_routes_tests.rs | 4 ++++ src/tests/documents_tests.rs | 8 ++++++++ src/tests/enhanced_search_tests.rs | 2 ++ src/tests/file_service_tests.rs | 10 ++++++++++ src/tests/ignored_files_tests.rs | 2 ++ 7 files changed, 30 insertions(+) diff --git a/src/tests/db_tests.rs b/src/tests/db_tests.rs index 123c3f7..dca299c 100644 --- a/src/tests/db_tests.rs +++ b/src/tests/db_tests.rs @@ -52,6 +52,8 @@ mod tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } diff --git a/src/tests/document_response_serialization_tests.rs b/src/tests/document_response_serialization_tests.rs index d97741f..5b69fec 100644 --- a/src/tests/document_response_serialization_tests.rs +++ b/src/tests/document_response_serialization_tests.rs @@ -103,6 +103,8 @@ mod tests { original_created_at: Some(DateTime::parse_from_rfc3339("2023-12-01T10:00:00Z").unwrap().with_timezone(&Utc)), original_modified_at: Some(DateTime::parse_from_rfc3339("2023-12-15T15:30:00Z").unwrap().with_timezone(&Utc)), source_metadata: Some(serde_json::json!({"permissions": "644", "owner": "user1"})), + ocr_retry_count: None, + ocr_failure_reason: None, }; // Convert to DocumentResponse diff --git a/src/tests/document_routes_tests.rs b/src/tests/document_routes_tests.rs index 5582a4b..3e7af69 100644 --- a/src/tests/document_routes_tests.rs +++ b/src/tests/document_routes_tests.rs @@ -63,6 +63,8 @@ mod document_routes_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } @@ -400,6 +402,8 @@ mod document_routes_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } diff --git a/src/tests/documents_tests.rs b/src/tests/documents_tests.rs index 2340ff9..9973974 100644 --- a/src/tests/documents_tests.rs +++ b/src/tests/documents_tests.rs @@ -29,6 +29,8 @@ fn create_test_document(user_id: Uuid) -> Document { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } @@ -57,6 +59,8 @@ fn create_test_document_without_ocr(user_id: Uuid) -> Document { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } @@ -85,6 +89,8 @@ fn create_test_document_with_ocr_error(user_id: Uuid) -> Document { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } @@ -1564,6 +1570,8 @@ mod deletion_error_handling_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, } } diff --git a/src/tests/enhanced_search_tests.rs b/src/tests/enhanced_search_tests.rs index dc609ae..4da7afd 100644 --- a/src/tests/enhanced_search_tests.rs +++ b/src/tests/enhanced_search_tests.rs @@ -942,6 +942,8 @@ mod tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; db.create_document(document).await.unwrap(); diff --git a/src/tests/file_service_tests.rs b/src/tests/file_service_tests.rs index a1ad8b4..ba3a6b6 100644 --- a/src/tests/file_service_tests.rs +++ b/src/tests/file_service_tests.rs @@ -195,6 +195,8 @@ mod file_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; ( @@ -333,6 +335,8 @@ mod file_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; // Try to delete nonexistent files (should not fail) @@ -387,6 +391,8 @@ mod file_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; // Verify files exist @@ -445,6 +451,8 @@ mod file_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; // Verify files exist @@ -494,6 +502,8 @@ mod file_deletion_tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; // Verify file exists diff --git a/src/tests/ignored_files_tests.rs b/src/tests/ignored_files_tests.rs index 7128c25..dc93cc9 100644 --- a/src/tests/ignored_files_tests.rs +++ b/src/tests/ignored_files_tests.rs @@ -84,6 +84,8 @@ mod tests { original_created_at: None, original_modified_at: None, source_metadata: None, + ocr_retry_count: None, + ocr_failure_reason: None, }; sqlx::query("INSERT INTO documents (id, filename, original_filename, file_path, file_size, mime_type, content, ocr_text, ocr_confidence, ocr_word_count, ocr_processing_time_ms, ocr_status, ocr_error, ocr_completed_at, tags, created_at, updated_at, user_id, file_hash) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19)")