InvoiceSplitterAutoExtractionTestFunctional.php

<?php

declare(strict_types=1);

namespace V1\Image;

use Mindee\Input\PathInput;
use Mindee\V1\Pdf\PdfExtractor;
use Mindee\V1\Client;
use Mindee\V1\Parsing\Common\Document;
use Mindee\V1\Product\Invoice\InvoiceV4;
use Mindee\V1\Product\InvoiceSplitter\InvoiceSplitterV1;
use PHPUnit\Framework\TestCase;
use TestingUtilities;

require_once(__DIR__ . "/../../TestingUtilities.php");

class InvoiceSplitterAutoExtractionTestFunctional extends TestCase
{
    private const PRODUCT_DATA_DIR = '/tests/resources/products';

    private function prepareInvoiceReturn(string $rstFilePath, Document $invoicePrediction): string
    {
        $fileContent = file_get_contents($rstFilePath);
        $parsingVersion = $invoicePrediction->inference->product->version;
        $parsingId = $invoicePrediction->id;
        $rstContent = str_replace(TestingUtilities::getVersion($fileContent), $parsingVersion, $fileContent);
        return str_replace(TestingUtilities::getId($rstContent), $parsingId, $rstContent);
    }

    /**
     *
     * @group functional
     */
    public function testPdfShouldExtractInvoicesStrict(): void
    {
        $client = new Client();
        $invoiceSplitterInput = new PathInput(
            filePath: TestingUtilities::getV1DataDir() . '/products/invoice_splitter/default_sample.pdf'
        );
        $response = $client->enqueueAndParse(InvoiceSplitterV1::class, $invoiceSplitterInput);
        $inference = $response->document->inference;
        $pdfExtractor = new PdfExtractor($invoiceSplitterInput);
        self::assertSame(2, $pdfExtractor->pageCount);

        $extractedPdfsStrict = $pdfExtractor->extractInvoices($inference->prediction->invoicePageGroups);

        self::assertCount(2, $extractedPdfsStrict);
        self::assertSame('default_sample_001-001.pdf', $extractedPdfsStrict[0]->getFilename());
        self::assertSame('default_sample_002-002.pdf', $extractedPdfsStrict[1]->getFilename());

        $invoice0 = $client->parse(InvoiceV4::class, $extractedPdfsStrict[0]->asInputSource());

        $testStringRstInvoice0 = $this->prepareInvoiceReturn(
            rstFilePath: TestingUtilities::getV1DataDir() . '/products/invoices/response_v4/summary_full_invoice_p1.rst',
            invoicePrediction: $invoice0->document
        );

        self::assertGreaterThan(
            0.90,
            TestingUtilities::levenshteinRatio(
                $testStringRstInvoice0,
                (string) $invoice0->document
            )
        );
    }
}