Wikibase\EntityStore\Tests\Integration\BatchingEntityFetcherTest

Complexity

Total Complexity

9

Size/Duplication

Total Lines	91
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	setUp()	0	5	1
A	insertEntities()	0	9	2
A	getFetcherThatContinuesFrom()	0	13	1
A	testWhenFetchingLessEntitiesThenExist_fullBatchOfEntitiesIsReturned()	0	6	1
A	testStartingPositionIsHeldIntoAccountAndIsIncremented()	0	19	1
A	assertAreItemsWithIds()	0	11	2
A	testRewindSetsThePositionBackToTheInitialValue()	0	14	1

<?php

namespace Wikibase\EntityStore\Tests\Integration;

use Wikibase\DataModel\Entity\EntityId;
use Wikibase\DataModel\Entity\Item;
use Wikibase\DataModel\Entity\ItemId;
use Wikibase\EntityStore\BatchingEntityFetcher;
use Wikibase\EntityStore\BatchingEntityIdFetcherBuilder;
use Wikibase\Repo\WikibaseRepo;

/**
 * @covers Wikibase\EntityStore\BatchingEntityFetcher
 *
 * @group Database
 *
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class BatchingEntityFetcherTest extends \MediaWikiTestCase {

	public function setUp() {
		parent::setUp();

		$this->insertEntities();
	}

	private function insertEntities() {
		$entityStore = WikibaseRepo::getDefaultInstance()->getStore()->getEntityStore();

		foreach ( range( 1000070, 1000075 ) as $itemId ) {
			$item = Item::newEmpty();

The method Wikibase\DataModel\Entity\Item::newEmpty() has been deprecated with message: since 2.5, use new Item() instead.

			$item->setId( $itemId );
			$entityStore->saveEntity( $item, __CLASS__, $GLOBALS['wgUser'] );
		}
	}

	private function getFetcherThatContinuesFrom( EntityId $previousId = null ) {
		$repo = WikibaseRepo::getDefaultInstance();

		$fetcherBuilder = new BatchingEntityIdFetcherBuilder(
			$repo->getStore()->newEntityPerPage(),
			$previousId
		);

		return new BatchingEntityFetcher(
			$fetcherBuilder->getFetcher(),
			$repo->getEntityLookup()
		);
	}

	public function testWhenFetchingLessEntitiesThenExist_fullBatchOfEntitiesIsReturned() {
		$entities = $this->getFetcherThatContinuesFrom( null )->fetchNext( 5 );

		$this->assertCount( 5, $entities );
		$this->assertContainsOnlyInstancesOf( 'Wikibase\DataModel\Entity\Item', $entities );
	}

	public function testStartingPositionIsHeldIntoAccountAndIsIncremented() {
		$fetcher = $this->getFetcherThatContinuesFrom( new ItemId( 'Q1000070' ) );

		$this->assertAreItemsWithIds(
			array(
				new ItemId( 'Q1000071' ),
				new ItemId( 'Q1000072' ),
			),
			$fetcher->fetchNext( 2 )
		);

		$this->assertAreItemsWithIds(
			array(
				new ItemId( 'Q1000073' ),
				new ItemId( 'Q1000074' ),
			),
			$fetcher->fetchNext( 2 )
		);
	}

	/**
	 * @param ItemId[] $expectedIds
	 * @param Item[] $items
	 */
	private function assertAreItemsWithIds( array $expectedIds, array $items ) {
		$this->assertContainsOnlyInstancesOf( 'Wikibase\DataModel\Entity\Item', $items );

		$actualIds = array();

		foreach ( $items as $item ) {
			$actualIds[] = $item->getId();
		}

		$this->assertEquals( $expectedIds, $actualIds );
	}

	public function testRewindSetsThePositionBackToTheInitialValue() {
		$fetcher = $this->getFetcherThatContinuesFrom( new ItemId( 'Q1000070' ) );

		$fetcher->fetchNext( 5 );
		$fetcher->rewind();

		$this->assertAreItemsWithIds(
			array(
				new ItemId( 'Q1000071' ),
				new ItemId( 'Q1000072' ),
			),
			$fetcher->fetchNext( 2 )
		);
	}

}