app/bundles/UserBundle/Tests/Security/UserTokenSetterTest.php

<?php

declare(strict_types=1);

namespace Mautic\UserBundle\Tests\Security;

use Mautic\CoreBundle\Helper\UserHelper;
use Mautic\CoreBundle\Test\AbstractMauticTestCase;
use Mautic\UserBundle\Entity\User;
use Mautic\UserBundle\Entity\UserRepository;
use Mautic\UserBundle\Security\UserTokenSetter;
use PHPUnit\Framework\MockObject\MockObject;

class UserTokenSetterTest extends AbstractMauticTestCase
{
    public function testSetUserMakesTheUserAvailableToUserHelper(): void
    {
        /** @var MockObject|UserRepository $userRepo */
        $userRepo = $this->createMock(UserRepository::class);
        $user     = new User();

        $userRepo->method('getEntity')
            ->with(1)
            ->willReturn($user);

        $userTokenSetter = new UserTokenSetter($userRepo, $this->getContainer()->get('security.token_storage'));

        $userTokenSetter->setUser(1);

        /** @var UserHelper $userHelper */
        $userHelper = $this->getContainer()->get('mautic.helper.user');

        $this->assertSame($user, $userHelper->getUser());
    }
}