From d60dd8a20855440625feca3758968b784b5fca9d Mon Sep 17 00:00:00 2001
From: Konrad Abicht <hi@inspirito.de>
Date: Thu, 11 Feb 2021 09:56:09 +0100
Subject: [PATCH] added unit tests for LoginFlowV2Service::startLoginFlow

Signed-off-by: Konrad Abicht <hi@inspirito.de>
---
 .../Service/LoginFlowV2ServiceUnitTest.php    | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/tests/Core/Service/LoginFlowV2ServiceUnitTest.php b/tests/Core/Service/LoginFlowV2ServiceUnitTest.php
index 511c9be3271..58c558aada4 100644
--- a/tests/Core/Service/LoginFlowV2ServiceUnitTest.php
+++ b/tests/Core/Service/LoginFlowV2ServiceUnitTest.php
@@ -21,6 +21,7 @@
 
 namespace Tests\Core\Data;
 
+use Exception;
 use OC\Core\Service\LoginFlowV2Service;
 use OC\Core\Db\LoginFlowV2Mapper;
 use OC\Core\Exception\LoginFlowV2NotFoundException;
@@ -244,4 +245,43 @@ class LoginFlowV2ServiceUnitTest extends TestCase {
 
 		$this->subjectUnderTest->getByLoginToken('test_token');
 	}
+
+	/*
+	 * Tests for startLoginFlow
+	 */
+
+	public function testStartLoginFlow() {
+		$loginFlowV2 = new LoginFlowV2();
+
+		$this->mapper->expects($this->once())
+			->method('getByLoginToken')
+			->willReturn($loginFlowV2);
+
+		$this->mapper->expects($this->once())
+			->method('update');
+
+		$this->assertTrue($this->subjectUnderTest->startLoginFlow('test_token'));
+	}
+
+	public function testStartLoginFlowDoesNotExistException() {
+		$this->mapper->expects($this->once())
+			->method('getByLoginToken')
+			->willThrowException(new DoesNotExistException(''));
+
+		$this->assertFalse($this->subjectUnderTest->startLoginFlow('test_token'));
+	}
+
+	/**
+	 * If an exception not of type DoesNotExistException is thrown,
+	 * it is expected that it is not being handled by startLoginFlow.
+	 */
+	public function testStartLoginFlowException() {
+		$this->expectException(Exception::class);
+
+		$this->mapper->expects($this->once())
+			->method('getByLoginToken')
+			->willThrowException(new Exception(''));
+
+		$this->subjectUnderTest->startLoginFlow('test_token');
+	}
 }