1 files changed, 68 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/collation/CustomUppercaseCollationTest.php b/www/wiki/tests/phpunit/includes/collation/CustomUppercaseCollationTest.php
new file mode 100644
index 00000000..f9e0bc9b
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/collation/CustomUppercaseCollationTest.php
@@ -0,0 +1,68 @@
+<?php
+
+/**
+ * @covers CustomUppercaseCollation
+ */
+class CustomUppercaseCollationTest extends MediaWikiTestCase {
+
+	public function setUp() {
+		$this->collation = new CustomUppercaseCollation( [
+			'D',
+			'C',
+			'Cs',
+			'B'
+		], Language::factory( 'en' ) );
+
+		parent::setUp();
+	}
+
+	/**
+	 * @dataProvider providerOrder
+	 */
+	public function testOrder( $first, $second, $msg ) {
+		$sortkey1 = $this->collation->getSortKey( $first );
+		$sortkey2 = $this->collation->getSortKey( $second );
+
+		$this->assertTrue( strcmp( $sortkey1, $sortkey2 ) < 0, $msg );
+	}
+
+	public function providerOrder() {
+		return [
+			[ 'X', 'Z', 'Maintain order of unrearranged' ],
+			[ 'D', 'C', 'Actually resorts' ],
+			[ 'D', 'B', 'resort test 2' ],
+			[ 'Adobe', 'Abode', 'not first letter' ],
+			[ '💩 ', 'C', 'Test relocated to end' ],
+			[ 'c', 'b', 'lowercase' ],
+			[ 'x', 'z', 'lowercase original' ],
+			[ 'Cz', 'Cs', 'digraphs' ],
+			[ 'C50D', 'C100', 'Numbers' ]
+		];
+	}
+
+	/**
+	 * @dataProvider provideGetFirstLetter
+	 */
+	public function testGetFirstLetter( $string, $first ) {
+		$this->assertSame( $this->collation->getFirstLetter( $string ), $first );
+	}
+
+	public function provideGetFirstLetter() {
+		return [
+			[ 'Do', 'D' ],
+			[ 'do', 'D' ],
+			[ 'Ao', 'A' ],
+			[ 'afdsa', 'A' ],
+			[ "\xF3\xB3\x80\x80Foo", 'D' ],
+			[ "\xF3\xB3\x80\x81Foo", 'C' ],
+			[ "\xF3\xB3\x80\x82Foo", 'Cs' ],
+			[ "\xF3\xB3\x80\x83Foo", 'B' ],
+			[ "\xF3\xB3\x80\x84Foo", "\xF3\xB3\x80\x84" ],
+			[ 'C', 'C' ],
+			[ 'Cz', 'C' ],
+			[ 'Cs', 'Cs' ],
+			[ 'CS', 'Cs' ],
+			[ 'cs', 'Cs' ],
+		];
+	}
+}