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:
Marius Hoch 2014-10-20 00:22:02 +02:00
parent 5aa4042bd2
commit ee9166a10e
3 changed files with 66 additions and 5 deletions

View file

@ -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();

View file

@ -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
*

View file

@ -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;
}
/**