Readur/src/tests/document_response_serializa...

135 lines
5.4 KiB
Rust

#[cfg(test)]
mod tests {
use crate::models::DocumentResponse;
use serde_json;
use uuid::Uuid;
use chrono::{DateTime, Utc};
#[test]
fn test_document_response_deserializes_without_new_metadata_fields() {
// Simulate an old server response without the new metadata fields
let old_response_json = r#"{
"id": "550e8400-e29b-41d4-a716-446655440000",
"filename": "test.pdf",
"original_filename": "test.pdf",
"file_path": "/storage/test.pdf",
"file_size": 1024,
"mime_type": "application/pdf",
"tags": [],
"labels": [],
"created_at": "2024-01-01T12:00:00Z",
"updated_at": "2024-01-01T12:00:00Z",
"user_id": "550e8400-e29b-41d4-a716-446655440001",
"has_ocr_text": false,
"ocr_confidence": null,
"ocr_word_count": null,
"ocr_processing_time_ms": null,
"ocr_status": null
}"#;
// Test deserialization - this should work with our serde defaults
let result: Result<DocumentResponse, _> = serde_json::from_str(old_response_json);
assert!(result.is_ok(), "Should deserialize successfully without metadata fields");
let doc = result.unwrap();
assert_eq!(doc.filename, "test.pdf");
assert!(doc.original_created_at.is_none());
assert!(doc.original_modified_at.is_none());
assert!(doc.source_metadata.is_none());
}
#[test]
fn test_document_response_deserializes_with_new_metadata_fields() {
// Simulate a new server response with the metadata fields
let new_response_json = r#"{
"id": "550e8400-e29b-41d4-a716-446655440000",
"filename": "test.pdf",
"original_filename": "test.pdf",
"file_path": "/storage/test.pdf",
"file_size": 1024,
"mime_type": "application/pdf",
"tags": [],
"labels": [],
"created_at": "2024-01-01T12:00:00Z",
"updated_at": "2024-01-01T12:00:00Z",
"user_id": "550e8400-e29b-41d4-a716-446655440001",
"has_ocr_text": false,
"ocr_confidence": null,
"ocr_word_count": null,
"ocr_processing_time_ms": null,
"ocr_status": null,
"original_created_at": "2023-12-01T10:00:00Z",
"original_modified_at": "2023-12-15T15:30:00Z",
"source_metadata": {"permissions": "644", "owner": "user1"}
}"#;
// Test deserialization
let result: Result<DocumentResponse, _> = serde_json::from_str(new_response_json);
assert!(result.is_ok(), "Should deserialize successfully with metadata fields");
let doc = result.unwrap();
assert_eq!(doc.filename, "test.pdf");
assert!(doc.original_created_at.is_some());
assert!(doc.original_modified_at.is_some());
assert!(doc.source_metadata.is_some());
// Verify metadata content
let metadata = doc.source_metadata.unwrap();
assert_eq!(metadata["permissions"], "644");
assert_eq!(metadata["owner"], "user1");
}
#[test]
fn test_document_response_serializes_with_metadata_fields() {
use crate::models::Document;
// Create a test document with metadata
let doc = Document {
id: Uuid::parse_str("550e8400-e29b-41d4-a716-446655440000").unwrap(),
filename: "test.pdf".to_string(),
original_filename: "test.pdf".to_string(),
file_path: "/test/test.pdf".to_string(),
file_size: 1024,
mime_type: "application/pdf".to_string(),
content: None,
ocr_text: None,
ocr_confidence: None,
ocr_word_count: None,
ocr_processing_time_ms: None,
ocr_status: None,
ocr_error: None,
ocr_completed_at: None,
tags: vec![],
created_at: DateTime::parse_from_rfc3339("2024-01-01T12:00:00Z").unwrap().with_timezone(&Utc),
updated_at: DateTime::parse_from_rfc3339("2024-01-01T12:00:00Z").unwrap().with_timezone(&Utc),
user_id: Uuid::new_v4(),
file_hash: Some("abcd1234".to_string()),
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_path: Some("/original/path/test.pdf".to_string()),
source_type: Some("filesystem".to_string()),
source_id: None,
file_permissions: Some(644),
file_owner: Some("user1".to_string()),
file_group: Some("users".to_string()),
source_metadata: Some(serde_json::json!({"permissions": "644", "owner": "user1"})),
ocr_retry_count: None,
ocr_failure_reason: None,
};
// Convert to DocumentResponse
let response: DocumentResponse = doc.into();
// Serialize to JSON
let json = serde_json::to_string(&response).unwrap();
// Should contain the metadata fields
assert!(json.contains("original_created_at"));
assert!(json.contains("original_modified_at"));
assert!(json.contains("source_metadata"));
assert!(json.contains("permissions"));
assert!(json.contains("644"));
}
}