magento2/app/code/Magento/Security/Test/Unit/Observer/AfterAdminUserSaveTest.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Security\Test\Unit\Observer;

use Magento\Framework\Event;
use Magento\Framework\Event\Observer;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
use Magento\Security\Model\ResourceModel\UserExpiration;
use Magento\Security\Model\UserExpirationFactory;
use Magento\Security\Observer\AfterAdminUserSave;
use Magento\User\Model\User;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;

/**
 * Test class for \Magento\Security\Observer\AfterAdminUserSave
 */
class AfterAdminUserSaveTest extends TestCase
{

    /**
     * @var MockObject|UserExpirationFactory
     */
    private $userExpirationFactoryMock;

    /**
     * @var MockObject|UserExpiration
     */
    private $userExpirationResourceMock;

    /**
     * @var AfterAdminUserSave
     */
    private $observer;

    /**
     * @var ObjectManager
     */
    private $objectManager;

    /**
     * @var MockObject|Observer
     */
    private $eventObserverMock;

    /**
     * @var MockObject|Event
     */
    private $eventMock;

    /**
     * @var MockObject|User
     */
    private $userMock;

    /**
     * @var MockObject|\Magento\Security\Model\UserExpiration
     */
    private $userExpirationMock;

    protected function setUp(): void
    {
        $this->objectManager = new ObjectManager($this);

        $this->userExpirationFactoryMock = $this->createMock(UserExpirationFactory::class);
        $this->userExpirationResourceMock = $this->createPartialMock(
            UserExpiration::class,
            ['load', 'save', 'delete']
        );
        $this->observer = $this->objectManager->getObject(
            AfterAdminUserSave::class,
            [
                'userExpirationFactory' => $this->userExpirationFactoryMock,
                'userExpirationResource' => $this->userExpirationResourceMock,
            ]
        );
        $this->eventObserverMock = $this->createPartialMock(Observer::class, ['getEvent']);
        $this->eventMock = $this->getMockBuilder(Event::class)
            ->addMethods(['getObject'])
            ->disableOriginalConstructor()
            ->getMock();
        $this->userMock = $this->getMockBuilder(User::class)
            ->addMethods(['getExpiresAt'])
            ->onlyMethods(['getId', 'hasData'])
            ->disableOriginalConstructor()
            ->getMock();
        $this->userExpirationMock = $this->createPartialMock(
            \Magento\Security\Model\UserExpiration::class,
            ['getId', 'getExpiresAt', 'setId', 'setExpiresAt']
        );
    }

    /**
     * @return void
     */
    public function testSaveNewUserExpiration(): void
    {
        $userId = '123';
        $this->eventObserverMock->expects(static::once())->method('getEvent')->willReturn($this->eventMock);
        $this->eventMock->expects(static::once())->method('getObject')->willReturn($this->userMock);
        $this->userMock->expects(static::exactly(3))->method('getId')->willReturn($userId);
        $this->userMock->expects(static::once())->method('getExpiresAt')->willReturn($this->getExpiresDateTime());
        $this->userMock->expects(static::once())
            ->method('hasData')
            ->with('expires_at')
            ->willReturn(true);
        $this->userExpirationFactoryMock->expects(static::once())->method('create')
            ->willReturn($this->userExpirationMock);
        $this->userExpirationResourceMock->expects(static::once())->method('load')
            ->willReturn($this->userExpirationMock);

        $this->userExpirationMock->expects(static::once())->method('getId')->willReturn(null);
        $this->userExpirationMock->expects(static::once())->method('setId')->willReturn($this->userExpirationMock);
        $this->userExpirationMock->expects(static::once())->method('setExpiresAt')
            ->willReturn($this->userExpirationMock);
        $this->userExpirationResourceMock->expects(static::once())->method('save')
            ->willReturn($this->userExpirationResourceMock);
        $this->observer->execute($this->eventObserverMock);
    }

    /**
     * @throws \Exception
     */
    public function testClearUserExpiration(): void
    {
        $userId = '123';
        $this->userExpirationMock->setId($userId);

        $this->eventObserverMock->expects(static::once())->method('getEvent')->willReturn($this->eventMock);
        $this->eventMock->expects(static::once())->method('getObject')->willReturn($this->userMock);
        $this->userMock->expects(static::exactly(2))->method('getId')->willReturn($userId);
        $this->userMock->expects(static::once())->method('getExpiresAt')->willReturn(null);
        $this->userMock->expects(static::once())
            ->method('hasData')
            ->with('expires_at')
            ->willReturn(true);
        $this->userExpirationFactoryMock->expects(static::once())->method('create')
            ->willReturn($this->userExpirationMock);
        $this->userExpirationResourceMock->expects(static::once())->method('load')
            ->willReturn($this->userExpirationMock);

        $this->userExpirationMock->expects(static::once())->method('getId')->willReturn($userId);
        $this->userExpirationResourceMock->expects(static::once())->method('delete')
            ->willReturn($this->userExpirationResourceMock);
        $this->observer->execute($this->eventObserverMock);
    }

    /**
     * @return void
     */
    public function testChangeUserExpiration(): void
    {
        $userId = '123';
        $this->userExpirationMock->setId($userId);

        $this->eventObserverMock->expects(static::once())->method('getEvent')->willReturn($this->eventMock);
        $this->eventMock->expects(static::once())->method('getObject')->willReturn($this->userMock);
        $this->userMock->expects(static::exactly(2))->method('getId')->willReturn($userId);
        $this->userMock->expects(static::once())->method('getExpiresAt')->willReturn($this->getExpiresDateTime());
        $this->userMock->expects(static::once())
            ->method('hasData')
            ->with('expires_at')
            ->willReturn(true);
        $this->userExpirationFactoryMock->expects(static::once())->method('create')
            ->willReturn($this->userExpirationMock);
        $this->userExpirationResourceMock->expects(static::once())->method('load')
            ->willReturn($this->userExpirationMock);

        $this->userExpirationMock->expects(static::once())->method('getId')->willReturn($userId);
        $this->userExpirationMock->expects(static::once())->method('setExpiresAt')
            ->willReturn($this->userExpirationMock);
        $this->userExpirationResourceMock->expects(static::once())->method('save')
            ->willReturn($this->userExpirationResourceMock);
        $this->observer->execute($this->eventObserverMock);
    }

    /**
     * @return void
     */
    public function testExecuteWithoutUserExpiration(): void
    {
        $userId = '123';
        $this->userExpirationMock->setId($userId);

        $this->eventObserverMock->expects(static::once())->method('getEvent')->willReturn($this->eventMock);
        $this->eventMock->expects(static::once())->method('getObject')->willReturn($this->userMock);
        $this->userMock->expects(static::once())->method('getId')->willReturn($userId);
        $this->userMock->expects(static::once())
            ->method('hasData')
            ->with('expires_at')
            ->willReturn(false);
        $this->userExpirationFactoryMock->expects(static::never())->method('create');
        $this->userExpirationResourceMock->expects(static::never())->method('load');

        $this->userExpirationMock->expects(static::never())->method('getId');
        $this->userExpirationMock->expects(static::never())->method('setExpiresAt');
        $this->userExpirationResourceMock->expects(static::never())->method('save');
        $this->observer->execute($this->eventObserverMock);
    }

    /**
     * @return string
     * @throws \Exception
     */
    private function getExpiresDateTime(): string
    {
        $testDate = new \DateTime();
        $testDate->modify('+10 days');
        return $testDate->format('Y-m-d H:i:s');
    }
}