Wikibase\JsonDumpReader\Tests\Integration\EntityDumpIteratorTest

Complexity

Total Complexity

13

Size/Duplication

Total Lines	118
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes

0

11 Methods

Rating	Name	Duplication	Size	Complexity
A	newIteratorForFile()	0	7	1
A	assertFindsEntities()	0	9	2
A	testGivenFileWithNoEntities_noEntitiesAreReturned()	0	5	1
A	testGivenFileWithOneEntity_oneEntityIsFound()	0	5	1
A	testGivenFileWithFiveEntities_fiveEntityAreFound()	0	5	1
A	testGivenFileWithInvalidEntity_noEntityIsFound()	0	4	1
A	testGivenFileWithInvalidEntities_validEntitiesAreFound()	0	4	1
A	testCanDoMultipleIterations()	0	6	1
A	testInitialPosition()	0	26	1
A	testGivenFileWithInvalidEntities_errorsAreReported()	0	18	2
A	testGivenFileWithInvalidJsonLine_errorIsRecorded()	0	15	1

<?php

declare( strict_types = 1 );

namespace Wikibase\JsonDumpReader\Tests\Integration;

use Iterator;
use PHPUnit\Framework\TestCase;
use Wikibase\JsonDumpReader\Iterator\EntityDumpIterator;
use Wikibase\JsonDumpReader\JsonDumpFactory;
use Wikibase\JsonDumpReader\Reader\ExtractedDumpReader;
use Wikibase\JsonDumpReader\Tests\TestFactory;

/**
 * @covers \Wikibase\JsonDumpReader\Iterator\EntityDumpIterator
 * @covers \Wikibase\JsonDumpReader\JsonDumpFactory
 *
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class EntityDumpIteratorTest extends TestCase {

	private function newIteratorForFile( $filePath, callable $onError = null ) {
		return ( new JsonDumpFactory() )->newEntityDumpIterator(
			new ExtractedDumpReader( $filePath ),
			TestFactory::newInstance()->newEntityDeserializer(),
			$onError
		);
	}

	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 ),
			TestFactory::newInstance()->newEntityDeserializer()
		);

		$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 ),
			TestFactory::newInstance()->newEntityDeserializer()
		);

		$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 );

$errors is of type array, but the function expects a object<Countable>|object<PHPUnit\Framework\iterable>.

	}

	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 );

$errors is of type array, but the function expects a object<Countable>|object<PHPUnit\Framework\iterable>.

	}

}