Page MenuHomeWickedGov Phorge
Files F4145671

UserEditTrackerTest.php
No OneTemporary
Actions

Size
6 KB
Referenced Files
None
Subscribers
None

UserEditTrackerTest.php
View Options

<?php
use MediaWiki\CommentStore\CommentStoreComment;
use MediaWiki\Content\WikitextContent;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\Deferred\UserEditCountUpdate;
use MediaWiki\Revision\MutableRevisionRecord;
use MediaWiki\Revision\SlotRecord;
use MediaWiki\Tests\User\TempUser\TempUserTestTrait;
use MediaWiki\Title\Title;
use MediaWiki\User\UserIdentity;
use MediaWiki\User\UserIdentityValue;
use MediaWiki\User\UserRigorOptions;
/**
* @covers \MediaWiki\User\UserEditTracker
* @group Database
*/
class UserEditTrackerTest extends MediaWikiIntegrationTestCase {
use TempUserTestTrait;
/**
* Do an edit
*
* @param UserIdentity $user
* @param string $timestamp
*/
private function editTrackerDoEdit( $user, $timestamp ) {
$title = Title::newFromText( __FUNCTION__ );
$page = $this->getServiceContainer()->getWikiPageFactory()->newFromTitle( $title );
if ( !$page->exists() ) {
$page->insertOn( $this->getDb() );
}
$rev = new MutableRevisionRecord( $title );
$rev->setContent( SlotRecord::MAIN, new WikitextContent( $timestamp ) );
$rev->setComment( CommentStoreComment::newUnsavedComment( '' ) );
$rev->setTimestamp( $timestamp );
$rev->setUser( $user );
$rev->setPageId( $page->getId() );
$this->getServiceContainer()->getRevisionStore()->insertRevisionOn( $rev, $this->getDb() );
}
/**
* Change the user_editcount field in the DB
*
* @param UserIdentity $user
* @param int|null $count
*/
private function setDbEditCount( $user, $count ) {
$this->getDb()->newUpdateQueryBuilder()
->update( 'user' )
->set( [ 'user_editcount' => $count ] )
->where( [ 'user_id' => $user->getId() ] )
->caller( __METHOD__ )
->execute();
}
public function testGetUserEditCount() {
// Set user_editcount to 5
$user = $this->getMutableTestUser()->getUser();
$update = new UserEditCountUpdate( $user, 5 );
$update->doUpdate();
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->assertSame( 5, $tracker->getUserEditCount( $user ) );
// Now fetch from cache
$this->assertSame( 5, $tracker->getUserEditCount( $user ) );
}
public function testGetUserEditCount_anon() {
// getUserEditCount returns null if the user is unregistered
$anon = UserIdentityValue::newAnonymous( '1.2.3.4' );
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->assertNull( $tracker->getUserEditCount( $anon ) );
}
public function testGetUserEditCount_null() {
// getUserEditCount doesn't find a value in user_editcount and calls
// initializeUserEditCount
$user = $this->getMutableTestUser()->getUserIdentity();
$this->setDbEditCount( $user, null );
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->assertSame( 0, $tracker->getUserEditCount( $user ) );
}
public function testInitializeUserEditCount() {
$user = $this->getMutableTestUser()->getUser();
$this->editTrackerDoEdit( $user, '20200101000000' );
$tracker = $this->getServiceContainer()->getUserEditTracker();
$tracker->initializeUserEditCount( $user );
$this->runJobs();
$this->assertSame( 1, $tracker->getUserEditCount( $user ) );
}
public function testGetEditTimestamp() {
$user = $this->getMutableTestUser()->getUser();
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->assertFalse( $tracker->getFirstEditTimestamp( $user ) );
$this->assertFalse( $tracker->getLatestEditTimestamp( $user ) );
$ts1 = '20010101000000';
$ts2 = '20020101000000';
$ts3 = '20030101000000';
$this->editTrackerDoEdit( $user, $ts3 );
$this->editTrackerDoEdit( $user, $ts2 );
$this->editTrackerDoEdit( $user, $ts1 );
$this->assertSame( $ts1, $tracker->getFirstEditTimestamp( $user ) );
$this->assertSame( $ts3, $tracker->getLatestEditTimestamp( $user ) );
}
public function testGetEditTimestamp_anon() {
$this->disableAutoCreateTempUser();
$user = $this->getServiceContainer()->getUserFactory()
->newFromName( '127.0.0.1', UserRigorOptions::RIGOR_NONE );
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->editTrackerDoEdit( $user, '20200101000000' );
$this->assertFalse( $tracker->getFirstEditTimestamp( $user ) );
$this->assertFalse( $tracker->getLatestEditTimestamp( $user ) );
}
public function testClearUserEditCache() {
$user = $this->getMutableTestUser()->getUser();
$tracker = $this->getServiceContainer()->getUserEditTracker();
$this->assertSame( 0, $tracker->getUserEditCount( $user ) );
$this->setDbEditCount( $user, 1 );
$this->assertSame( 0, $tracker->getUserEditCount( $user ) );
$tracker->clearUserEditCache( $user );
$this->assertSame( 1, $tracker->getUserEditCount( $user ) );
}
public function testIncrementUserEditCount() {
$tracker = $this->getServiceContainer()->getUserEditTracker();
$user = $this->getMutableTestUser()->getUser();
$editCountStart = $tracker->getUserEditCount( $user );
$this->getDb()->startAtomic( __METHOD__ ); // let deferred updates queue up
$tracker->incrementUserEditCount( $user );
$this->assertSame(
1,
DeferredUpdates::pendingUpdatesCount(),
'Update queued for registered user'
);
$tracker->incrementUserEditCount( UserIdentityValue::newAnonymous( '1.1.1.1' ) );
$this->assertSame(
1,
DeferredUpdates::pendingUpdatesCount(),
'No update queued for anonymous user'
);
$this->getDb()->endAtomic( __METHOD__ ); // run deferred updates
$this->assertSame(
0,
DeferredUpdates::pendingUpdatesCount(),
'deferred updates ran'
);
$editCountEnd = $tracker->getUserEditCount( $user );
$this->assertSame(
1,
$editCountEnd - $editCountStart,
'Edit count was incremented'
);
}
public function testManualCache() {
// Make sure manually setting the cached value overrides the database, in case
// User::loadFromRow() is called with a row containing user_editcount that is
// different from the actual database value, the row takes precedence
$user = new UserIdentityValue( 123, __METHOD__ );
$this->setDbEditCount( $user, 5 );
$tracker = $this->getServiceContainer()->getUserEditTracker();
$tracker->setCachedUserEditCount( $user, 10 );
$this->assertSame( 10, $tracker->getUserEditCount( $user ) );
}
}

File Metadata

Mime Type
text/x-php
Expires
Aug 19 2026, 17:43 (5 w, 4 d ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
25/ea/adc694344216939d25154f701369
Default Alt Text
UserEditTrackerTest.php (6 KB)

Event Timeline