mediawiki-extensions-Echo/tests/phpunit/ThankYouEditTest.php
Aaron Schulz 88ab531e17 Explicitly call DeferredUpdates::tryOpportunisticExecute() in ThankYouEditTest
Change-Id: I423f9a856fddab30e691d78cbcd0b70a292e3916
2024-11-07 12:16:46 -08:00

85 lines
2.7 KiB
PHP

<?php
namespace MediaWiki\Extension\Notifications\Test;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\Extension\Notifications\DbFactory;
use MediaWiki\Extension\Notifications\Mapper\NotificationMapper;
use MediaWiki\Title\Title;
use MediaWikiIntegrationTestCase;
use Wikimedia\Rdbms\Platform\ISQLPlatform;
/**
* @group Echo
* @group Database
*/
class ThankYouEditTest extends MediaWikiIntegrationTestCase {
private function deleteEchoData() {
$db = DbFactory::newFromDefault()->getEchoDb( DB_PRIMARY );
$db->newDeleteQueryBuilder()
->deleteFrom( 'echo_event' )
->where( ISQLPlatform::ALL_ROWS )
->caller( __METHOD__ )
->execute();
$db->newDeleteQueryBuilder()
->deleteFrom( 'echo_notification' )
->where( ISQLPlatform::ALL_ROWS )
->caller( __METHOD__ )
->execute();
}
/**
* @covers \MediaWiki\Extension\Notifications\Hooks::onPageSaveComplete
*/
public function testFirstEdit() {
// setup
$this->deleteEchoData();
$user = $this->getMutableTestUser()->getUser();
$title = Title::newFromText( 'Help:MWEchoThankYouEditTest_testFirstEdit' );
// action
$this->editPage( $title, 'this is my first edit', '', NS_MAIN, $user );
DeferredUpdates::tryOpportunisticExecute();
// assertions
$notificationMapper = new NotificationMapper();
$notifications = $notificationMapper->fetchByUser( $user, 10, null, [ 'thank-you-edit' ] );
$this->assertCount( 1, $notifications );
/** @var Notification $notification */
$notification = reset( $notifications );
$this->assertSame( 1, $notification->getEvent()->getExtraParam( 'editCount', 'not found' ) );
}
/**
* @covers \MediaWiki\Extension\Notifications\Hooks::onPageSaveComplete
*/
public function testTenthEdit() {
// setup
$this->deleteEchoData();
$user = $this->getMutableTestUser()->getUser();
$title = Title::newFromText( 'Help:MWEchoThankYouEditTest_testTenthEdit' );
$page = $this->getServiceContainer()->getWikiPageFactory()->newFromTitle( $title );
// action
// we could fast-forward the edit-count to speed things up
// but this is the only way to make sure duplicate notifications
// are not generated
for ( $i = 0; $i < 12; $i++ ) {
$this->editPage( $page, "this is edit #$i", '', NS_MAIN, $user );
}
DeferredUpdates::tryOpportunisticExecute();
$user->clearInstanceCache();
// assertions
$notificationMapper = new NotificationMapper();
$notifications = $notificationMapper->fetchByUser( $user, 10, null, [ 'thank-you-edit' ] );
$this->assertCount( 2, $notifications );
/** @var Notification $notification */
$notification = reset( $notifications );
$this->assertSame( 10, $notification->getEvent()->getExtraParam( 'editCount', 'not found' ) );
}
}