Current File : /home/jvzmxxx/wiki/extensions/Wikibase/lib/tests/phpunit/Store/RevisionBasedEntityLookupTest.php

<?php

namespace Wikibase\Test;

use Wikibase\DataModel\Entity\EntityId;
use Wikibase\DataModel\Entity\Item;
use Wikibase\DataModel\Entity\ItemId;
use Wikibase\DataModel\Services\Lookup\EntityLookupException;
use Wikibase\Lib\Store\EntityRevisionLookup;
use Wikibase\Lib\Store\RevisionBasedEntityLookup;
use Wikibase\Lib\Store\RevisionedUnresolvedRedirectException;
use Wikibase\Lib\Tests\MockRepository;

/**
 * @covers Wikibase\Lib\Store\RevisionBasedEntityLookup
 *
 * @group WikibaseLib
 * @group WikibaseEntityLookup
 * @group Wikibase
 *
 * @license GPL-2.0+
 * @author Daniel Kinzler
 */
class RevisionBasedEntityLookupTest extends \PHPUnit_Framework_TestCase {

	public function getEntityProvider() {
		$q10 = new ItemId( 'Q10' );
		$q11 = new ItemId( 'Q11' );

		$item10 = new Item( $q10 );
		$item10->setLabel( 'en', 'ten' );

		$repo = new MockRepository();
		$repo->putEntity( $item10 );

		return array(
			'found' => array( $repo, $q10, $q10 ),
			'not found' => array( $repo, $q11, null ),
		);
	}

	/**
	 * @dataProvider getEntityProvider
	 *
	 * @param EntityRevisionLookup $revisionLookup
	 * @param EntityId $id
	 * @param EntityId|null $expected
	 */
	public function testGetEntity( EntityRevisionLookup $revisionLookup, EntityId $id, EntityId $expected = null ) {
		$entityLookup = new RevisionBasedEntityLookup( $revisionLookup );
		$entity = $entityLookup->getEntity( $id );

		if ( $expected === null ) {
			$this->assertNull( $entity );
		} else {
			$this->assertTrue( $expected->equals( $entity->getId() ) );
		}
	}

	public function hasEntityProvider() {
		$cases = $this->getEntityProvider();

		$cases = array_map( function( $case ) {
			// true if set an id is expected, false otherwise.
			$case[2] = $case[2] !== null;

			return $case;
		}, $cases );

		return $cases;
	}

	/**
	 * @dataProvider hasEntityProvider
	 *
	 * @param EntityRevisionLookup $revisionLookup
	 * @param EntityId $id
	 * @param bool $exists
	 */
	public function testHasEntity( EntityRevisionLookup $revisionLookup, EntityId $id, $exists ) {
		$entityLookup = new RevisionBasedEntityLookup( $revisionLookup );
		$actual = $entityLookup->hasEntity( $id );

		$this->assertEquals( $exists, $actual );
	}

	public function testWhenEntityLookupExceptionIsThrown_getEntityPassesItAlong() {
		$entityLookup = new RevisionBasedEntityLookup( $this->newEntityLookupExceptionThrowingRevisionLookup() );

		$this->setExpectedException( RevisionedUnresolvedRedirectException::class );
		$entityLookup->getEntity( new ItemId( 'Q1' ) );
	}

	/**
	 * @return EntityRevisionLookup
	 */
	private function newEntityLookupExceptionThrowingRevisionLookup() {
		$revisionLookup = $this->getMock( EntityRevisionLookup::class );

		$revisionLookup->expects( $this->any() )
			->method( 'getEntityRevision' )
			->will( $this->throwException( new RevisionedUnresolvedRedirectException(
				new ItemId( 'Q1' ),
				new ItemId( 'Q2' )
			) ) );

		$revisionLookup->expects( $this->any() )
			->method( 'getLatestRevisionId' )
			->will( $this->throwException( new RevisionedUnresolvedRedirectException(
				new ItemId( 'Q1' ),
				new ItemId( 'Q2' )
			) ) );

		return $revisionLookup;
	}

	public function testWhenEntityLookupExceptionIsThrown_hasEntityPassesItAlong() {
		$entityLookup = new RevisionBasedEntityLookup( $this->newEntityLookupExceptionThrowingRevisionLookup() );

		$this->setExpectedException( RevisionedUnresolvedRedirectException::class );
		$entityLookup->hasEntity( new ItemId( 'Q1' ) );
	}

	public function testWhenBadExceptionIsThrown_hasEntityRethrowsAsEntityLookupException() {
		$entityLookup = new RevisionBasedEntityLookup( $this->newBadExceptionThrowingRevisionLookup() );

		$this->setExpectedException( EntityLookupException::class );
		$entityLookup->hasEntity( new ItemId( 'Q1' ) );
	}

	/**
	 * @return EntityRevisionLookup
	 */
	private function newBadExceptionThrowingRevisionLookup() {
		$revisionLookup = $this->getMock( EntityRevisionLookup::class );

		$revisionLookup->expects( $this->any() )
			->method( 'getEntityRevision' )
			->will( $this->throwException( new \Exception( 'Someone killed a kitten' ) ) );

		$revisionLookup->expects( $this->any() )
			->method( 'getLatestRevisionId' )
			->will( $this->throwException( new \Exception( 'Someone killed a kitten' ) ) );

		return $revisionLookup;
	}

	public function testWhenBadExceptionIsThrown_getEntityRethrowsAsEntityLookupException() {
		$entityLookup = new RevisionBasedEntityLookup( $this->newBadExceptionThrowingRevisionLookup() );

		$this->setExpectedException( EntityLookupException::class );
		$entityLookup->getEntity( new ItemId( 'Q1' ) );
	}

}