readur/tests/integration_ocr_retry_db_tests.rs

#[cfg(test)]
mod tests {
    use readur::db::ocr_retry::*;
    use readur::test_utils::{TestContext, TestAuthHelper};
    use sqlx::Row;
    use uuid::Uuid;

    #[tokio::test]
    async fn test_simple_retry_record() {
        let ctx = TestContext::new().await;
        let auth_helper = TestAuthHelper::new(ctx.app.clone());
        let user = auth_helper.create_test_user().await;
        
        // Create a test document using the TestContext database
        let doc_id = Uuid::new_v4();
        sqlx::query("INSERT INTO documents (id, filename, original_filename, user_id, mime_type, file_size, created_at, updated_at, file_path) VALUES ($1, 'test.pdf', 'test.pdf', $2, 'application/pdf', 1024, NOW(), NOW(), '/test/test.pdf')")
            .bind(doc_id)
            .bind(user.user_response.id)
            .execute(&ctx.state.db.pool)
            .await
            .expect("Failed to create test document");
        
        // Test the record_ocr_retry function
        let retry_id = record_ocr_retry(
            &ctx.state.db.pool,
            doc_id,
            user.user_response.id,
            "manual_retry",
            10,
            None,
        ).await.expect("Failed to record retry");
        
        // Verify the retry was recorded
        let count: i64 = sqlx::query_scalar("SELECT COUNT(*) FROM ocr_retry_history WHERE id = $1")
            .bind(retry_id)
            .fetch_one(&ctx.state.db.pool)
            .await
            .expect("Failed to count retries");
            
        assert_eq!(count, 1);
    }
}