1 files changed, 121 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/TitleArrayFromResultTest.php b/www/wiki/tests/phpunit/includes/TitleArrayFromResultTest.php
new file mode 100644
index 00000000..af49ecf7
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/TitleArrayFromResultTest.php
@@ -0,0 +1,121 @@
+<?php
+
+/**
+ * @author Addshore
+ * @covers TitleArrayFromResult
+ */
+class TitleArrayFromResultTest extends PHPUnit\Framework\TestCase {
+
+	use MediaWikiCoversValidator;
+
+	private function getMockResultWrapper( $row = null, $numRows = 1 ) {
+		$resultWrapper = $this->getMockBuilder( Wikimedia\Rdbms\ResultWrapper::class )
+			->disableOriginalConstructor();
+
+		$resultWrapper = $resultWrapper->getMock();
+		$resultWrapper->expects( $this->atLeastOnce() )
+			->method( 'current' )
+			->will( $this->returnValue( $row ) );
+		$resultWrapper->expects( $this->any() )
+			->method( 'numRows' )
+			->will( $this->returnValue( $numRows ) );
+
+		return $resultWrapper;
+	}
+
+	private function getRowWithTitle( $namespace = 3, $title = 'foo' ) {
+		$row = new stdClass();
+		$row->page_namespace = $namespace;
+		$row->page_title = $title;
+		return $row;
+	}
+
+	private function getTitleArrayFromResult( $resultWrapper ) {
+		return new TitleArrayFromResult( $resultWrapper );
+	}
+
+	/**
+	 * @covers TitleArrayFromResult::__construct
+	 */
+	public function testConstructionWithFalseRow() {
+		$row = false;
+		$resultWrapper = $this->getMockResultWrapper( $row );
+
+		$object = $this->getTitleArrayFromResult( $resultWrapper );
+
+		$this->assertEquals( $resultWrapper, $object->res );
+		$this->assertSame( 0, $object->key );
+		$this->assertEquals( $row, $object->current );
+	}
+
+	/**
+	 * @covers TitleArrayFromResult::__construct
+	 */
+	public function testConstructionWithRow() {
+		$namespace = 0;
+		$title = 'foo';
+		$row = $this->getRowWithTitle( $namespace, $title );
+		$resultWrapper = $this->getMockResultWrapper( $row );
+
+		$object = $this->getTitleArrayFromResult( $resultWrapper );
+
+		$this->assertEquals( $resultWrapper, $object->res );
+		$this->assertSame( 0, $object->key );
+		$this->assertInstanceOf( Title::class, $object->current );
+		$this->assertEquals( $namespace, $object->current->mNamespace );
+		$this->assertEquals( $title, $object->current->mTextform );
+	}
+
+	public static function provideNumberOfRows() {
+		return [
+			[ 0 ],
+			[ 1 ],
+			[ 122 ],
+		];
+	}
+
+	/**
+	 * @dataProvider provideNumberOfRows
+	 * @covers TitleArrayFromResult::count
+	 */
+	public function testCountWithVaryingValues( $numRows ) {
+		$object = $this->getTitleArrayFromResult( $this->getMockResultWrapper(
+			$this->getRowWithTitle(),
+			$numRows
+		) );
+		$this->assertEquals( $numRows, $object->count() );
+	}
+
+	/**
+	 * @covers TitleArrayFromResult::current
+	 */
+	public function testCurrentAfterConstruction() {
+		$namespace = 0;
+		$title = 'foo';
+		$row = $this->getRowWithTitle( $namespace, $title );
+		$object = $this->getTitleArrayFromResult( $this->getMockResultWrapper( $row ) );
+		$this->assertInstanceOf( Title::class, $object->current() );
+		$this->assertEquals( $namespace, $object->current->mNamespace );
+		$this->assertEquals( $title, $object->current->mTextform );
+	}
+
+	public function provideTestValid() {
+		return [
+			[ $this->getRowWithTitle(), true ],
+			[ false, false ],
+		];
+	}
+
+	/**
+	 * @dataProvider provideTestValid
+	 * @covers TitleArrayFromResult::valid
+	 */
+	public function testValid( $input, $expected ) {
+		$object = $this->getTitleArrayFromResult( $this->getMockResultWrapper( $input ) );
+		$this->assertEquals( $expected, $object->valid() );
+	}
+
+	// @todo unit test for key()
+	// @todo unit test for next()
+	// @todo unit test for rewind()
+}