EndpointControllerTest.php

<?php

declare(strict_types=1);
/**
 * SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
 * SPDX-FileCopyrightText: 2016 ownCloud, Inc.
 * SPDX-License-Identifier: AGPL-3.0-only
 */

namespace OCA\Notifications\Tests\Unit\Controller;

use OCA\Notifications\Controller\EndpointController;
use OCA\Notifications\Exceptions\NotificationNotFoundException;
use OCA\Notifications\Handler;
use OCA\Notifications\Push;
use OCA\Notifications\Service\ClientService;
use OCP\AppFramework\Http;
use OCP\AppFramework\Http\DataResponse;
use OCP\AppFramework\Utility\ITimeFactory;
use OCP\IRequest;
use OCP\IUser;
use OCP\IUserSession;
use OCP\L10N\IFactory;
use OCP\Notification\IAction;
use OCP\Notification\IManager;
use OCP\Notification\INotification;
use OCP\UserStatus\IManager as IUserStatusManager;
use PHPUnit\Framework\Attributes\DataProvider;
use PHPUnit\Framework\MockObject\MockObject;
use Test\TestCase;

class EndpointControllerTest extends TestCase {
	protected IRequest&MockObject $request;
	protected Handler&MockObject $handler;
	protected IManager&MockObject $manager;
	protected IFactory&MockObject $l10nFactory;
	protected IUserSession&MockObject $session;
	protected IUserStatusManager&MockObject $userStatusManager;
	protected IUser&MockObject $user;
	protected ITimeFactory&MockObject $timeFactory;
	protected ClientService&MockObject $clientService;
	protected Push&MockObject $push;
	protected EndpointController $controller;

	protected function setUp(): void {
		parent::setUp();

		$this->request = $this->createMock(IRequest::class);
		$this->handler = $this->createMock(Handler::class);
		$this->manager = $this->createMock(IManager::class);
		$this->l10nFactory = $this->createMock(IFactory::class);
		$this->session = $this->createMock(IUserSession::class);
		$this->userStatusManager = $this->createMock(IUserStatusManager::class);
		$this->user = $this->createMock(IUser::class);
		$this->timeFactory = $this->createMock(ITimeFactory::class);
		$this->clientService = $this->createMock(ClientService::class);
		$this->push = $this->createMock(Push::class);

		$this->session->expects($this->any())
			->method('getUser')
			->willReturn($this->user);
	}

	protected function getController(array $methods = [], $username = 'username') {
		$this->user->expects($this->any())
			->method('getUID')
			->willReturn($username);

		if (empty($methods)) {
			return new EndpointController(
				'notifications',
				$this->request,
				$this->handler,
				$this->manager,
				$this->l10nFactory,
				$this->session,
				$this->timeFactory,
				$this->userStatusManager,
				$this->clientService,
				$this->push,
			);
		}

		return $this->getMockBuilder(EndpointController::class)
			->setConstructorArgs([
				'notifications',
				$this->request,
				$this->handler,
				$this->manager,
				$this->l10nFactory,
				$this->session,
				$this->timeFactory,
				$this->userStatusManager,
				$this->clientService,
				$this->push,
			])
			->onlyMethods($methods)
			->getMock();
	}

	public static function dataListNotifications(): array {
		return [
			[
				'v2',
				[],
				'"' . md5(json_encode([])) . '"',
				[],
			],
			[
				'v2',
				[
					1,
					3,
				],
				'"' . md5(json_encode([1, 3])) . '"',
				[['$notification'], ['$notification']],
			],
			[
				'v2',
				[
					42,
				],
				'"' . md5(json_encode([42])) . '"',
				[['$notification']],
			],
		];
	}

