Page MenuHomeWickedGov Phorge
Files F4126062

EntitySearchTest.php
No OneTemporary
Actions

Size
5 KB
Referenced Files
None
Subscribers
None

EntitySearchTest.php
View Options

<?php
declare( strict_types = 1 );
namespace MediaWiki\Extension\Translate\TranslatorInterface;
use Generator;
use HashBagOStuff;
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\Extension\Translate\MessageLoading\MessageIndex;
use MediaWiki\Extension\Translate\Services;
use MediaWikiIntegrationTestCase;
use MessageGroup;
use WANObjectCache;
/**
* @covers \MediaWiki\Extension\Translate\TranslatorInterface\EntitySearch
* @license GPL-2.0-or-later
*/
class EntitySearchTest extends MediaWikiIntegrationTestCase {
protected function setUp(): void {
parent::setUp();
$this->overrideConfigValue( 'TranslateMessageIndex', 'hash' );
}
public function getMessageGroupFactoryStub(): MessageGroups {
$data =
<<<'EOF'
Page
Translatable page
Pägë
Translatable pägë
Page 1
Page 10
Page 2
Page Page Page Page
page page page page
foo Page
EOF;
$data = explode( "\n", $data );
$stubGroups = [];
foreach ( $data as $dataItem ) {
$stubGroup = $this->createStub( MessageGroup::class );
$stubGroup->method( 'getLabel' )->willReturn( $dataItem );
$stubGroup->method( 'getId' )->willReturn( $this->makeGroupId( $dataItem ) );
$stubGroups[] = $stubGroup;
}
$stub = $this->createStub( MessageGroups::class );
$stub->method( 'getGroups' )->willReturn( $stubGroups );
return $stub;
}
private function makeGroupId( string $x ): string {
return "page-$x";
}
/** @dataProvider provideTestSearchStaticMessageGroups */
public function testSearchStaticMessageGroups( string $query, int $maxResults, array $expected ) {
$mediaWikiServices = $this->getServiceContainer();
$entitySearch = new EntitySearch(
new WANObjectCache( [ 'cache' => new HashBagOStuff() ] ),
$mediaWikiServices->getCollationFactory()->makeCollation( 'uca-default-u-kn' ),
$this->getMessageGroupFactoryStub(),
$mediaWikiServices->getNamespaceInfo(),
Services::getInstance()->getMessageIndex(),
$mediaWikiServices->getTitleFormatter(),
$mediaWikiServices->getTitleParser()
);
$actual = $entitySearch->searchStaticMessageGroups( $query, $maxResults );
$this->assertEquals( $expected, $actual );
}
public static function provideTestSearchStaticMessageGroups(): Generator {
yield [
'Page',
10,
[
[
'label' => 'Page',
'group' => 'page-Page',
],
[
'label' => 'Page 1',
'group' => 'page-Page 1',
],
[
'label' => 'Page 2',
'group' => 'page-Page 2',
],
[
'label' => 'Page 10',
'group' => 'page-Page 10',
],
[
'label' => 'page page page page',
'group' => 'page-page page page page',
],
[
'label' => 'Page Page Page Page',
'group' => 'page-Page Page Page Page',
],
[
'label' => 'foo Page',
'group' => 'page-foo Page'
],
[
'label' => 'Translatable page',
'group' => 'page-Translatable page',
],
]
];
yield [
'P',
1,
[
[
'label' => 'Page',
'group' => 'page-Page',
],
]
];
yield [
'Pägë',
10,
[
[
'label' => 'Pägë',
'group' => 'page-Pägë',
],
[
'label' => 'Translatable pägë',
'group' => 'page-Translatable pägë',
],
]
];
yield [
'Book',
10,
[]
];
}
public function getMessageIndexStub(): MessageIndex {
$data =
<<<'EOF'
8:title
8:page title
8:CAPITAL TITLE
8:big_bunny
8:prefix
9:prefix-1
9:prefix-2
9:prefix-3
EOF;
$data = explode( "\n", $data );
$stub = $this->createStub( MessageIndex::class );
$stub->method( 'getKeys' )->willReturn( $data );
return $stub;
}
/** @dataProvider provideTestSearchMessages */
public function testSearchMessages( string $query, int $maxResults, array $expected ) {
$mediaWikiServices = $this->getServiceContainer();
$entitySearch = new EntitySearch(
new WANObjectCache( [ 'cache' => new HashBagOStuff() ] ),
$mediaWikiServices->getCollationFactory()->makeCollation( 'uca-default-u-kn' ),
$this->createStub( MessageGroups::class ),
$mediaWikiServices->getNamespaceInfo(),
$this->getMessageIndexStub(),
$mediaWikiServices->getTitleFormatter(),
$mediaWikiServices->getTitleParser()
);
$actual = $entitySearch->searchMessages( $query, $maxResults );
$this->assertEquals( $expected, $actual );
}
public static function provideTestSearchMessages(): Generator {
yield 'prefix and infix case-insensitive matching at word boundaries' => [
'title',
10,
[
[
'pattern' => 'MediaWiki:CAPITAL TITLE',
'count' => 1,
],
[
'pattern' => 'MediaWiki:Page title',
'count' => 1,
],
[
'pattern' => 'MediaWiki:Title',
'count' => 1,
],
]
];
yield 'matching at underscore boundaries' => [
'bunny',
1,
[
[
'pattern' => 'MediaWiki:Big bunny',
'count' => 1,
],
]
];
yield 'prefix collapsing' => [
'prefix',
2,
[
[
'pattern' => 'MediaWiki talk:Prefix*',
'count' => 3,
],
[
'pattern' => 'MediaWiki:Prefix',
'count' => 1,
],
]
];
yield [
'No match',
10,
[]
];
}
}

File Metadata

Mime Type
text/x-php
Expires
Wed, Aug 19, 07:53 (2 w, 2 d ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
e0/0f/e9ed763922a0b89d2e06480c2c5a
Default Alt Text
EntitySearchTest.php (5 KB)

Event Timeline