MenuBuilderTest.php

Go to the documentation of this file.

<?php
namespace Magento\Backend\Test\Unit\Model;

use Magento\Backend\Model\Setup\MenuBuilder;

class MenuBuilderTest extends \PHPUnit\Framework\TestCase
{
    public function testAfterGetResult($isPub, $times)
    {
        $docRootLocator = $this->createMock(\Magento\Framework\App\DocRootLocator::class);
        $docRootLocator->expects($this->once())->method('isPub')->willReturn($isPub);
        $model = new MenuBuilder($docRootLocator);
        $menu = $this->createMock(\Magento\Backend\Model\Menu::class);
        $menu->expects($this->exactly($times))->method('remove')->willReturn(true);

        $menuBuilder = $this->createMock(\Magento\Backend\Model\Menu\Builder::class);

        $this->assertInstanceOf(
            \Magento\Backend\Model\Menu::class,
            $model->afterGetResult($menuBuilder, $menu)
        );
    }

    public function afterGetResultDataProvider()
    {
        return [[true, 1], [false, 0],];
    }
}