Wikibase\EntityStore\MongoDBTest

Complexity

Total Complexity

5

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	15

Importance

Changes	16
Bugs	0	Features	6

4 Methods

Rating	Name	Duplication	Size	Complexity
A	getEntityStoreFromConfiguration()	0	4	1
A	setupMongoDb()	0	4	1
A	testMongoDbStore()	0	60	2
A	importCommand()	0	10	1

<?php

namespace Wikibase\EntityStore;

use Ask\Language\Description\AnyValue;
use Ask\Language\Description\SomeProperty;
use Ask\Language\Description\ValueDescription;
use Ask\Language\Option\QueryOptions;
use DataValues\StringValue;
use MongoConnectionException;
use Symfony\Component\Console\Input\ArrayInput;
use Symfony\Component\Console\Output\NullOutput;
use Wikibase\DataModel\Entity\EntityIdValue;
use Wikibase\DataModel\Entity\ItemId;
use Wikibase\DataModel\Entity\PropertyId;
use Wikibase\DataModel\Term\Term;
use Wikibase\EntityStore\Config\EntityStoreFromConfigurationBuilder;
use Wikibase\EntityStore\Console\CliApplicationFactory;

/**
 * @licence GNU GPL v2+
 * @author Thomas Pellissier Tanon
 */
class MongoDBTest extends \PHPUnit_Framework_TestCase {

	public function testMongoDbStore() {
		try {
			$this->setupMongoDb();
		} catch( MongoConnectionException $e ) {
			$this->markTestSkipped( 'MongoDB is not running: ' . $e->getMessage() );
			return;
		}

		$store = $this->getEntityStoreFromConfiguration();

		$this->assertEquals(
			new ItemId( 'Q1' ),
			$store->getItemLookup()->getItemForId( new ItemId( 'Q1' ) )->getId()
		);

		$this->assertEquals(
			new PropertyId( 'P1' ),
			$store->getPropertyLookup()->getPropertyForId( new PropertyId( 'P1' ) )->getId()
		);

		$results = $store->getEntityDocumentLookup()->getEntityDocumentsForIds(
			[ new ItemId( 'Q1' ), new ItemId( 'Q1000' ) ]
		);
		$this->assertEquals( 1, count( $results ) );
		$this->assertEquals( new ItemId( 'Q1' ), $results[0]->getId() );

		$this->assertEquals(
			[ new ItemId( 'Q1' ) ],
			$store->getItemIdForTermLookup()->getItemIdsForTerm( new Term( 'en', 'universe' ) )
		);

		$this->assertEquals(
			[],
			$store->getItemIdForTermLookup()->getItemIdsForTerm( new Term( 'pl', 'Kosmos' ) )
		);

		$this->assertEquals(
			[ new PropertyId( 'P16' ) ],
			$store->getPropertyIdForTermLookup()->getPropertyIdsForTerm( new Term( 'en', 'highway system' ) )
		);

		$this->assertEquals(
			[ new PropertyId( 'P16' ) ],
			$store->getPropertyIdForQueryLookup()->getPropertyIdsForQuery(
				new AnyValue(),
				new QueryOptions( 1, 0 )
			)
		);

		$this->assertEquals(
			[ new ItemId( 'Q1' ) ],
			$store->getItemIdForQueryLookup()->getItemIdsForQuery(
				new SomeProperty(
					new EntityIdValue( new PropertyId( 'P18' ) ),
					new ValueDescription( new StringValue( 'Hubble ultra deep field.jpg' ) )
				),
				new QueryOptions( 10, 0 )
			)
		);
	}

	private function setupMongoDb() {
		$this->importCommand( 'import-json-dump', 'valid.json' );
		$this->importCommand( 'import-incremental-xml-dump', 'valid-incremental.xml' );
	}

	private function importCommand( $command, $file ) {
		$applicationFactory = new CliApplicationFactory();
		$importCommand = $applicationFactory->newApplication()->find( $command );
		$input = new ArrayInput( [
			'command' => $command,
			'file' => __DIR__ . '/../data/' . $file,
			'configuration' => __DIR__ . '/../data/valid-config-mongodb.json'
		] );
		$importCommand->run( $input, new NullOutput() );
	}

	private function getEntityStoreFromConfiguration() {
		$configBuilder = new EntityStoreFromConfigurationBuilder();
		return $configBuilder->buildEntityStore( __DIR__ . '/../data/valid-config-mongodb.json' );
	}
}