1 files changed, 420 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Maintenance/DataRebuilderTest.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Maintenance/DataRebuilderTest.php
new file mode 100644
index 00000000..a3b3a441
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Maintenance/DataRebuilderTest.php
@@ -0,0 +1,420 @@
+<?php
+
+namespace SMW\Tests\Maintenance;
+
+use SMW\Maintenance\DataRebuilder;
+use SMW\Options;
+use Title;
+use SMW\Tests\PHPUnitCompat;
+use SMW\Tests\TestEnvironment;
+
+/**
+ * @covers \SMW\Maintenance\DataRebuilder
+ * @group semantic-mediawiki
+ * @group medium
+ *
+ * @license GNU GPL v2+
+ * @since 1.9.2
+ *
+ * @author mwjames
+ */
+class DataRebuilderTest extends \PHPUnit_Framework_TestCase {
+
+	use PHPUnitCompat;
+
+	protected $obLevel;
+	private $connectionManager;
+	private $testEnvironment;
+
+	// The Store writes to the output buffer during drop/setupStore, to avoid
+	// inappropriate buffer settings which can cause interference during unit
+	// testing, we clean the output buffer
+	protected function setUp() {
+
+		$nullJob = $this->getMockBuilder( '\SMW\MediaWiki\Jobs\NullJob' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$jobFactory = $this->getMockBuilder( '\SMW\MediaWiki\JobFactory' )
+			->disableOriginalConstructor()
+			->setMethods( [ 'newUpdateJob' ] )
+			->getMock();
+
+		$jobFactory->expects( $this->any() )
+			->method( 'newUpdateJob' )
+			->will( $this->returnValue( $nullJob ) );
+
+		$this->testEnvironment = new TestEnvironment();
+		$this->testEnvironment->registerObject( 'JobFactory', $jobFactory );
+
+		$connection = $this->getMockBuilder( '\SMW\MediaWiki\Database' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$connection->expects( $this->any() )
+			->method( 'select' )
+			->will( $this->returnValue( [] ) );
+
+		$this->connectionManager = $this->getMockBuilder( '\SMW\Connection\ConnectionManager' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$this->connectionManager->expects( $this->any() )
+			->method( 'getConnection' )
+			->will( $this->returnValue( $connection ) );
+
+		$this->obLevel = ob_get_level();
+		ob_start();
+
+		parent::setUp();
+	}
+
+	protected function tearDown() {
+		parent::tearDown();
+		$this->testEnvironment->tearDown();
+
+		while ( ob_get_level() > $this->obLevel ) {
+			ob_end_clean();
+		}
+	}
+
+	public function testCanConstruct() {
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->getMockForAbstractClass();
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$this->assertInstanceOf(
+			'\SMW\Maintenance\DataRebuilder',
+			new DataRebuilder( $store, $titleFactory )
+		);
+	}
+
+	/**
+	 * @depends testCanConstruct
+	 */
+	public function testRebuildAllWithoutOptions() {
+
+		$entityRebuildDispatcher = $this->getMockBuilder( '\SMW\SQLStore\EntityRebuildDispatcher' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$entityRebuildDispatcher->expects( $this->once() )
+			->method( 'rebuild' )
+			->will( $this->returnCallback( [ $this, 'refreshDataOnMockCallback' ] ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getMaxId' )
+			->will( $this->returnValue( 1000 ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getDispatchedEntities' )
+			->will( $this->returnValue( [] ) );
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->setMethods( [ 'refreshData' ] )
+			->getMockForAbstractClass();
+
+		$store->expects( $this->once() )
+			->method( 'refreshData' )
+			->will( $this->returnValue( $entityRebuildDispatcher ) );
+
+		$store->setConnectionManager( $this->connectionManager );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		// Needs an end otherwise phpunit is caught up in an infinite loop
+		$instance->setOptions( new Options( [
+			'e' => 1
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	/**
+	 * @depends testCanConstruct
+	 */
+	public function testRebuildAllWithFullDelete() {
+
+		$entityRebuildDispatcher = $this->getMockBuilder( '\SMW\SQLStore\EntityRebuildDispatcher' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$entityRebuildDispatcher->expects( $this->atLeastOnce() )
+			->method( 'rebuild' )
+			->will( $this->returnCallback( [ $this, 'refreshDataOnMockCallback' ] ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getMaxId' )
+			->will( $this->returnValue( 1000 ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getDispatchedEntities' )
+			->will( $this->returnValue( [] ) );
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->setMethods( [
+				'refreshData',
+				'drop' ] )
+			->getMockForAbstractClass();
+
+		$store->expects( $this->once() )
+			->method( 'refreshData' )
+			->will( $this->returnValue( $entityRebuildDispatcher ) );
+
+		$store->expects( $this->once() )
+			->method( 'drop' );
+
+		$store->setConnectionManager( $this->connectionManager );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			'e' => 1,
+			'f' => true,
+			'verbose' => false
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	/**
+	 * @depends testCanConstruct
+	 */
+	public function testRebuildAllWithStopRangeOption() {
+
+		$entityRebuildDispatcher = $this->getMockBuilder( '\SMW\SQLStore\EntityRebuildDispatcher' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$entityRebuildDispatcher->expects( $this->exactly( 6 ) )
+			->method( 'rebuild' )
+			->will( $this->returnCallback( [ $this, 'refreshDataOnMockCallback' ] ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getMaxId' )
+			->will( $this->returnValue( 1000 ) );
+
+		$entityRebuildDispatcher->expects( $this->any() )
+			->method( 'getDispatchedEntities' )
+			->will( $this->returnValue( [] ) );
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->setMethods( [ 'refreshData' ] )
+			->getMockForAbstractClass();
+
+		$store->expects( $this->once() )
+			->method( 'refreshData' )
+			->will( $this->returnValue( $entityRebuildDispatcher ) );
+
+		$store->setConnectionManager( $this->connectionManager );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			's' => 2,
+			'n' => 5,
+			'verbose' => false
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	/**
+	 * @depends testCanConstruct
+	 */
+	public function testRebuildSelectedPagesWithQueryOption() {
+
+		$subject = $this->getMockBuilder( '\SMW\DIWikiPage' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$subject->expects( $this->once() )
+			->method( 'getTitle' )
+			->will( $this->returnValue( Title::newFromText( __METHOD__ ) ) );
+
+		$queryResult = $this->getMockBuilder( '\SMWQueryResult' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$queryResult->expects( $this->once() )
+			->method( 'getResults' )
+			->will( $this->returnValue( [ $subject ] ) );
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->getMockForAbstractClass();
+
+		$store->expects( $this->at( 0 ) )
+			->method( 'getQueryResult' )
+			->will( $this->returnValue( 1 ) );
+
+		$store->expects( $this->at( 1 ) )
+			->method( 'getQueryResult' )
+			->will( $this->returnValue( $queryResult ) );
+
+		$store->setConnectionManager( $this->connectionManager );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			'query' => '[[Category:Foo]]'
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	public function testRebuildSelectedPagesWithCategoryNamespaceFilter() {
+
+		$row = new \stdClass;
+		$row->cat_title = 'Foo';
+
+		$database = $this->getMockBuilder( '\SMW\MediaWiki\Database' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$database->expects( $this->any() )
+			->method( 'select' )
+			->with( $this->stringContains( 'category' ),
+				$this->anything(),
+				$this->anything(),
+				$this->anything(),
+				$this->anything() )
+			->will( $this->returnValue( [ $row ] ) );
+
+		$store = $this->getMockBuilder( '\SMWSQLStore3' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$store->expects( $this->once() )
+			->method( 'getConnection' )
+			->will( $this->returnValue( $database ) );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			'categories' => true
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	public function testRebuildSelectedPagesWithPropertyNamespaceFilter() {
+
+		$row = new \stdClass;
+		$row->page_namespace = SMW_NS_PROPERTY;
+		$row->page_title = 'Bar';
+
+		$database = $this->getMockBuilder( '\SMW\MediaWiki\Database' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$database->expects( $this->any() )
+			->method( 'select' )
+			->with( $this->anything(),
+				$this->anything(),
+				$this->equalTo( [ 'page_namespace' => SMW_NS_PROPERTY ] ),
+				$this->anything(),
+				$this->anything() )
+			->will( $this->returnValue( [ $row ] ) );
+
+		$store = $this->getMockBuilder( '\SMWSQLStore3' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$store->expects( $this->once() )
+			->method( 'getConnection' )
+			->will( $this->returnValue( $database ) );
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			'p' => true
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+	}
+
+	public function testRebuildSelectedPagesWithPageOption() {
+
+		$store = $this->getMockBuilder( '\SMW\Store' )
+			->disableOriginalConstructor()
+			->getMockForAbstractClass();
+
+		$titleFactory = $this->getMockBuilder( '\SMW\MediaWiki\TitleFactory' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$titleFactory->expects( $this->at( 0 ) )
+			->method( 'newFromText' )
+			->with( $this->equalTo( 'Main page' ) )
+			->will( $this->returnValue( Title::newFromText( 'Main page' ) ) );
+
+		$titleFactory->expects( $this->at( 1 ) )
+			->method( 'newFromText' )
+			->with( $this->equalTo( 'Some other page' ) )
+			->will( $this->returnValue( Title::newFromText( 'Some other page' ) ) );
+
+		$titleFactory->expects( $this->at( 2 ) )
+			->method( 'newFromText' )
+			->with( $this->equalTo( 'Help:Main page' ) )
+			->will( $this->returnValue( Title::newFromText( 'Main page', NS_HELP ) ) );
+
+		$titleFactory->expects( $this->at( 3 ) )
+			->method( 'newFromText' )
+			->with( $this->equalTo( 'Main page' ) )
+			->will( $this->returnValue( Title::newFromText( 'Main page' ) ) );
+
+		$instance = new DataRebuilder( $store, $titleFactory );
+
+		$instance->setOptions( new Options( [
+			'page'  => 'Main page|Some other page|Help:Main page|Main page'
+		] ) );
+
+		$this->assertTrue( $instance->rebuild() );
+
+		$this->assertEquals(
+			3,
+			$instance->getRebuildCount()
+		);
+	}
+
+	/**
+	 * @see Store::refreshData
+	 */
+	public function refreshDataOnMockCallback( &$index ) {
+		$index++;
+	}
+
+}