magento2/app/code/Magento/AdminNotification/Test/Unit/Model/NotificationServiceTest.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

/**
 * Test class for \Magento\AdminNotification\Model\NotificationService
 */
namespace Magento\AdminNotification\Test\Unit\Model;

use Magento\AdminNotification\Model\Inbox;
use Magento\AdminNotification\Model\InboxFactory;
use Magento\AdminNotification\Model\NotificationService;
use Magento\Framework\Exception\LocalizedException;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;

class NotificationServiceTest extends TestCase
{
    /**
     * Retrieve instance of notification service model
     *
     * @param $notificationId
     * @return NotificationService
     */
    protected function _getServiceInstanceForMarkAsReadTest($notificationId)
    {
        /**
         * @var MockObject|InboxFactory $notificationFactory
         */
        $notificationFactory = $this->createPartialMock(
            InboxFactory::class,
            ['create']
        );
        $notification = $this->getMockBuilder(Inbox::class)
            ->addMethods(['setIsRead'])
            ->onlyMethods(['load', 'getId', 'save', 'setData', '__sleep', '__wakeup'])
            ->disableOriginalConstructor()
            ->getMock();
        $notification->expects($this->once())->method('load')->with($notificationId)->willReturnSelf();
        $notification->expects($this->once())->method('getId')->willReturn($notificationId);

        // when notification Id is valid, add additional expectations
        if ($notificationId) {
            $notification->expects($this->once())->method('save')->willReturnSelf();
            $notification->expects($this->once())->method('setIsRead')->with(1)->willReturnSelf();
        }

        $notificationFactory->expects($this->once())->method('create')->willReturn($notification);
        return new NotificationService($notificationFactory);
    }

    public function testMarkAsRead()
    {
        $notificationId = 1;
        $service = $this->_getServiceInstanceForMarkAsReadTest($notificationId);
        $service->markAsRead($notificationId);
    }

    public function testMarkAsReadThrowsExceptionWhenNotificationIdIsInvalid()
    {
        $this->expectException(LocalizedException::class);
        $this->expectExceptionMessage('Wrong notification ID specified.');

        $notificationId = null;
        $service = $this->_getServiceInstanceForMarkAsReadTest($notificationId);
        $service->markAsRead($notificationId);
    }
}