Make TestUser way faster
By only updating the user row in the database, if needed and by making use of md5 for passwords instead of slower hashing. This cut down run time of some Wikibase API tests to 20% of the original value for me! Also it reduces the run time of MediaWiki's test suite by about 45s on jenkins. Change-Id: I7024b287a71fe9b327dbcdc5427cd8edb5047606
This commit is contained in:
parent
5aa4042bd2
commit
ee9166a10e
3 changed files with 66 additions and 5 deletions
|
|
@ -3817,7 +3817,6 @@ class User implements IDBAccessObject {
|
|||
return false;
|
||||
}
|
||||
|
||||
$passwordFactory = self::getPasswordFactory();
|
||||
if ( !$this->mPassword->equals( $password ) ) {
|
||||
if ( $wgLegacyEncoding ) {
|
||||
// Some wikis were converted from ISO 8859-1 to UTF-8, the passwords can't be converted
|
||||
|
|
@ -3831,6 +3830,7 @@ class User implements IDBAccessObject {
|
|||
}
|
||||
}
|
||||
|
||||
$passwordFactory = self::getPasswordFactory();
|
||||
if ( $passwordFactory->needsUpdate( $this->mPassword ) ) {
|
||||
$this->mPassword = $passwordFactory->newFromPlaintext( $password );
|
||||
$this->saveSettings();
|
||||
|
|
|
|||
|
|
@ -68,6 +68,15 @@ final class PasswordFactory {
|
|||
$this->default = $type;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the default password type
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
public function getDefaultType() {
|
||||
return $this->default;
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the internal static variables using the global variables
|
||||
*
|
||||
|
|
|
|||
|
|
@ -63,9 +63,9 @@ class TestUser {
|
|||
}
|
||||
|
||||
// Update the user to use the password and other details
|
||||
$this->user->setPassword( $this->password );
|
||||
$this->user->setEmail( $email );
|
||||
$this->user->setRealName( $realname );
|
||||
$change = $this->setPassword( $this->password ) ||
|
||||
$this->setEmail( $email ) ||
|
||||
$this->setRealName( $realname );
|
||||
|
||||
// Adjust groups by adding any missing ones and removing any extras
|
||||
$currentGroups = $this->user->getGroups();
|
||||
|
|
@ -75,7 +75,59 @@ class TestUser {
|
|||
foreach ( array_diff( $currentGroups, $groups ) as $group ) {
|
||||
$this->user->removeGroup( $group );
|
||||
}
|
||||
$this->user->saveSettings();
|
||||
if ( $change ) {
|
||||
$this->user->saveSettings();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $realname
|
||||
* @return bool
|
||||
*/
|
||||
private function setRealName( $realname ) {
|
||||
if ( $this->user->getRealName() !== $realname ) {
|
||||
$this->user->setRealName( $realname );
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $email
|
||||
* @return bool
|
||||
*/
|
||||
private function setEmail( $email ) {
|
||||
if ( $this->user->getEmail() !== $email ) {
|
||||
$this->user->setEmail( $email );
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $password
|
||||
* @return bool
|
||||
*/
|
||||
private function setPassword( $password ) {
|
||||
$passwordFactory = $this->user->getPasswordFactory();
|
||||
$oldDefaultType = $passwordFactory->getDefaultType();
|
||||
|
||||
// B is salted MD5 (thus fast) ... we don't care about security here, this is test only
|
||||
$passwordFactory->setDefaultType( 'B' ); // @TODO: Change this to A once that is fixed: https://gerrit.wikimedia.org/r/167523
|
||||
$newPassword = $passwordFactory->newFromPlaintext( $password , $this->user->getPassword() );
|
||||
|
||||
$change = false;
|
||||
if ( !$this->user->getPassword()->equals( $newPassword ) ) {
|
||||
// Password changed
|
||||
$this->user->setPassword( $password );
|
||||
$change = true;
|
||||
}
|
||||
|
||||
$passwordFactory->setDefaultType( $oldDefaultType );
|
||||
|
||||
return $change;
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
Loading…
Reference in a new issue