Current File : /home/jvzmxxx/wiki1/extensions/Flow/tests/phpunit/Parsoid/ReferenceExtractorTest.php

<?php

namespace Flow\Tests\Parsoid;

use Flow\Container;
use Flow\Exception\WikitextException;
use Flow\Model\UUID;
use Flow\Parsoid\ReferenceFactory;
use Flow\Conversion\Utils;
use Flow\Tests\FlowTestCase;
use ReflectionMethod;
use Title;

/**
 * @group Database
 * @group Flow
 */
class ReferenceExtractorTestCase extends FlowTestCase {
	protected function setUp() {
		parent::setUp();

		// Check for Parsoid
		try {
			Utils::convert( 'html', 'wikitext', 'Foo', Title::newFromText( 'UTPage' ) );
		} catch ( WikitextException $excep ) {
			$this->markTestSkipped( 'Parsoid not enabled' );
		}
	}

	public static function referenceExtractorProvider() {
		return array(
			array(
				'Normal link',
				// source wiki text
				'[[My page]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'link',
				// expected target
				'title:My_page',
			),
			array(
				'Link with URL encoding issues',
				// source wiki text
				'[[User talk:Werdna?]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'link',
				// expected target
				'title:User_talk:Werdna?',
			),
			array(
				'Subpage link',
				// source wiki text
				'[[/Subpage]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'link',
				// expected target
				'title:Talk:UTPage/Subpage',
				// ???
				'Talk:UTPage',
			),
			array(
				'External link',
				// source wiki text
				'[http://www.google.com Google]',
				// expected factory method
				'Flow\Model\UrlReference',
				// expected type
				'link',
				// expected target
				'url:http://www.google.com',
			),
			array(
				'File',
				// source wiki text
				'[[File:Image.png]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'file',
				// expected target
				'title:File:Image.png',
			),
			array(
				'File with parameters',
				// source wiki text
				'[[File:Image.png|25px]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'file',
				// expected target
				'title:File:Image.png',
			),
			array(
				'File with encoding issues',
				// source wiki text
				'[[File:Image?.png]]',
				// expected class
				'Flow\Model\WikiReference',
				// expected type
				'file',
				// expected target
				'title:File:Image?.png',
			),
			array(
				'Template',
				// source wiki text
				'{{Foo}}',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'template',
				// expected target
				'title:Template:Foo',
			),

			array(
				'Non-existent File',
				// source wiki text
				'[[File:Some/Files/Really/Should_Not_Ex/ist.png]]',
				// expected factory method
				'Flow\Model\WikiReference',
				// expected type
				'file',
				// expected target
				'title:File:Some/Files/Really/Should_Not_Ex/ist.png',
			)
		);
	}

	/**
	 * @dataProvider referenceExtractorProvider
	 */
	public function testReferenceExtractor(
		$description,
		$wikitext,
		$expectedClass,
		$expectedType,
		$expectedTarget,
		$page = 'UTPage'
	) {
		$referenceExtractor = Container::get( 'reference.extractor' );

		$workflow = $this->getMock( 'Flow\Model\Workflow' );
		$workflow->expects( $this->any() )
			->method( 'getId' )
			->will( $this->returnValue( UUID::create() ) );
		$workflow->expects( $this->any() )
			->method( 'getArticleTitle' )
			->will( $this->returnValue( Title::newMainPage() ) );
		$factory = new ReferenceFactory( $workflow, 'foo', UUID::create() );

		$reflMethod = new ReflectionMethod( $referenceExtractor, 'extractReferences' );
		$reflMethod->setAccessible( true );

		$reflProperty = new \ReflectionProperty( $referenceExtractor, 'extractors' );
		$reflProperty->setAccessible( true );
		$extractors = $reflProperty->getValue( $referenceExtractor );

		$html = Utils::convert( 'wt', 'html', $wikitext, Title::newFromText( $page ) );
		$result = $reflMethod->invoke(
			$referenceExtractor,
			$factory,
			$extractors['post'],
			$html
		);
		$this->assertCount( 1, $result, $html );

		$result = reset( $result );
		$this->assertInstanceOf( $expectedClass, $result, $description );
		$this->assertEquals( $expectedType, $result->getType(), $description );
		$this->assertEquals( $expectedTarget, $result->getTargetIdentifier(), $description );
	}
}