1 files changed, 67 insertions, 0 deletions
diff --git a/MLEB/Translate/tests/phpunit/TranslatorInterface/Aid/TranslationAidDataProviderTest.php b/MLEB/Translate/tests/phpunit/TranslatorInterface/Aid/TranslationAidDataProviderTest.php
new file mode 100644
index 00000000..51e9fe27
--- /dev/null
+++ b/MLEB/Translate/tests/phpunit/TranslatorInterface/Aid/TranslationAidDataProviderTest.php
@@ -0,0 +1,67 @@
+<?php
+
+namespace MediaWiki\Extension\Translate\TranslatorInterface\Aid;
+
+use HashMessageIndex;
+use MediaWikiIntegrationTestCase;
+use MessageGroups;
+use MessageHandle;
+use MessageIndex;
+use MockWikiMessageGroup;
+use ObjectCache;
+use Title;
+use WANObjectCache;
+
+/**
+ * @group Database
+ * @covers \MediaWiki\Extension\Translate\TranslatorInterface\Aid\TranslationAidDataProvider
+ */
+class TranslationAidDataProviderTest extends MediaWikiIntegrationTestCase {
+
+	protected function setUp(): void {
+		parent::setUp();
+		$this->setMwGlobals( [
+			'wgTranslateMessageNamespaces' => [ NS_MEDIAWIKI ]
+		] );
+		$this->setTemporaryHook( 'TranslatePostInitGroups', [ $this, 'getTestGroups' ] );
+
+		$mg = MessageGroups::singleton();
+		$mg->setCache( new WANObjectCache( [ 'cache' => ObjectCache::getInstance( 'hash' ) ] ) );
+		$mg->recache();
+		MessageIndex::setInstance( new HashMessageIndex() );
+		MessageIndex::singleton()->rebuild();
+	}
+
+	public function getTestGroups( &$list ) {
+		$messages = [
+			'TestPage' => 'bunny',
+		];
+		$list['test-group'] = new MockWikiMessageGroup( 'test-group', $messages );
+		return false;
+	}
+
+	/**
+	 * @covers \MediaWiki\Extension\Translate\TranslatorInterface\Aid\TranslationAidDataProvider::getGoodTranslations
+	 * @throws MWException
+	 */
+	public function testGetGoodTranslations() {
+		$title = 'MediaWiki:TestPage';
+		// Create some translations
+		$this->assertTrue(
+			$this->editPage( $title . '/fi', 'Test Finnish Translation' )->isGood(),
+			'Sanity: must successfully edit ' . $title . '/fi page'
+		);
+		$this->assertTrue(
+			$this->editPage( $title . '/ru', 'Test Russian Translation' )->isGood(),
+			'Sanity: must successfully edit ' . $title . '/ru page'
+		);
+
+		$messageHandle = new MessageHandle( Title::newFromText( $title ) );
+		$this->assertTrue( $messageHandle->isValid(), 'Sanity: MessageHandle must be valid' );
+		$dataProvider = new TranslationAidDataProvider( $messageHandle );
+		$this->assertEquals( [
+			'ru' => 'Test Russian Translation',
+			'fi' => 'Test Finnish Translation'
+		], $dataProvider->getGoodTranslations() );
+	}
+}