Tests\Wikibase\JsonDumpReader\EntityDumpIteratorTest

Complexity

Total Complexity

15

Size/Duplication

Total Lines	144
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Importance

Changes

0

<?php
class EntityDumpIteratorTest extends TestCase {

	private function newIteratorForFile( $filePath, callable $onError = null ) {
		return ( new JsonDumpFactory() )->newEntityDumpIterator(
			new ExtractedDumpReader( $filePath ),
			$this->newCurrentEntityDeserializer(),
			$onError
		);
	}

	private function newCurrentEntityDeserializer() {
		$factory = new DeserializerFactory(
			$this->newDataValueDeserializer(),
			new BasicEntityIdParser()
		);

		return $factory->newEntityDeserializer();
	}

	private function newDataValueDeserializer() {
		$dataValueClasses = [
			'boolean' => BooleanValue::class,
			'number' => NumberValue::class,SeekableDumpReader
			'string' => StringValue::class,

This code did not parse for me. Apparently, there is an error somewhere around this line:

Syntax error, unexpected T_CONSTANT_ENCAPSED_STRING, expecting ']'

			'unknown' => UnknownValue::class,
			'globecoordinate' => GlobeCoordinateValue::class,
			'monolingualtext' => MonolingualTextValue::class,
			'multilingualtext' => MultilingualTextValue::class,
			'quantity' => QuantityValue::class,
			'time' => TimeValue::class,
			'wikibase-entityid' => EntityIdValue::class,
		];

		return new DataValueDeserializer( $dataValueClasses );
	}

	private function assertFindsEntities( array $expectedIds, Iterator $dumpIterator, $message = '' ) {
		$actualIds = [];

		foreach ( $dumpIterator as $entity ) {
			$actualIds[] = $entity->getId()->getSerialization();
		}

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

	public function testGivenFileWithNoEntities_noEntitiesAreReturned() {
		$iterator = $this->newIteratorForFile( ( new \JsonDumpData() )->getEmptyDumpPath() );

		$this->assertFindsEntities( [], $iterator );
	}

	public function testGivenFileWithOneEntity_oneEntityIsFound() {
		$iterator = $this->newIteratorForFile( ( new \JsonDumpData() )->getOneItemDumpPath() );

		$this->assertFindsEntities( [ 'Q1' ], $iterator );
	}

	public function testGivenFileWithFiveEntities_fiveEntityAreFound() {
		$iterator = $this->newIteratorForFile( ( new \JsonDumpData() )->getFiveEntitiesDumpPath() );

		$this->assertFindsEntities( [ 'Q1', 'Q8', 'P16', 'P19', 'P22' ], $iterator );
	}

	public function testGivenFileWithInvalidEntity_noEntityIsFound() {
		$iterator = $this->newIteratorForFile( __DIR__ . '/../data/invalid-item.json' );
		$this->assertFindsEntities( [], $iterator );
	}

	public function testGivenFileWithInvalidEntities_validEntitiesAreFound() {
		$iterator = $this->newIteratorForFile( __DIR__ . '/../data/3valid-2invalid.json' );
		$this->assertFindsEntities( [ 'Q1', 'P16', 'P22' ], $iterator );
	}

	public function testCanDoMultipleIterations() {
		$iterator = $this->newIteratorForFile( ( new \JsonDumpData() )->getFiveEntitiesDumpPath() );

		$this->assertFindsEntities( [ 'Q1', 'Q8', 'P16', 'P19', 'P22' ], $iterator, 'first iteration' );
		$this->assertFindsEntities( [ 'Q1', 'Q8', 'P16', 'P19', 'P22' ], $iterator, 'second iteration' );
	}

	public function testInitialPosition() {
		$reader = new ExtractedDumpReader( ( new \JsonDumpData() )->getFiveEntitiesDumpPath() );

		$iterator = new EntityDumpIterator(
			( new JsonDumpFactory() )->newObjectDumpIterator( $reader ),
			$this->newCurrentEntityDeserializer()
		);

		$iterator->rewind();
		$this->assertSame( 'Q1', $iterator->current()->getId()->getSerialization() );

		$iterator->next();
		$this->assertSame( 'Q8', $iterator->current()->getId()->getSerialization() );

		$newReader = new ExtractedDumpReader(
			( new \JsonDumpData() )->getFiveEntitiesDumpPath(),
			$reader->getPosition()
		);

		$newIterator = new EntityDumpIterator(
			( new JsonDumpFactory() )->newObjectDumpIterator( $newReader ),
			$this->newCurrentEntityDeserializer()
		);

		$this->assertFindsEntities( [ 'P16', 'P19', 'P22' ], $newIterator );
	}

	public function testGivenFileWithInvalidEntities_errorsAreReported() {
		$errors = [];

		$iterator = $this->newIteratorForFile(
			__DIR__ . '/../data/3valid-2invalid.json',
			function( $errorMessage ) use ( &$errors ) {
				$errors[] = $errorMessage;
			}
		);

		$iterator->rewind();
		while ( $iterator->valid() ) {
			$iterator->next();
		}

		$this->assertContainsOnly( 'string', $errors );
		$this->assertCount( 2, $errors );
	}

	public function testGivenFileWithInvalidJsonLine_errorIsRecorded() {
		$errors = [];

		$iterator = $this->newIteratorForFile(
			__DIR__ . '/../data/invalid-json.json',
			function( $errorMessage ) use ( &$errors ) {
				$errors[] = $errorMessage;
			}
		);

		iterator_to_array( $iterator );

		$this->assertContainsOnly( 'string', $errors );
		$this->assertCount( 1, $errors );
	}

}