Fix AppPassword 2FA auth

Signed-off-by: Roeland Jago Douma <roeland@famdouma.nl>

1 files changed, 16 insertions, 1 deletions

diff --git a/tests/lib/Authentication/TwoFactorAuth/ManagerTest.php b/tests/lib/Authentication/TwoFactorAuth/ManagerTest.php
index 4fa3b3d7e14..9db27edd70c 100644
--- a/tests/lib/Authentication/TwoFactorAuth/ManagerTest.php
+++ b/tests/lib/Authentication/TwoFactorAuth/ManagerTest.php
@@ -388,9 +388,13 @@ class ManagerTest extends TestCase {
 		$user = $this->createMock(IUser::class);
 		$this->session->expects($this->at(0))
 			->method('exists')
+			->with('app_password')
+			->willReturn(false);
+		$this->session->expects($this->at(1))
+			->method('exists')
 			->with('two_factor_auth_uid')
 			->will($this->returnValue(false));
-		$this->session->expects($this->at(1))
+		$this->session->expects($this->at(2))
 			->method('exists')
 			->with(Manager::SESSION_UID_DONE)
 			->willReturn(false);
@@ -523,6 +527,8 @@ class ManagerTest extends TestCase {
 			->will($this->returnCallback(function($var) {
 				if ($var === Manager::SESSION_UID_KEY) {
 					return false;
+				} else if ($var === 'app_password') {
+					return false;
 				}
 				return true;
 			}));
@@ -585,4 +591,13 @@ class ManagerTest extends TestCase {
 
 		$this->assertFalse($this->manager->needsSecondFactor($user));
 	}
+
+	public function testNeedsSecondFactorAppPassword() {
+		$user = $this->createMock(IUser::class);
+		$this->session->method('exists')
+			->with('app_password')
+			->willReturn(true);
+
+		$this->assertFalse($this->manager->needsSecondFactor($user));
+	}
 }