96 lines
3.4 KiB
C
96 lines
3.4 KiB
C
/******************************************************************************
|
|
* @file test_pdf_analysis.c
|
|
* @brief Tests synthétiques de l'analyse PDF.
|
|
******************************************************************************/
|
|
#include "core/pdf_analysis.h"
|
|
#include <glib.h>
|
|
#include <glib/gstdio.h>
|
|
|
|
static PdfAnalysisTools fake_tools(void)
|
|
{
|
|
PdfAnalysisTools tools = {
|
|
.pdfinfo = "tests/fake_document_tool",
|
|
.pdftotext = "tests/fake_document_tool",
|
|
.pdftoppm = "tests/fake_document_tool",
|
|
.tesseract = "tests/fake_document_tool"
|
|
};
|
|
return tools;
|
|
}
|
|
|
|
static char *create_pdf(const char *directory, const char *name)
|
|
{
|
|
char *path = g_build_filename(directory, name, NULL);
|
|
g_assert_true(g_file_set_contents(path, "%PDF-synthetic", -1, NULL));
|
|
return path;
|
|
}
|
|
|
|
static void test_encrypted_and_native(void)
|
|
{
|
|
GError *error = NULL;
|
|
char *directory = g_dir_make_tmp("labfy-pdf-XXXXXX", &error);
|
|
PdfAnalysisTools tools = fake_tools();
|
|
char *encrypted = create_pdf(directory, "encrypted.pdf");
|
|
PdfAnalysisResult *result = pdf_analysis_run(
|
|
&tools, encrypted, "fra", NULL, &error);
|
|
g_assert_no_error(error);
|
|
g_assert_true(result->encrypted);
|
|
g_assert_cmpuint(result->pages->len, ==, 0);
|
|
pdf_analysis_result_free(result);
|
|
char *native = create_pdf(directory, "native.pdf");
|
|
result = pdf_analysis_run(&tools, native, "fra", NULL, &error);
|
|
g_assert_no_error(error);
|
|
g_assert_true(result->native_text_usable);
|
|
g_assert_cmpuint(result->pages->len, ==, 2);
|
|
PdfPageAnalysis *page = g_ptr_array_index(result->pages, 0);
|
|
g_assert_cmpint(page->method, ==, PDF_PAGE_METHOD_NATIVE);
|
|
pdf_analysis_result_free(result);
|
|
g_remove(encrypted);
|
|
g_remove(native);
|
|
g_rmdir(directory);
|
|
g_free(encrypted);
|
|
g_free(native);
|
|
g_free(directory);
|
|
}
|
|
|
|
static void test_ocr_fallback_order_and_cleanup(void)
|
|
{
|
|
GError *error = NULL;
|
|
char *directory = g_dir_make_tmp("labfy-pdf-XXXXXX", &error);
|
|
char *scan = create_pdf(directory, "scan.pdf");
|
|
PdfAnalysisTools tools = fake_tools();
|
|
PdfAnalysisResult *result = pdf_analysis_run(
|
|
&tools, scan, "fra+eng", NULL, &error);
|
|
g_assert_no_error(error);
|
|
g_assert_false(result->native_text_usable);
|
|
g_assert_cmpuint(result->pages->len, ==, 2);
|
|
PdfPageAnalysis *first = g_ptr_array_index(result->pages, 0);
|
|
PdfPageAnalysis *second = g_ptr_array_index(result->pages, 1);
|
|
g_assert_cmpuint(first->page_number, ==, 1);
|
|
g_assert_cmpuint(second->page_number, ==, 2);
|
|
g_assert_cmpint(first->method, ==, PDF_PAGE_METHOD_OCR);
|
|
g_assert_nonnull(strstr(second->text, "page deux"));
|
|
pdf_analysis_result_free(result);
|
|
g_remove(scan);
|
|
g_rmdir(directory);
|
|
g_free(scan);
|
|
g_free(directory);
|
|
}
|
|
|
|
static void test_heuristic(void)
|
|
{
|
|
g_assert_false(pdf_analysis_text_is_usable(""));
|
|
g_assert_false(pdf_analysis_text_is_usable("court"));
|
|
g_assert_true(pdf_analysis_text_is_usable(
|
|
"Texte synthétique imprimable et suffisamment long pour le test."));
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
g_test_init(&argc, &argv, NULL);
|
|
g_test_add_func("/pdf-analysis/encrypted-native",
|
|
test_encrypted_and_native);
|
|
g_test_add_func("/pdf-analysis/ocr-fallback-cleanup",
|
|
test_ocr_fallback_order_and_cleanup);
|
|
g_test_add_func("/pdf-analysis/heuristic", test_heuristic);
|
|
return g_test_run();
|
|
}
|