1 files changed, 118 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/collation/CollationTest.php b/www/wiki/tests/phpunit/includes/collation/CollationTest.php
new file mode 100644
index 00000000..b92e651e
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/collation/CollationTest.php
@@ -0,0 +1,118 @@
+<?php
+
+/**
+ * Class CollationTest
+ * @covers Collation
+ * @covers IcuCollation
+ * @covers IdentityCollation
+ * @covers UppercaseCollation
+ */
+class CollationTest extends MediaWikiLangTestCase {
+	protected function setUp() {
+		parent::setUp();
+		$this->checkPHPExtension( 'intl' );
+	}
+
+	/**
+	 * Test to make sure, that if you
+	 * have "X" and "XY", the binary
+	 * sortkey also has "X" being a
+	 * prefix of "XY". Our collation
+	 * code makes this assumption.
+	 *
+	 * @param string $lang Language code for collator
+	 * @param string $base
+	 * @param string $extended String containing base as a prefix.
+	 *
+	 * @dataProvider prefixDataProvider
+	 */
+	public function testIsPrefix( $lang, $base, $extended ) {
+		$cp = Collator::create( $lang );
+		$cp->setStrength( Collator::PRIMARY );
+		$baseBin = $cp->getSortKey( $base );
+		// Remove sortkey terminator
+		$baseBin = rtrim( $baseBin, "\0" );
+		$extendedBin = $cp->getSortKey( $extended );
+		$this->assertStringStartsWith( $baseBin, $extendedBin, "$base is not a prefix of $extended" );
+	}
+
+	public static function prefixDataProvider() {
+		return [
+			[ 'en', 'A', 'AA' ],
+			[ 'en', 'A', 'AAA' ],
+			[ 'en', 'Д', 'ДЂ' ],
+			[ 'en', 'Д', 'ДA' ],
+			// 'Ʒ' should expand to 'Z ' (note space).
+			[ 'fi', 'Z', 'Ʒ' ],
+			// 'Þ' should expand to 'th'
+			[ 'sv', 't', 'Þ' ],
+			// Javanese is a limited use alphabet, so should have 3 bytes
+			// per character, so do some tests with it.
+			[ 'en', 'ꦲ', 'ꦲꦤ' ],
+			[ 'en', 'ꦲ', 'ꦲД' ],
+			[ 'en', 'A', 'Aꦲ' ],
+		];
+	}
+
+	/**
+	 * Opposite of testIsPrefix
+	 *
+	 * @dataProvider notPrefixDataProvider
+	 */
+	public function testNotIsPrefix( $lang, $base, $extended ) {
+		$cp = Collator::create( $lang );
+		$cp->setStrength( Collator::PRIMARY );
+		$baseBin = $cp->getSortKey( $base );
+		// Remove sortkey terminator
+		$baseBin = rtrim( $baseBin, "\0" );
+		$extendedBin = $cp->getSortKey( $extended );
+		$this->assertStringStartsNotWith( $baseBin, $extendedBin, "$base is a prefix of $extended" );
+	}
+
+	public static function notPrefixDataProvider() {
+		return [
+			[ 'en', 'A', 'B' ],
+			[ 'en', 'AC', 'ABC' ],
+			[ 'en', 'Z', 'Ʒ' ],
+			[ 'en', 'A', 'ꦲ' ],
+		];
+	}
+
+	/**
+	 * Test correct first letter is fetched.
+	 *
+	 * @param string $collation Collation name (aka uca-en)
+	 * @param string $string String to get first letter of
+	 * @param string $firstLetter Expected first letter.
+	 *
+	 * @dataProvider firstLetterProvider
+	 */
+	public function testGetFirstLetter( $collation, $string, $firstLetter ) {
+		$col = Collation::factory( $collation );
+		$this->assertEquals( $firstLetter, $col->getFirstLetter( $string ) );
+	}
+
+	function firstLetterProvider() {
+		return [
+			[ 'uppercase', 'Abc', 'A' ],
+			[ 'uppercase', 'abc', 'A' ],
+			[ 'identity', 'abc', 'a' ],
+			[ 'uca-en', 'abc', 'A' ],
+			[ 'uca-en', ' ', ' ' ],
+			[ 'uca-en', 'Êveryone', 'E' ],
+			[ 'uca-vi', 'Êveryone', 'Ê' ],
+			// Make sure thorn is not a first letter.
+			[ 'uca-sv', 'The', 'T' ],
+			[ 'uca-sv', 'Å', 'Å' ],
+			[ 'uca-hu', 'dzsdo', 'Dzs' ],
+			[ 'uca-hu', 'dzdso', 'Dz' ],
+			[ 'uca-hu', 'CSD', 'Cs' ],
+			[ 'uca-root', 'CSD', 'C' ],
+			[ 'uca-fi', 'Ǥ', 'G' ],
+			[ 'uca-fi', 'Ŧ', 'T' ],
+			[ 'uca-fi', 'Ʒ', 'Z' ],
+			[ 'uca-fi', 'Ŋ', 'N' ],
+			[ 'uppercase-ba', 'в', 'В' ],
+		];
+	}
+}