146 lines
3.4 KiB
PHP
146 lines
3.4 KiB
PHP
<?php
|
|
|
|
namespace SMW\Tests\ParserFunctions;
|
|
|
|
use SMW\ParserFunctions\ExpensiveFuncExecutionWatcher;
|
|
|
|
/**
|
|
* @covers \SMW\ParserFunctions\ExpensiveFuncExecutionWatcher
|
|
* @group semantic-mediawiki
|
|
*
|
|
* @license GPL-2.0-or-later
|
|
* @since 3.0
|
|
*
|
|
* @author mwjames
|
|
*/
|
|
class ExpensiveFuncExecutionWatcherTest extends \PHPUnit\Framework\TestCase {
|
|
|
|
private $parserData;
|
|
|
|
protected function setUp(): void {
|
|
parent::setUp();
|
|
|
|
$this->parserData = $this->getMockBuilder( '\SMW\ParserData' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
}
|
|
|
|
public function testCanConstruct() {
|
|
$this->assertInstanceOf(
|
|
'\SMW\ParserFunctions\ExpensiveFuncExecutionWatcher',
|
|
new ExpensiveFuncExecutionWatcher( $this->parserData )
|
|
);
|
|
}
|
|
|
|
public function testHasReachedExpensiveLimit() {
|
|
$parserOutput = $this->getMockBuilder( '\MediaWiki\Parser\ParserOutput' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$parserOutput->expects( $this->once() )
|
|
->method( 'getExtensionData' )
|
|
->willReturn( 42 );
|
|
|
|
$query = $this->getMockBuilder( '\SMWQuery' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$query->expects( $this->once() )
|
|
->method( 'getLimit' )
|
|
->willReturn( 100 );
|
|
|
|
$this->parserData->expects( $this->once() )
|
|
->method( 'getOutput' )
|
|
->willReturn( $parserOutput );
|
|
|
|
$instance = new ExpensiveFuncExecutionWatcher(
|
|
$this->parserData
|
|
);
|
|
|
|
$instance->setExpensiveExecutionLimit( 1 );
|
|
|
|
$this->assertTrue(
|
|
$instance->hasReachedExpensiveLimit( $query )
|
|
);
|
|
}
|
|
|
|
public function testIncrementExpensiveCountOnExsitingCounter() {
|
|
$parserOutput = $this->getMockBuilder( '\MediaWiki\Parser\ParserOutput' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$parserOutput->expects( $this->once() )
|
|
->method( 'getExtensionData' )
|
|
->willReturn( 42 );
|
|
|
|
$parserOutput->expects( $this->once() )
|
|
->method( 'setExtensionData' )
|
|
->with(
|
|
ExpensiveFuncExecutionWatcher::EXPENSIVE_COUNTER,
|
|
43 );
|
|
|
|
$query = $this->getMockBuilder( '\SMWQuery' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$query->expects( $this->once() )
|
|
->method( 'getLimit' )
|
|
->willReturn( 100 );
|
|
|
|
$query->expects( $this->once() )
|
|
->method( 'getOption' )
|
|
->willReturn( 100 );
|
|
|
|
$this->parserData->expects( $this->any() )
|
|
->method( 'getOutput' )
|
|
->willReturn( $parserOutput );
|
|
|
|
$instance = new ExpensiveFuncExecutionWatcher(
|
|
$this->parserData
|
|
);
|
|
|
|
$instance->setExpensiveThreshold( 1 );
|
|
$instance->setExpensiveExecutionLimit( 1 );
|
|
|
|
$instance->incrementExpensiveCount( $query );
|
|
}
|
|
|
|
public function testIncrementExpensiveCountOnNull() {
|
|
$parserOutput = $this->getMockBuilder( '\MediaWiki\Parser\ParserOutput' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$parserOutput->expects( $this->once() )
|
|
->method( 'setExtensionData' )
|
|
->with(
|
|
ExpensiveFuncExecutionWatcher::EXPENSIVE_COUNTER,
|
|
1 );
|
|
|
|
$query = $this->getMockBuilder( '\SMWQuery' )
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$query->expects( $this->once() )
|
|
->method( 'getLimit' )
|
|
->willReturn( 100 );
|
|
|
|
$query->expects( $this->once() )
|
|
->method( 'getOption' )
|
|
->willReturn( 100 );
|
|
|
|
$this->parserData->expects( $this->any() )
|
|
->method( 'getOutput' )
|
|
->willReturn( $parserOutput );
|
|
|
|
$instance = new ExpensiveFuncExecutionWatcher(
|
|
$this->parserData
|
|
);
|
|
|
|
$instance->setExpensiveThreshold( 1 );
|
|
$instance->setExpensiveExecutionLimit( 1 );
|
|
|
|
$instance->incrementExpensiveCount( $query );
|
|
}
|
|
|
|
}
|