Improve AutomaticLogoutListenerTest

tests/Authentication/AutomaticLogoutListenerTest.phpt

@@ -28,6 +28,23 @@ final class AutomaticLogoutListenerTest extends \Tester\TestCase
 		$listener->check();
 
 		Assert::false($userStorage->isAuthenticated());
+		Assert::same($identity, $userStorage->getIdentity());
+	}
+
+	public function testCheckNotFoundEntity(): void
+	{
+		$userStorage = new UserStorage();
+		$identity = new Identity('foo', 'foo', '', false);
+		$userStorage->setAuthenticated(true);
+		$userStorage->setIdentity($identity);
+		$listener = new AutomaticLogoutListener(new AuthenticationManager($userStorage));
+
+		$identity->deactivate();
+		$listener->check();
+
+		Assert::false($userStorage->isAuthenticated());
+		Assert::type('null', $userStorage->getIdentity());
+		//
 	}
 
 }

tests/Authentication/Mocks/Identity.php

@@ -2,17 +2,29 @@
 
 namespace GrandMediaTests\SecurityDoctrine\Authentication\Mocks;
 
+use Doctrine\ORM\EntityNotFoundException;
+
 final class Identity extends \GrandMedia\SecurityDoctrine\Authentication\Identity
 {
 
 	/** @var string */
 	private $name;
 
-	public function __construct(string $name, string $password, string $role = '')
+	/** @var bool */
+	private $entityFound;
+
+	public function __construct(
+		string $name,
+		string $password,
+		string $role = '',
+		bool $entityFound = true
+	)
 	{
 		$this->name = $name;
 
 		parent::__construct($password, $role);
+
+		$this->entityFound = $entityFound;
 	}
 
 	public function getId(): string
@@ -22,7 +34,37 @@ public function getId(): string
 
 	public function getName(): string
 	{
+		$this->loadProxy();
+
 		return $this->name;
 	}
 
+	public function verify(string $password): bool
+	{
+		$this->loadProxy();
+
+		return parent::verify($password);
+	}
+
+	public function isActive(): bool
+	{
+		$this->loadProxy();
+
+		return parent::isActive();
+	}
+
+	public function getRoles(): array
+	{
+		$this->loadProxy();
+
+		return parent::getRoles();
+	}
+
+	private function loadProxy(): void
+	{
+		if (!$this->entityFound) {
+			throw new EntityNotFoundException();
+		}
+	}
+
 }