tigitz / php-spellchecker

tests/PhpSpellcheck/Tests/Spellchecker/MultipleSpellCheckersTest.php

<?php

declare(strict_types=1);

namespace PhpSpellcheck\Tests\Spellchecker;

use PhpSpellcheck\Misspelling;
use PhpSpellcheck\Spellchecker\MultiSpellchecker;
use PhpSpellcheck\Spellchecker\SpellcheckerInterface;
use PhpSpellcheck\Utils\TextEncoding;
use PHPUnit\Framework\TestCase;

class MultipleSpellcheckersTest extends TestCase
{

    public function testCheckAndMergeSuggestions()
    {
        $spellChecker1 = $this->createMock(SpellcheckerInterface::class);
        $misspelling1 = new Misspelling('mispelling1', 1);
        $misspelling2a = new Misspelling('mispelling2', 2, 2, ['suggestionA']);
        $misspelling2b = new Misspelling('mispelling2', 2, 2, ['suggestionB']);

        $spellChecker1->method('check')
            ->willReturn([$misspelling1, $misspelling2a]);
        $spellChecker2 = $this->createMock(SpellcheckerInterface::class);
        $misspelling3 = new Misspelling('mispelling3', 3, 3);
        $spellChecker2->method('check')
            ->willReturn([$misspelling2b, $misspelling3]);

        $multiSpellchecker = new MultiSpellchecker([$spellChecker1, $spellChecker2]);

        $misspellings = $multiSpellchecker->check('test', ['en'], ['ctx'], TextEncoding::UTF8);
        $this->assertEquals(
            [
                $misspelling1->setContext(['ctx']),
                new Misspelling('mispelling2', 2, 2, ['suggestionA', 'suggestionB'], ['ctx']),
                $misspelling3->setContext(['ctx']),
            ],
            $misspellings
        );
    }

    public function testCheckAndNotMergeSuggestions()
    {
        $spellChecker1 = $this->createMock(SpellcheckerInterface::class);
        $misspelling1 = new Misspelling('mispelling1', 1);
        $misspelling2a = new Misspelling('mispelling2', 2, 2, ['suggestionA']);
        $misspelling2b = new Misspelling('mispelling2', 2, 2, ['suggestionB']);

        $spellChecker1->method('check')
            ->willReturn([$misspelling1, $misspelling2a]);
        $spellChecker2 = $this->createMock(SpellcheckerInterface::class);
        $misspelling3 = new Misspelling('mispelling3', 3, 3);
        $spellChecker2->method('check')
            ->willReturn([$misspelling2b, $misspelling3]);

        $multiSpellchecker = new MultiSpellchecker([$spellChecker1, $spellChecker2], false);

        $misspellings = $multiSpellchecker->check('test', ['en'], ['ctx'], TextEncoding::UTF8);
        $this->assertEquals(
            [
                $misspelling1->setContext(['ctx']),
                $misspelling2a->setContext(['ctx']),
                $misspelling2b->setContext(['ctx']),
                $misspelling3->setContext(['ctx']),
            ],
            iterator_to_array($misspellings)

$misspellings of type array is incompatible with the type Traversable expected by parameter $iterator of iterator_to_array().

        );
    }

    public function testGetSupportedLanguages()
    {
        $spellChecker1 = $this->createMock(SpellcheckerInterface::class);
        $spellChecker1->method('getSupportedLanguages')
            ->willReturn(['en', 'fr']);
        $spellChecker2 = $this->createMock(SpellcheckerInterface::class);
        $spellChecker2->method('getSupportedLanguages')
            ->willReturn(['fr', 'ru']);

        $multipleSpellchecker = new MultiSpellchecker([$spellChecker1, $spellChecker2]);

        $this->assertSame(['en', 'fr', 'ru'], $multipleSpellchecker->getSupportedLanguages());
    }
}