1 files changed, 144 insertions, 0 deletions
diff --git a/tests/lib/Authentication/TwoFactorAuth/RegistryTest.php b/tests/lib/Authentication/TwoFactorAuth/RegistryTest.php
new file mode 100644
index 00000000000..2018dc1a634
--- /dev/null
+++ b/tests/lib/Authentication/TwoFactorAuth/RegistryTest.php
@@ -0,0 +1,144 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2018 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace Test\Authentication\TwoFactorAuth;
+
+use OC\Authentication\TwoFactorAuth\Db\ProviderUserAssignmentDao;
+use OC\Authentication\TwoFactorAuth\Registry;
+use OCP\Authentication\TwoFactorAuth\IProvider;
+use OCP\Authentication\TwoFactorAuth\IRegistry;
+use OCP\Authentication\TwoFactorAuth\RegistryEvent;
+use OCP\Authentication\TwoFactorAuth\TwoFactorProviderDisabled;
+use OCP\Authentication\TwoFactorAuth\TwoFactorProviderForUserRegistered;
+use OCP\Authentication\TwoFactorAuth\TwoFactorProviderForUserUnregistered;
+use OCP\Authentication\TwoFactorAuth\TwoFactorProviderUserDeleted;
+use OCP\EventDispatcher\IEventDispatcher;
+use OCP\IUser;
+use PHPUnit\Framework\MockObject\MockObject;
+use Test\TestCase;
+
+class RegistryTest extends TestCase {
+	/** @var ProviderUserAssignmentDao|MockObject */
+	private $dao;
+
+	/** @var IEventDispatcher|MockObject */
+	private $dispatcher;
+
+	/** @var Registry */
+	private $registry;
+
+	protected function setUp(): void {
+		parent::setUp();
+
+		$this->dao = $this->createMock(ProviderUserAssignmentDao::class);
+		$this->dispatcher = $this->createMock(IEventDispatcher::class);
+
+		$this->registry = new Registry($this->dao, $this->dispatcher);
+	}
+
+	public function testGetProviderStates(): void {
+		$user = $this->createMock(IUser::class);
+		$user->expects($this->once())->method('getUID')->willReturn('user123');
+		$state = [
+			'twofactor_totp' => true,
+		];
+		$this->dao->expects($this->once())->method('getState')->willReturn($state);
+
+		$actual = $this->registry->getProviderStates($user);
+
+		$this->assertEquals($state, $actual);
+	}
+
+	public function testEnableProvider(): void {
+		$user = $this->createMock(IUser::class);
+		$provider = $this->createMock(IProvider::class);
+		$user->expects($this->once())->method('getUID')->willReturn('user123');
+		$provider->expects($this->once())->method('getId')->willReturn('p1');
+		$this->dao->expects($this->once())->method('persist')->with('p1', 'user123',
+			true);
+
+		$this->dispatcher->expects($this->once())
+			->method('dispatch')
+			->with(
+				$this->equalTo(IRegistry::EVENT_PROVIDER_ENABLED),
+				$this->callback(function (RegistryEvent $e) use ($user, $provider) {
+					return $e->getUser() === $user && $e->getProvider() === $provider;
+				})
+			);
+		$this->dispatcher->expects($this->once())
+			->method('dispatchTyped')
+			->with(new TwoFactorProviderForUserRegistered(
+				$user,
+				$provider,
+			));
+
+		$this->registry->enableProviderFor($provider, $user);
+	}
+
+	public function testDisableProvider(): void {
+		$user = $this->createMock(IUser::class);
+		$provider = $this->createMock(IProvider::class);
+		$user->expects($this->once())->method('getUID')->willReturn('user123');
+		$provider->expects($this->once())->method('getId')->willReturn('p1');
+		$this->dao->expects($this->once())->method('persist')->with('p1', 'user123',
+			false);
+
+
+		$this->dispatcher->expects($this->once())
+			->method('dispatch')
+			->with(
+				$this->equalTo(IRegistry::EVENT_PROVIDER_DISABLED),
+				$this->callback(function (RegistryEvent $e) use ($user, $provider) {
+					return $e->getUser() === $user && $e->getProvider() === $provider;
+				})
+			);
+		$this->dispatcher->expects($this->once())
+			->method('dispatchTyped')
+			->with(new TwoFactorProviderForUserUnregistered(
+				$user,
+				$provider,
+			));
+
+		$this->registry->disableProviderFor($provider, $user);
+	}
+
+	public function testDeleteUserData(): void {
+		$user = $this->createMock(IUser::class);
+		$user->expects($this->once())->method('getUID')->willReturn('user123');
+		$this->dao->expects($this->once())
+			->method('deleteByUser')
+			->with('user123')
+			->willReturn([
+				[
+					'provider_id' => 'twofactor_u2f',
+				]
+			]);
+
+		$calls = [
+			[new TwoFactorProviderDisabled('twofactor_u2f')],
+			[new TwoFactorProviderUserDeleted($user, 'twofactor_u2f')],
+		];
+		$this->dispatcher->expects($this->exactly(2))
+			->method('dispatchTyped')
+			->willReturnCallback(function () use (&$calls): void {
+				$expected = array_shift($calls);
+				$this->assertEquals($expected, func_get_args());
+			});
+
+		$this->registry->deleteUserData($user);
+	}
+
+	public function testCleanUp(): void {
+		$this->dao->expects($this->once())
+			->method('deleteAll')
+			->with('twofactor_u2f');
+
+		$this->registry->cleanUp('twofactor_u2f');
+	}
+}