	#[DataProvider(methodName: 'dataListNotifications')]
	public function testListNotifications(string $apiVersion, array $notifications, string $expectedETag, array $expectedData): void {
		$notifications = array_fill_keys($notifications, $this->createMock(INotification::class));

		$controller = $this->getController([
			'notificationToArray',
		]);
		$controller->expects($this->exactly(\count($notifications)))
			->method('notificationToArray')
			->willReturn(['$notification']);

		$filter = $this->getMockBuilder(INotification::class)
			->getMock();
		$filter->expects($this->once())
			->method('setUser')
			->willReturn($filter);

		$this->manager->expects($this->once())
			->method('hasNotifiers')
			->willReturn(true);
		$this->manager->expects($this->once())
			->method('createNotification')
			->willReturn($filter);
		$this->manager->expects($this->exactly(\count($notifications)))
			->method('prepare')
			->willReturnArgument(0);

		$this->l10nFactory
			->method('getUserLanguage')
			->with($this->user)
			->willReturn('en');

		$this->handler->expects($this->once())
			->method('get')
			->with($filter)
			->willReturn($notifications);

		$response = $controller->listNotifications($apiVersion);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_OK, $response->getStatus());

		$headers = $response->getHeaders();
		$this->assertArrayHasKey('ETag', $headers);
		$this->assertSame($expectedETag, $headers['ETag']);
		$this->assertSame($expectedData, $response->getData());
	}

	public static function dataListNotificationsThrows(): array {
		return [
			[
				'v2',
				[
					1,
					3,
				],
				'"' . md5(json_encode([3])) . '"',
				[['$notification']],
			],
		];
	}

	#[DataProvider(methodName: 'dataListNotificationsThrows')]
	public function testListNotificationsThrows(string $apiVersion, array $notifications, string $expectedETag, array $expectedData): void {
		$notifications = array_fill_keys($notifications, $this->createMock(INotification::class));

		$controller = $this->getController([
			'notificationToArray',
		]);
		$controller->expects($this->exactly(1))
			->method('notificationToArray')
			->willReturn(['$notification']);

		$filter = $this->getMockBuilder(INotification::class)
			->getMock();
		$filter->expects($this->once())
			->method('setUser')
			->willReturn($filter);

		$this->manager->expects($this->once())
			->method('hasNotifiers')
			->willReturn(true);
		$this->manager->expects($this->once())
			->method('createNotification')
			->willReturn($filter);
		$this->manager->expects($this->once())
			->method('defer')
			->willReturn(true);
		$this->manager->expects($this->once())
			->method('flush');

		$throw = true;
		$this->manager->expects($this->exactly(2))
			->method('prepare')
			->willReturnCallback(function ($arg) use (&$throw) {
				if ($throw) {
					$throw = false;
					throw new \InvalidArgumentException();
				}
				return $arg;
			});

		$this->l10nFactory
			->method('getUserLanguage')
			->with($this->user)
			->willReturn('en');

		$this->handler->expects($this->once())
			->method('get')
			->with($filter)
			->willReturn($notifications);

		$response = $controller->listNotifications($apiVersion);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_OK, $response->getStatus());

		$headers = $response->getHeaders();
		$this->assertArrayHasKey('ETag', $headers);
		$this->assertSame($expectedETag, $headers['ETag']);
		$this->assertSame($expectedData, $response->getData());
	}

	public static function dataListNotificationsNoNotifiers(): array {
		return [
			['v1'],
			['v2'],
		];
	}

	#[DataProvider(methodName: 'dataListNotificationsNoNotifiers')]
	public function testListNotificationsNoNotifiers(string $apiVersion): void {
		$controller = $this->getController();
		$this->manager->expects($this->once())
			->method('hasNotifiers')
			->willReturn(false);

		$response = $controller->listNotifications($apiVersion);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_NO_CONTENT, $response->getStatus());
	}

	public static function dataGetNotification(): array {
		return [
			['v1', 42, 'username1'],
			['v2', 21, 'username2'],
		];
	}

	#[DataProvider(methodName: 'dataGetNotification')]
	public function testGetNotification(string $apiVersion, int $id, string $username): void {
		$controller = $this->getController([
			'notificationToArray',
		], $username);

		$notification = $this->getMockBuilder(INotification::class)
			->getMock();

		$this->manager->expects($this->once())
			->method('hasNotifiers')
			->willReturn(true);
		$this->manager->expects($this->once())
			->method('prepare')
			->with($notification)
			->willReturn($notification);

		$this->handler->expects($this->once())
			->method('getById')
			->with($id, $username)
			->willReturn($notification);

		$controller->expects($this->exactly(1))
			->method('notificationToArray')
			->with($id, $notification)
			->willReturn(['$notification']);

		$this->l10nFactory
			->method('getUserLanguage')
			->with($this->user)
			->willReturn('en');

		$response = $controller->getNotification($apiVersion, $id);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_OK, $response->getStatus());
	}

	public static function dataGetNotificationNoId(): array {
		return [
			['v1', false, 42, false, new NotificationNotFoundException()], // No notifiers
			['v1', true, 42, true, new NotificationNotFoundException()], // Not found in database
			['v1', true, 42, true, null], // Not handled on prepare
			['v2', true, 42, true, null], // Not handled on prepare
		];
	}

	#[DataProvider(methodName: 'dataGetNotificationNoId')]
	public function testGetNotificationNoId(string $apiVersion, bool $hasNotifiers, int $id, bool $called, ?NotificationNotFoundException $notification): void {
		if ($notification === null) {
			$notification = $this->createMock(INotification::class);
		}
		$controller = $this->getController();

		$this->manager->expects($this->once())
			->method('hasNotifiers')
			->willReturn($hasNotifiers);

		if ($notification instanceof NotificationNotFoundException) {
			$this->handler->expects($called ? $this->once() : $this->never())
				->method('getById')
				->willThrowException($notification);

			$this->manager->expects($called && !$notification instanceof NotificationNotFoundException ? $this->once() : $this->never())
				->method('prepare')
				->willThrowException(new \InvalidArgumentException());
		} else {
			$this->handler->expects($this->once())
				->method('getById')
				->willReturn($notification);

			$this->l10nFactory
				->method('getUserLanguage')
				->with($this->user)
				->willReturn('en');

			$this->manager->expects($this->once())
				->method('prepare')
				->willThrowException(new \InvalidArgumentException());
		}

		$response = $controller->getNotification($apiVersion, $id);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_NOT_FOUND, $response->getStatus());
	}

	public static function dataDeleteNotification(): array {
		return [
			[42, 'username1'],
			[21, 'username2'],
		];
	}

	#[DataProvider(methodName: 'dataDeleteNotification')]
	public function testDeleteNotification(int $id, string $username): void {
		$controller = $this->getController([], $username);

		$this->handler->expects($this->once())
			->method('deleteById')
			->with($id, $username);

		$response = $controller->deleteNotification($id);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_OK, $response->getStatus());
	}

	public function testDeleteNotificationNoId(): void {
		$controller = $this->getController();

		$this->handler->expects($this->never())
			->method('deleteById');

		$response = $controller->deleteNotification(0);
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_NOT_FOUND, $response->getStatus());
	}

	#[DataProvider(methodName: 'dataDeleteNotification')]
	public function testDeleteAllNotifications(int $_, string $username): void {
		$controller = $this->getController([], $username);

		$this->handler->expects($this->once())
			->method('deleteByUser')
			->with($username);
		$this->manager->expects($this->once())
			->method('defer')
			->willReturn(true);
		$this->manager->expects($this->once())
			->method('flush');

		$response = $controller->deleteAllNotifications();
		$this->assertInstanceOf(DataResponse::class, $response);
		$this->assertSame(Http::STATUS_OK, $response->getStatus());
	}

	public static function dataNotificationToArray(): array {
		return [
			['v1', 42, 'app1', 'user1', 1234, 'type1', '42', 'subject1', '', [], 'message1', 'richMessage 1', ['richMessage param'], 'link1', 'icon1', 0, []],
			['v1', 1337, 'app2', 'user2', 1337, 'type2', '21', 'subject2', 'richSubject 2', ['richSubject param'], 'message2', '', [], 'link2', 'icon2', 2, [['action'], ['action']]],
			['v2', 42, 'app1', 'user1', 1234, 'type1', '42', 'subject1', '', [], 'message1', 'richMessage 1', ['richMessage param'], 'link1', 'icon1', 0, []],
			['v2', 1337, 'app2', 'user2', 1337, 'type2', '21', 'subject2', 'richSubject 2', ['richSubject param'], 'message2', '', [], 'link2', 'icon2', 2, [['action'], ['action']]],
		];
	}

	#[DataProvider(methodName: 'dataNotificationToArray')]
	public function testNotificationToArray(string $apiVersion, int $id, string $app, string $user, int $timestamp, string $objectType, string $objectId, string $subject, string $subjectRich, array $subjectRichParameters, string $message, string $messageRich, array $messageRichParameters, string $link, string $icon, int $actionsCount, array $actionsExpected): void {
		$actions = [];
		for ($i = 0; $i < $actionsCount; $i++) {
			$actions[] = $this->createMock(IAction::class);
		}

		$notification = $this->getMockBuilder(INotification::class)
			->getMock();

		$notification->expects($this->once())
			->method('getApp')
			->willReturn($app);

		$notification->expects($this->once())
			->method('getUser')
			->willReturn($user);

		$dateTime = new \DateTime();
		$dateTime->setTimestamp($timestamp);
		$notification->expects($this->once())
			->method('getDateTime')
			->willReturn($dateTime);

		$notification->expects($this->once())
			->method('getObjectType')
			->willReturn($objectType);

		$notification->expects($this->once())
			->method('getObjectId')
			->willReturn($objectId);

		$notification->expects($this->once())
			->method('getParsedSubject')
			->willReturn($subject);

		$notification->expects($apiVersion === 'v1' ? $this->never() : $this->once())
			->method('getRichSubject')
			->willReturn($subjectRich);

		$notification->expects($apiVersion === 'v1' ? $this->never() : $this->once())
			->method('getRichSubjectParameters')
			->willReturn($subjectRichParameters);

		$notification->expects($this->once())
			->method('getParsedMessage')
			->willReturn($message);

		$notification->expects($apiVersion === 'v1' ? $this->never() : $this->once())
			->method('getRichMessage')
			->willReturn($messageRich);

		$notification->expects($apiVersion === 'v1' ? $this->never() : $this->once())
			->method('getRichMessageParameters')
			->willReturn($messageRichParameters);

		$notification->expects($this->once())
			->method('getLink')
			->willReturn($link);

		$notification->expects($apiVersion === 'v1' ? $this->never() : $this->once())
			->method('getIcon')
			->willReturn($icon);

		$notification->expects($this->once())
			->method('getParsedActions')
			->willReturn($actions);

		$controller = $this->getController([
			'actionToArray'
		]);
		$controller->expects($this->exactly(\count($actions)))
			->method('actionToArray')
			->willReturn(['action']);

		$expected = [
			'notification_id' => $id,
			'app' => $app,
			'user' => $user,
			'datetime' => date('c', $timestamp),
			'object_type' => $objectType,
			'object_id' => $objectId,
			'subject' => $subject,
			'message' => $message,
			'link' => $link,
			'actions' => $actionsExpected,
		];

		if ($apiVersion !== 'v1') {
			$expected = array_merge($expected, [
				'subjectRich' => $subjectRich,
				'subjectRichParameters' => $subjectRichParameters,
				'messageRich' => $messageRich,
				'messageRichParameters' => $messageRichParameters,
				'icon' => $icon,
				'shouldNotify' => true,
			]);
		}

		$this->assertEquals($expected, $this->invokePrivate($controller, 'notificationToArray', [$id, $notification, $apiVersion])
		);
	}

	public static function dataActionToArray(): array {
		return [
			['label1', 'link1', 'GET', false],
			['label2', 'link2', 'POST', true],
		];
	}

	#[DataProvider(methodName: 'dataActionToArray')]
	public function testActionToArray(string $label, string $link, string $requestType, bool $isPrimary): void {
		$action = $this->getMockBuilder(IAction::class)
			->getMock();

		$action->expects($this->once())
			->method('getParsedLabel')
			->willReturn($label);

		$action->expects($this->once())
			->method('getLink')
			->willReturn($link);

		$action->expects($this->once())
			->method('getRequestType')
			->willReturn($requestType);

		$action->expects($this->once())
			->method('isPrimary')
			->willReturn($isPrimary);

		$this->assertEquals([
			'label' => $label,
			'link' => $link,
			'type' => $requestType,
			'primary' => $isPrimary,
		],
			$this->invokePrivate($this->getController(), 'actionToArray', [$action])
		);
	}
}