OcrControllerTest.php

<?php
declare( strict_types = 1 );

namespace App\Tests\Controller;

use App\Controller\OcrController;
use App\Engine\EngineFactory;
use App\Engine\GoogleCloudVisionEngine;
use App\Engine\TesseractEngine;
use App\Engine\TranskribusClient;
use App\Engine\TranskribusEngine;
use App\Tests\OcrTestCase;
use Krinkle\Intuition\Intuition;
use Symfony\Component\Cache\Adapter\FilesystemAdapter;
use Symfony\Component\Cache\Adapter\NullAdapter;
use Symfony\Component\HttpClient\MockHttpClient;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\RequestStack;
use Symfony\Component\HttpFoundation\Session\Session;
use Symfony\Component\HttpFoundation\Session\Storage\MockArraySessionStorage;
use thiagoalessio\TesseractOCR\TesseractOCR;

class OcrControllerTest extends OcrTestCase {

	/**
	 * @dataProvider provideGetLang
	 * @covers OcrController::getLangs
	 * @param string[] $getParams
	 * @param string[] $expectedLangs
	 */
	public function testGetLang( array $getParams, array $expectedLangs ): void {
		$request = new Request( $getParams );
		$requestStack = new RequestStack();
		$requestStack->push( $request );
		$request->setSession( new Session( new MockArraySessionStorage() ) );
		$intuition = new Intuition( [] );
		$gcv = new GoogleCloudVisionEngine(
			dirname( __DIR__ ) . '/fixtures/google-account-keyfile.json',
			$intuition,
			$this->projectDir,
			new MockHttpClient()
		);
		$controller = new OcrController(
			$requestStack,
			$intuition,
			new EngineFactory(
				$gcv,
				new TesseractEngine( new MockHttpClient(), $intuition, $this->projectDir, new TesseractOCR() ),
				new TranskribusEngine(
					new TranskribusClient(
						getenv( 'APP_TRANSKRIBUS_USERNAME' ),
						getenv( 'APP_TRANSKRIBUS_PASSWORD' ),
						new MockHttpClient(),
						new NullAdapter(),
						new NullAdapter()
					),
					$intuition,
					$this->projectDir,
					new MockHttpClient()
				),
			),
			new FilesystemAdapter()
		);
		$this->assertSame( $expectedLangs, $controller->getLangs( $request ) );
	}

	/**
	 * @return mixed[]
	 */
	public function provideGetLang(): array {
		return [
			[
				[ 'lang' => 'ar' ],
				[ 'ar' ],
			],
			[
				[ 'langs' => [ 'a|b', 'c!', 'ab' ] ],
				[ 'ab', 'c' ],
			],
			'special characters' => [
				[ 'langs' => [ 'sr-Latn', 'Canadian_Aboriginal' ] ],
				[ 'sr-Latn', 'Canadian_Aboriginal' ],
			],
			'numbers' => [
				[ 'langs' => [ 'ru-petr1708' ] ],
				[ 'ru-petr1708' ],
			],
		];
	}

	/**
	 * @dataProvider provideRotationNormalization
	 * @covers OcrController
	 * @param int $inputRotation
	 * @param int $expectedRotation
	 */
	public function testRotationNormalization( int $inputRotation, int $expectedRotation ): void {
		$request = new Request( [ 'rotate' => $inputRotation ] );
		$requestStack = new RequestStack();
		$requestStack->push( $request );
		$request->setSession( new Session( new MockArraySessionStorage() ) );
		$intuition = new Intuition( [] );
		$controller = new OcrController(
			$requestStack,
			$intuition,
			new EngineFactory(
				new GoogleCloudVisionEngine(
					dirname( __DIR__ ) . '/fixtures/google-account-keyfile.json',
					$intuition,
					$this->projectDir,
					new MockHttpClient()
				),
				new TesseractEngine( new MockHttpClient(), $intuition, $this->projectDir, new TesseractOCR() ),
				new TranskribusEngine(
					new TranskribusClient(
						getenv( 'APP_TRANSKRIBUS_USERNAME' ),
						getenv( 'APP_TRANSKRIBUS_PASSWORD' ),
						new MockHttpClient(),
						new NullAdapter(),
						new NullAdapter()
					),
					$intuition,
					$this->projectDir,
					new MockHttpClient()
				),
			),
			new FilesystemAdapter()
		);
		// Trigger the setup to process rotation
		$reflection = new \ReflectionMethod( $controller, 'setup' );
		$reflection->setAccessible( true );
		$reflection->invoke( $controller );

		$this->assertSame( $expectedRotation, OcrController::$params['rotate'] );
	}

	/**
	 * @return int[][]
	 */
	public function provideRotationNormalization(): array {
		return [
			'zero rotation' => [ 0, 0 ],
			'positive rotation' => [ 90, 90 ],
			'negative rotation' => [ -90, 270 ],
			'large positive rotation' => [ 450, 90 ],
			'large negative rotation' => [ -450, 270 ],
			'boundary 359' => [ 359, 359 ],
			'boundary 360' => [ 360, 0 ],
			'boundary -360' => [ -360, 0 ],
		];
	}
}