mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Echo
synced 2024-12-02 19:36:48 +00:00
760cfad682
To match the return type documentation of IReadableDatabase::select Also fix a default value for IReadableDatabase::selectRowCount Change-Id: I510bd261b64b5c2cef227c081a2fe34801f54121
98 lines
2.4 KiB
PHP
98 lines
2.4 KiB
PHP
<?php
|
|
|
|
use MediaWiki\Extension\Notifications\DbFactory;
|
|
use MediaWiki\Extension\Notifications\Mapper\TargetPageMapper;
|
|
use MediaWiki\Extension\Notifications\Model\TargetPage;
|
|
use Wikimedia\Rdbms\FakeResultWrapper;
|
|
use Wikimedia\Rdbms\IDatabase;
|
|
use Wikimedia\Rdbms\InsertQueryBuilder;
|
|
|
|
/**
|
|
* @covers \MediaWiki\Extension\Notifications\Mapper\TargetPageMapper
|
|
*/
|
|
class TargetPageMapperTest extends MediaWikiUnitTestCase {
|
|
|
|
public static function provideDataTestInsert() {
|
|
return [
|
|
[
|
|
'successful insert with next sequence = 1',
|
|
[ 'insert' => true, 'insertId' => 2 ],
|
|
1
|
|
],
|
|
[
|
|
'successful insert with insert id = 2',
|
|
[ 'insert' => true, 'insertId' => 2 ],
|
|
2
|
|
],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @dataProvider provideDataTestInsert
|
|
*/
|
|
public function testInsert( $message, $dbResult, $result ) {
|
|
$target = $this->mockTargetPage();
|
|
$targetMapper = new TargetPageMapper( $this->mockDbFactory( $dbResult ) );
|
|
$this->assertEquals( $result, $targetMapper->insert( $target ), $message );
|
|
}
|
|
|
|
/**
|
|
* Mock object of TargetPage
|
|
* @return TargetPage
|
|
*/
|
|
protected function mockTargetPage() {
|
|
$target = $this->createMock( TargetPage::class );
|
|
$target->method( 'toDbArray' )
|
|
->willReturn( [] );
|
|
$target->method( 'getPageId' )
|
|
->willReturn( 2 );
|
|
$target->method( 'getEventId' )
|
|
->willReturn( 3 );
|
|
|
|
return $target;
|
|
}
|
|
|
|
/**
|
|
* Mock object of DbFactory
|
|
* @param array $dbResult
|
|
* @return DbFactory
|
|
*/
|
|
protected function mockDbFactory( $dbResult ) {
|
|
$dbFactory = $this->createMock( DbFactory::class );
|
|
$dbFactory->method( 'getEchoDb' )
|
|
->willReturn( $this->mockDb( $dbResult ) );
|
|
|
|
return $dbFactory;
|
|
}
|
|
|
|
/**
|
|
* Returns a mock database object
|
|
* @param array $dbResult
|
|
* @return IDatabase
|
|
*/
|
|
protected function mockDb( array $dbResult ) {
|
|
$dbResult += [
|
|
'insert' => '',
|
|
'insertId' => '',
|
|
'select' => [],
|
|
'delete' => ''
|
|
];
|
|
$db = $this->createMock( IDatabase::class );
|
|
$db->method( 'insert' )
|
|
->willReturn( $dbResult['insert'] );
|
|
$db->method( 'insertId' )
|
|
->willReturn( $dbResult['insertId'] );
|
|
$db->method( 'select' )
|
|
->willReturn( new FakeResultWrapper( $dbResult['select'] ) );
|
|
$db->method( 'delete' )
|
|
->willReturn( $dbResult['delete'] );
|
|
$db->method( 'newInsertQueryBuilder' )
|
|
->willReturnCallback( static function () use ( $db ) {
|
|
return new InsertQueryBuilder( $db );
|
|
} );
|
|
|
|
return $db;
|
|
}
|
|
|
|
}
|