ElyBy-Mirror/accounts
1
0
Fork 0
mirror of https://github.com/elyby/accounts.git synced 2025-04-27 05:14:19 +05:30
Code Issues Packages Projects Releases Wiki Activity

Cleanup code, improve typings

Browse Source
This commit is contained in:
ErickSkrauch 2019-12-13 22:27:13 +03:00
parent 830a17612b
commit d9ef27b745
28 changed files with 189 additions and 225 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/controllers/AuthenticationController.php
View File

@ -107,7 +107,7 @@ class AuthenticationController extends Controller {
],
];
if ($model->getLoginAttribute() !== 'email') {
if (strpos($model->login, '@') === false) {
$response['data']['emailMask'] = StringHelper::getEmailMask($model->getAccount()->email);
}

35
api/models/authentication/ForgotPasswordForm.php
View File

@ -1,10 +1,11 @@
<?php
declare(strict_types=1);
namespace api\models\authentication;
use api\aop\annotations\CollectModelMetrics;
use api\components\ReCaptcha\Validator as ReCaptchaValidator;
use api\models\base\ApiForm;
use api\traits\AccountFinder;
use common\components\UserFriendlyRandomKey;
use common\helpers\Error as E;
use common\models\Account;
@ -15,13 +16,12 @@ use Yii;
use yii\base\ErrorException;
class ForgotPasswordForm extends ApiForm {
use AccountFinder;
public $captcha;
public $login;
public function rules() {
public function rules(): array {
return [
['captcha', ReCaptchaValidator::class],
['login', 'required', 'message' => E::LOGIN_REQUIRED],
@ -31,7 +31,7 @@ class ForgotPasswordForm extends ApiForm {
];
}
public function validateLogin($attribute) {
public function validateLogin(string $attribute): void {
if (!$this->hasErrors()) {
if ($this->getAccount() === null) {
$this->addError($attribute, E::LOGIN_NOT_EXIST);
@ -39,7 +39,7 @@ class ForgotPasswordForm extends ApiForm {
}
}
public function validateActivity($attribute) {
public function validateActivity(string $attribute): void {
if (!$this->hasErrors()) {
$account = $this->getAccount();
if ($account->status !== Account::STATUS_ACTIVE) {
@ -48,7 +48,7 @@ class ForgotPasswordForm extends ApiForm {
}
}
public function validateFrequency($attribute) {
public function validateFrequency(string $attribute): void {
if (!$this->hasErrors()) {
$emailConfirmation = $this->getEmailActivation();
if ($emailConfirmation !== null && !$emailConfirmation->canRepeat()) {
@ -57,12 +57,15 @@ class ForgotPasswordForm extends ApiForm {
}
}
public function getAccount(): ?Account {
return Account::find()->andWhereLogin($this->login)->one();
}
/**
* @CollectModelMetrics(prefix="authentication.forgotPassword")
* @return bool
* @throws ErrorException
*/
public function forgotPassword() {
public function forgotPassword(): bool {
if (!$this->validate()) {
return false;
}
@ -86,23 +89,13 @@ class ForgotPasswordForm extends ApiForm {
return true;
}
public function getLogin(): string {
return $this->login;
}
/**
* @return EmailActivation|null
* @throws ErrorException
*/
public function getEmailActivation() {
public function getEmailActivation(): ?EmailActivation {
$account = $this->getAccount();
if ($account === null) {
throw new ErrorException('Account not founded');
return null;
}
return $account->getEmailActivations()
->andWhere(['type' => EmailActivation::TYPE_FORGOT_PASSWORD_KEY])
->one();
return $account->getEmailActivations()->withType(EmailActivation::TYPE_FORGOT_PASSWORD_KEY)->one();
}
}

18
api/models/authentication/LoginForm.php
View File

@ -5,7 +5,6 @@ namespace api\models\authentication;
use api\aop\annotations\CollectModelMetrics;
use api\models\base\ApiForm;
use api\traits\AccountFinder;
use api\validators\TotpValidator;
use common\helpers\Error as E;
use common\models\Account;
@ -14,14 +13,25 @@ use Webmozart\Assert\Assert;
use Yii;
class LoginForm extends ApiForm {
use AccountFinder;
/**
* @var string
*/
public $login;
/**
* @var string
*/
public $password;
/**
* @var string|null
*/
public $totp;
/**
* @var bool
*/
public $rememberMe = false;
public function rules(): array {
@ -90,8 +100,8 @@ class LoginForm extends ApiForm {
}
}
public function getLogin(): string {
return $this->login;
public function getAccount(): ?Account {
return Account::find()->andWhereLogin($this->login)->one();
}
/**

33
api/models/authentication/RepeatAccountActivationForm.php
View File

@ -1,4 +1,6 @@
<?php
declare(strict_types=1);
namespace api\models\authentication;
use api\aop\annotations\CollectModelMetrics;
@ -21,7 +23,7 @@ class RepeatAccountActivationForm extends ApiForm {
private $emailActivation;
public function rules() {
public function rules(): array {
return [
['captcha', ReCaptchaValidator::class],
['email', 'filter', 'filter' => 'trim'],
@ -31,7 +33,7 @@ class RepeatAccountActivationForm extends ApiForm {
];
}
public function validateEmailForAccount($attribute) {
public function validateEmailForAccount(string $attribute): void {
if (!$this->hasErrors()) {
$account = $this->getAccount();
if ($account === null) {
@ -45,7 +47,7 @@ class RepeatAccountActivationForm extends ApiForm {
}
}
public function validateExistsActivation($attribute) {
public function validateExistsActivation(string $attribute): void {
if (!$this->hasErrors()) {
$activation = $this->getActivation();
if ($activation !== null && !$activation->canRepeat()) {
@ -58,11 +60,12 @@ class RepeatAccountActivationForm extends ApiForm {
* @CollectModelMetrics(prefix="signup.repeatEmail")
* @return bool
*/
public function sendRepeatMessage() {
public function sendRepeatMessage(): bool {
if (!$this->validate()) {
return false;
}
/** @var Account $account */
$account = $this->getAccount();
$transaction = Yii::$app->db->beginTransaction();
@ -85,27 +88,17 @@ class RepeatAccountActivationForm extends ApiForm {
return true;
}
/**
* @return Account|null
*/
public function getAccount() {
public function getAccount(): ?Account {
return Account::find()
->andWhere(['email' => $this->email])
->one();
}
/**
* @return EmailActivation|null
*/
public function getActivation() {
if ($this->emailActivation === null) {
$this->emailActivation = $this->getAccount()
->getEmailActivations()
->andWhere(['type' => EmailActivation::TYPE_REGISTRATION_EMAIL_CONFIRMATION])
->one();
}
return $this->emailActivation;
public function getActivation(): ?EmailActivation {
return $this->getAccount()
->getEmailActivations()
->withType(EmailActivation::TYPE_REGISTRATION_EMAIL_CONFIRMATION)
->one();
}
}

14
api/modules/accounts/models/SendEmailVerificationForm.php
View File

@ -1,4 +1,6 @@
<?php
declare(strict_types=1);
namespace api\modules\accounts\models;
use api\aop\annotations\CollectModelMetrics;
@ -26,7 +28,7 @@ class SendEmailVerificationForm extends AccountActionForm {
];
}
public function validateFrequency($attribute): void {
public function validateFrequency(string $attribute): void {
if (!$this->hasErrors()) {
$emailConfirmation = $this->getEmailActivation();
if ($emailConfirmation !== null && !$emailConfirmation->canRepeat()) {
@ -82,12 +84,10 @@ class SendEmailVerificationForm extends AccountActionForm {
public function getEmailActivation(): ?EmailActivation {
return $this->getAccount()
->getEmailActivations()
->andWhere([
'type' => [
EmailActivation::TYPE_CURRENT_EMAIL_CONFIRMATION,
EmailActivation::TYPE_NEW_EMAIL_CONFIRMATION,
],
])
->withType(
EmailActivation::TYPE_CURRENT_EMAIL_CONFIRMATION,
EmailActivation::TYPE_NEW_EMAIL_CONFIRMATION
)
->one();
}

5
api/modules/authserver/models/AuthenticationForm.php
View File

@ -75,10 +75,7 @@ class AuthenticationForm extends ApiForm {
// The previous authorization server implementation used the nickname field instead of username,
// so we keep such behavior
$attribute = $loginForm->getLoginAttribute();
if ($attribute === 'username') {
$attribute = 'nickname';
}
$attribute = strpos($this->username, '@') === false ? 'nickname' : 'email';
// TODO: эта логика дублируется с логикой в SignoutForm

5
api/modules/authserver/models/SignoutForm.php
View File

@ -47,10 +47,7 @@ class SignoutForm extends ApiForm {
// The previous authorization server implementation used the nickname field instead of username,
// so we keep such behavior
$attribute = $loginForm->getLoginAttribute();
if ($attribute === 'username') {
$attribute = 'nickname';
}
$attribute = strpos($this->username, '@') === false ? 'nickname' : 'email';
throw new ForbiddenOperationException("Invalid credentials. Invalid {$attribute} or password.");
}

11
api/request/RequestParser.php
View File

@ -1,4 +1,6 @@
<?php
declare(strict_types=1);
namespace api\request;
use Yii;
@ -18,7 +20,14 @@ use yii\web\RequestParserInterface;
*/
class RequestParser implements RequestParserInterface {
public function parse($rawBody, $contentType) {
/**
* @param string $rawBody
* @param string $contentType
*
* @return array
* @throws \yii\web\BadRequestHttpException
*/
public function parse($rawBody, $contentType): array {
if (!empty($_POST)) {
return $_POST;
}

4
api/tests/unit/models/authentication/ForgotPasswordFormTest.php
View File

@ -126,8 +126,8 @@ class ForgotPasswordFormTest extends TestCase {
return new class($params) extends ForgotPasswordForm {
public $key;
public function getEmailActivation() {
return EmailActivation::findOne($this->key);
public function getEmailActivation(): ?EmailActivation {
return EmailActivation::findOne(['key' => $this->key]);
}
};
}

2
api/tests/unit/models/authentication/RepeatAccountActivationFormTest.php
View File

@ -100,7 +100,7 @@ class RepeatAccountActivationFormTest extends TestCase {
return new class($params) extends RepeatAccountActivationForm {
public $emailKey;
public function getActivation() {
public function getActivation(): ?EmailActivation {
return EmailActivation::findOne($this->emailKey);
}
};

1
api/tests/unit/modules/accounts/models/ChangeEmailFormTest.php
View File

@ -29,7 +29,6 @@ class ChangeEmailFormTest extends TestCase {
'account_id' => $account->id,
'type' => EmailActivation::TYPE_NEW_EMAIL_CONFIRMATION,
]));
/** @noinspection UnserializeExploitsInspection */
$data = unserialize($newEmailConfirmationFixture['_data']);
$this->assertSame($data['newEmail'], $account->email);
}

52
api/tests/unit/traits/AccountFinderTest.php
View File

@ -1,52 +0,0 @@
<?php
namespace api\tests\_support\traits;
use api\tests\unit\TestCase;
use api\traits\AccountFinder;
use common\models\Account;
use common\tests\fixtures\AccountFixture;
class AccountFinderTest extends TestCase {
public function _fixtures(): array {
return [
'accounts' => AccountFixture::class,
];
}
public function testGetAccount() {
$model = new AccountFinderTestTestClass();
/** @var Account $account */
$accountFixture = $this->tester->grabFixture('accounts', 'admin');
$model->login = $accountFixture->email;
$account = $model->getAccount();
$this->assertInstanceOf(Account::class, $account);
$this->assertSame($accountFixture->id, $account->id, 'founded account for passed login data');
$model = new AccountFinderTestTestClass();
$model->login = 'unexpected';
$this->assertNull($account = $model->getAccount(), 'null, if account can\'t be found');
}
public function testGetLoginAttribute() {
$model = new AccountFinderTestTestClass();
$model->login = 'erickskrauch@ely.by';
$this->assertSame('email', $model->getLoginAttribute(), 'if login look like email value, then \'email\'');
$model = new AccountFinderTestTestClass();
$model->login = 'erickskrauch';
$this->assertSame('username', $model->getLoginAttribute(), 'username in any other case');
}
}
class AccountFinderTestTestClass {
use AccountFinder;
public $login;
public function getLogin(): string {
return $this->login;
}
}

24
api/traits/AccountFinder.php
View File

@ -1,24 +0,0 @@
<?php
namespace api\traits;
use common\models\Account;
trait AccountFinder {
private $account;
abstract public function getLogin(): string;
public function getAccount(): ?Account {
if ($this->account === null) {
$this->account = Account::findOne([$this->getLoginAttribute() => $this->getLogin()]);
}
return $this->account;
}
public function getLoginAttribute(): string {
return strpos($this->getLogin(), '@') ? 'email' : 'username';
}
}

4
api/validators/EmailActivationKeyValidator.php
View File

@ -24,7 +24,7 @@ class EmailActivationKeyValidator extends Validator {
public $skipOnEmpty = false;
public function validateAttribute($model, $attribute) {
public function validateAttribute($model, $attribute): void {
$value = $model->$attribute;
if (empty($value)) {
$this->addError($model, $attribute, $this->keyRequired);
@ -49,7 +49,7 @@ class EmailActivationKeyValidator extends Validator {
$query = EmailActivation::find();
$query->andWhere(['key' => $key]);
if ($type !== null) {
$query->andWhere(['type' => $type]);
$query->withType($type);
}
return $query->one();

6
common/models/Account.php
View File

@ -59,6 +59,10 @@ class Account extends ActiveRecord {
return '{{%accounts}}';
}
public static function find(): AccountQuery {
return new AccountQuery(self::class);
}
public function behaviors(): array {
return [
TimestampBehavior::class,
@ -88,7 +92,7 @@ class Account extends ActiveRecord {
$this->password_changed_at = time();
}
public function getEmailActivations(): ActiveQuery {
public function getEmailActivations(): EmailActivationQuery {
return $this->hasMany(EmailActivation::class, ['account_id' => 'id']);
}

21
common/models/AccountQuery.php Normal file
View File

@ -0,0 +1,21 @@
<?php
declare(strict_types=1);
namespace common\models;
use yii\db\ActiveQuery;
/**
* @see Account
*/
class AccountQuery extends ActiveQuery {
public function andWhereLogin(string $login): self {
return $this->andWhere([$this->getLoginAttribute($login) => $login]);
}
private function getLoginAttribute(string $login): string {
return strpos($login, '@') ? 'email' : 'username';
}
}

16
common/models/EmailActivation.php
View File

@ -1,4 +1,6 @@
<?php
declare(strict_types=1);
namespace common\models;
use common\behaviors\DataBehavior;
@ -33,11 +35,15 @@ class EmailActivation extends ActiveRecord {
public const TYPE_CURRENT_EMAIL_CONFIRMATION = 2;
public const TYPE_NEW_EMAIL_CONFIRMATION = 3;
public static function tableName() {
return '{{%email_activations}}';
public static function tableName(): string {
return 'email_activations';
}
public function behaviors() {
public static function find(): EmailActivationQuery {
return new EmailActivationQuery(static::class);
}
public function behaviors(): array {
return [
[
'class' => TimestampBehavior::class,
@ -61,7 +67,7 @@ class EmailActivation extends ActiveRecord {
];
}
public function getAccount() {
public function getAccount(): AccountQuery {
return $this->hasOne(Account::class, ['id' => 'account_id']);
}
@ -82,7 +88,7 @@ class EmailActivation extends ActiveRecord {
return new $classMap[$type]();
}
public static function getClassMap() {
public static function getClassMap(): array {
return [
self::TYPE_REGISTRATION_EMAIL_CONFIRMATION => confirmations\RegistrationConfirmation::class,
self::TYPE_FORGOT_PASSWORD_KEY => confirmations\ForgotPassword::class,

17
common/models/EmailActivationQuery.php Normal file
View File

@ -0,0 +1,17 @@
<?php
declare(strict_types=1);
namespace common\models;
use yii\db\ActiveQuery;
/**
* @see EmailActivation
*/
class EmailActivationQuery extends ActiveQuery {
public function withType(int ...$typeId): self {
return $this->andWhere(['type' => $typeId]);
}
}

50
common/models/OauthScopeQuery.php
View File

@ -1,50 +0,0 @@
<?php
namespace common\models;
use yii\helpers\ArrayHelper;
class OauthScopeQuery {
private $scopes;
private $internal;
private $owner;
public function __construct(array $scopes) {
$this->scopes = $scopes;
}
public function onlyPublic(): self {
$this->internal = false;
return $this;
}
public function onlyInternal(): self {
$this->internal = true;
return $this;
}
public function usersScopes(): self {
$this->owner = 'user';
return $this;
}
public function machineScopes(): self {
$this->owner = 'machine';
return $this;
}
public function all(): array {
return ArrayHelper::getColumn(array_filter($this->scopes, function($value) {
$shouldCheckInternal = $this->internal !== null;
$isInternalMatch = $value['internal'] === $this->internal;
$shouldCheckOwner = $this->owner !== null;
$isOwnerMatch = $value['owner'] === $this->owner;
return (!$shouldCheckInternal || $isInternalMatch)
&& (!$shouldCheckOwner || $isOwnerMatch);
}), 'value');
}
}

11
common/models/confirmations/CurrentEmailConfirmation.php
View File

@ -1,12 +1,19 @@
<?php
declare(strict_types=1);
namespace common\models\confirmations;
use common\models\EmailActivation;
use common\models\EmailActivationQuery;
use yii\helpers\ArrayHelper;
class CurrentEmailConfirmation extends EmailActivation {
public function behaviors() {
public static function find(): EmailActivationQuery {
return parent::find()->withType(EmailActivation::TYPE_CURRENT_EMAIL_CONFIRMATION);
}
public function behaviors(): array {
return ArrayHelper::merge(parent::behaviors(), [
'expirationBehavior' => [
'repeatTimeout' => 6 * 60 * 60, // 6h
@ -15,7 +22,7 @@ class CurrentEmailConfirmation extends EmailActivation {
]);
}
public function init() {
public function init(): void {
parent::init();
$this->type = EmailActivation::TYPE_CURRENT_EMAIL_CONFIRMATION;
}

11
common/models/confirmations/ForgotPassword.php
View File

@ -1,12 +1,19 @@
<?php
declare(strict_types=1);
namespace common\models\confirmations;
use common\models\EmailActivation;
use common\models\EmailActivationQuery;
use yii\helpers\ArrayHelper;
class ForgotPassword extends EmailActivation {
public function behaviors() {
public static function find(): EmailActivationQuery {
return parent::find()->withType(EmailActivation::TYPE_FORGOT_PASSWORD_KEY);
}
public function behaviors(): array {
return ArrayHelper::merge(parent::behaviors(), [
'expirationBehavior' => [
'repeatTimeout' => 30 * 60,
@ -15,7 +22,7 @@ class ForgotPassword extends EmailActivation {
]);
}
public function init() {
public function init(): void {
parent::init();
$this->type = EmailActivation::TYPE_FORGOT_PASSWORD_KEY;
}

11
common/models/confirmations/NewEmailConfirmation.php
View File

@ -1,7 +1,10 @@
<?php
declare(strict_types=1);
namespace common\models\confirmations;
use common\models\EmailActivation;
use common\models\EmailActivationQuery;
use yii\helpers\ArrayHelper;
/**
@ -10,7 +13,11 @@ use yii\helpers\ArrayHelper;
*/
class NewEmailConfirmation extends EmailActivation {
public function behaviors() {
public static function find(): EmailActivationQuery {
return parent::find()->withType(EmailActivation::TYPE_NEW_EMAIL_CONFIRMATION);
}
public function behaviors(): array {
return ArrayHelper::merge(parent::behaviors(), [
'expirationBehavior' => [
'repeatTimeout' => 5 * 60,
@ -21,7 +28,7 @@ class NewEmailConfirmation extends EmailActivation {
]);
}
public function init() {
public function init(): void {
parent::init();
$this->type = EmailActivation::TYPE_NEW_EMAIL_CONFIRMATION;
}

2
common/models/confirmations/NewEmailConfirmationBehavior.php
View File

@ -1,4 +1,6 @@
<?php
declare(strict_types=1);
namespace common\models\confirmations;
use common\behaviors\DataBehavior;

9
common/models/confirmations/RegistrationConfirmation.php
View File

@ -1,11 +1,18 @@
<?php
declare(strict_types=1);
namespace common\models\confirmations;
use common\models\EmailActivation;
use common\models\EmailActivationQuery;
class RegistrationConfirmation extends EmailActivation {
public function init() {
public static function find(): EmailActivationQuery {
return parent::find()->withType(EmailActivation::TYPE_REGISTRATION_EMAIL_CONFIRMATION);
}
public function init(): void {
parent::init();
$this->type = EmailActivation::TYPE_REGISTRATION_EMAIL_CONFIRMATION;
}

14
common/tests/unit/tasks/SendCurrentEmailConfirmationTest.php
View File

@ -1,7 +1,10 @@
<?php
declare(strict_types=1);
namespace common\tests\unit\tasks;
use common\models\Account;
use common\models\AccountQuery;
use common\models\confirmations\CurrentEmailConfirmation;
use common\tasks\SendCurrentEmailConfirmation;
use common\tests\unit\TestCase;
@ -15,13 +18,14 @@ class SendCurrentEmailConfirmationTest extends TestCase {
$account->email = 'mock@ely.by';
$account->lang = 'id';
/** @var \Mockery\Mock|CurrentEmailConfirmation $confirmation */
$confirmation = mock(CurrentEmailConfirmation::class)->makePartial();
$accountQuery = $this->createMock(AccountQuery::class);
$accountQuery->method('findFor')->willReturn($account);
$confirmation = $this->createPartialMock(CurrentEmailConfirmation::class, ['getAccount']);
$confirmation->method('getAccount')->willReturn($accountQuery);
$confirmation->key = 'ABCDEFG';
$confirmation->shouldReceive('getAccount')->andReturn($account);
$result = SendCurrentEmailConfirmation::createFromConfirmation($confirmation);
$this->assertInstanceOf(SendCurrentEmailConfirmation::class, $result);
$this->assertSame('mock-username', $result->username);
$this->assertSame('mock@ely.by', $result->email);
$this->assertSame('ABCDEFG', $result->code);
@ -33,7 +37,7 @@ class SendCurrentEmailConfirmationTest extends TestCase {
$task->email = 'mock@ely.by';
$task->code = 'GFEDCBA';
$task->execute(mock(Queue::class));
$task->execute($this->createMock(Queue::class));
$this->tester->canSeeEmailIsSent(1);
/** @var \yii\swiftmailer\Message $email */

16
common/tests/unit/tasks/SendNewEmailConfirmationTest.php
View File

@ -1,7 +1,10 @@
<?php
declare(strict_types=1);
namespace common\tests\unit\tasks;
use common\models\Account;
use common\models\AccountQuery;
use common\models\confirmations\NewEmailConfirmation;
use common\tasks\SendNewEmailConfirmation;
use common\tests\unit\TestCase;
@ -14,14 +17,15 @@ class SendNewEmailConfirmationTest extends TestCase {
$account->username = 'mock-username';
$account->lang = 'id';
/** @var \Mockery\Mock|NewEmailConfirmation $confirmation */
$confirmation = mock(NewEmailConfirmation::class)->makePartial();
$accountQuery = $this->createMock(AccountQuery::class);
$accountQuery->method('findFor')->willReturn($account);
$confirmation = $this->createPartialMock(NewEmailConfirmation::class, ['getAccount']);
$confirmation->method('getAccount')->willReturn($accountQuery);
$confirmation->setNewEmail('new-email@ely.by');
$confirmation->key = 'ABCDEFG';
$confirmation->shouldReceive('getAccount')->andReturn($account);
$confirmation->shouldReceive('getNewEmail')->andReturn('new-email@ely.by');
$result = SendNewEmailConfirmation::createFromConfirmation($confirmation);
$this->assertInstanceOf(SendNewEmailConfirmation::class, $result);
$this->assertSame('mock-username', $result->username);
$this->assertSame('new-email@ely.by', $result->email);
$this->assertSame('ABCDEFG', $result->code);
@ -33,7 +37,7 @@ class SendNewEmailConfirmationTest extends TestCase {
$task->email = 'mock@ely.by';
$task->code = 'GFEDCBA';
$task->execute(mock(Queue::class));
$task->execute($this->createMock(Queue::class));
$this->tester->canSeeEmailIsSent(1);
/** @var \yii\swiftmailer\Message $email */

11
common/tests/unit/tasks/SendPasswordRecoveryEmailTest.php
View File

@ -5,6 +5,7 @@ namespace common\tests\unit\tasks;
use common\emails\RendererInterface;
use common\models\Account;
use common\models\AccountQuery;
use common\models\confirmations\ForgotPassword;
use common\tasks\SendPasswordRecoveryEmail;
use common\tests\unit\TestCase;
@ -24,10 +25,12 @@ class SendPasswordRecoveryEmailTest extends TestCase {
$account->email = 'mock@ely.by';
$account->lang = 'id';
/** @var \Mockery\Mock|ForgotPassword $confirmation */
$confirmation = mock(ForgotPassword::class)->makePartial();
$accountQuery = $this->createMock(AccountQuery::class);
$accountQuery->method('findFor')->willReturn($account);
$confirmation = $this->createPartialMock(ForgotPassword::class, ['getAccount']);
$confirmation->method('getAccount')->willReturn($accountQuery);
$confirmation->key = 'ABCDEFG';
$confirmation->shouldReceive('getAccount')->andReturn($account);
$result = SendPasswordRecoveryEmail::createFromConfirmation($confirmation);
$this->assertSame('mock-username', $result->username);
@ -51,7 +54,7 @@ class SendPasswordRecoveryEmailTest extends TestCase {
'link' => 'https://account.ely.by/recover-password/ABCDEFG',
])->willReturn('mock-template');
$task->execute(mock(Queue::class));
$task->execute($this->createMock(Queue::class));
$this->tester->canSeeEmailIsSent(1);
/** @var \yii\swiftmailer\Message $email */

9
common/tests/unit/tasks/SendRegistrationEmailTest.php
View File

@ -5,6 +5,7 @@ namespace common\tests\unit\tasks;
use common\emails\RendererInterface;
use common\models\Account;
use common\models\AccountQuery;
use common\models\confirmations\RegistrationConfirmation;
use common\tasks\SendRegistrationEmail;
use common\tests\unit\TestCase;
@ -24,10 +25,12 @@ class SendRegistrationEmailTest extends TestCase {
$account->email = 'mock@ely.by';
$account->lang = 'ru';
/** @var \Mockery\Mock|RegistrationConfirmation $confirmation */
$confirmation = mock(RegistrationConfirmation::class)->makePartial();
$accountQuery = $this->createMock(AccountQuery::class);
$accountQuery->method('findFor')->willReturn($account);
$confirmation = $this->createPartialMock(RegistrationConfirmation::class, ['getAccount']);
$confirmation->method('getAccount')->willReturn($accountQuery);
$confirmation->key = 'ABCDEFG';
$confirmation->shouldReceive('getAccount')->andReturn($account);
$result = SendRegistrationEmail::createFromConfirmation($confirmation);
$this->assertSame('mock-username', $result->username);