MenuLinkTreeTest.php

<?php

namespace Drupal\KernelTests\Core\Menu;

use Drupal\Core\Menu\MenuLinkTreeElement;
use Drupal\Core\Menu\MenuTreeParameters;
use Drupal\KernelTests\KernelTestBase;
use Drupal\Tests\Core\Menu\MenuLinkMock;

/**
 * Tests the menu link tree.
 *
 * @group Menu
 *
 * @see \Drupal\Core\Menu\MenuLinkTree
 */
class MenuLinkTreeTest extends KernelTestBase {
  
  /**
   * The tested menu link tree.
   *
   * @var \Drupal\Core\Menu\MenuLinkTree
   */
  protected $linkTree;
  
  /**
   * The menu link plugin manager.
   *
   * @var \Drupal\Core\Menu\MenuLinkManagerInterface
   */
  protected $menuLinkManager;
  
  /**
   * Modules to enable.
   *
   * @var array
   */
  public static $modules = [
    'system',
    'menu_test',
    'menu_link_content',
    'field',
    'link',
    'user',
  ];
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    \Drupal::service('router.builder')->rebuild();
    $this->installEntitySchema('user');
    $this->installEntitySchema('menu_link_content');
    $this->linkTree = $this->container
      ->get('menu.link_tree');
    $this->menuLinkManager = $this->container
      ->get('plugin.manager.menu.link');
  }
  
  /**
   * Tests deleting all the links in a menu.
   */
  public function testDeleteLinksInMenu() {
    \Drupal::entityTypeManager()->getStorage('menu')
      ->create([
      'id' => 'menu1',
    ])
      ->save();
    \Drupal::entityTypeManager()->getStorage('menu')
      ->create([
      'id' => 'menu2',
    ])
      ->save();
    \Drupal::entityTypeManager()->getStorage('menu_link_content')
      ->create([
      'link' => [
        'uri' => 'internal:/menu_name_test',
      ],
      'menu_name' => 'menu1',
      'bundle' => 'menu_link_content',
      'title' => 'Link test',
    ])
      ->save();
    \Drupal::entityTypeManager()->getStorage('menu_link_content')
      ->create([
      'link' => [
        'uri' => 'internal:/menu_name_test',
      ],
      'menu_name' => 'menu1',
      'bundle' => 'menu_link_content',
      'title' => 'Link test',
    ])
      ->save();
    \Drupal::entityTypeManager()->getStorage('menu_link_content')
      ->create([
      'link' => [
        'uri' => 'internal:/menu_name_test',
      ],
      'menu_name' => 'menu2',
      'bundle' => 'menu_link_content',
      'title' => 'Link test',
    ])
      ->save();
    $output = $this->linkTree
      ->load('menu1', new MenuTreeParameters());
    $this->assertCount(2, $output);
    $output = $this->linkTree
      ->load('menu2', new MenuTreeParameters());
    $this->assertCount(1, $output);
    $this->menuLinkManager
      ->deleteLinksInMenu('menu1');
    $output = $this->linkTree
      ->load('menu1', new MenuTreeParameters());
    $this->assertCount(0, $output);
    $output = $this->linkTree
      ->load('menu2', new MenuTreeParameters());
    $this->assertCount(1, $output);
  }
  
  /**
   * Tests creating links with an expected tree structure.
   */
  public function testCreateLinksInMenu() {
    // This creates a tree with the following structure:
    // - 1
    // - 2
    //   - 3
    //     - 4
    // - 5
    //   - 7
    // - 6
    // - 8
    // With link 6 being the only external link.
    $links = [
      1 => MenuLinkMock::create([
        'id' => 'test.example1',
        'route_name' => 'example1',
        'title' => 'foo',
        'parent' => '',
      ]),
      2 => MenuLinkMock::create([
        'id' => 'test.example2',
        'route_name' => 'example2',
        'title' => 'bar',
        'parent' => 'test.example1',
        'route_parameters' => [
          'foo' => 'bar',
        ],
      ]),
      3 => MenuLinkMock::create([
        'id' => 'test.example3',
        'route_name' => 'example3',
        'title' => 'baz',
        'parent' => 'test.example2',
        'route_parameters' => [
          'baz' => 'qux',
        ],
      ]),
      4 => MenuLinkMock::create([
        'id' => 'test.example4',
        'route_name' => 'example4',
        'title' => 'qux',
        'parent' => 'test.example3',
      ]),
      5 => MenuLinkMock::create([
        'id' => 'test.example5',
        'route_name' => 'example5',
        'title' => 'foofoo',
        'parent' => '',
      ]),
      6 => MenuLinkMock::create([
        'id' => 'test.example6',
        'route_name' => '',
        'url' => 'https://www.drupal.org/',
        'title' => 'barbar',
        'parent' => '',
      ]),
      7 => MenuLinkMock::create([
        'id' => 'test.example7',
        'route_name' => 'example7',
        'title' => 'bazbaz',
        'parent' => '',
      ]),
      8 => MenuLinkMock::create([
        'id' => 'test.example8',
        'route_name' => 'example8',
        'title' => 'quxqux',
        'parent' => '',
      ]),
    ];
    foreach ($links as $instance) {
      $this->menuLinkManager
        ->addDefinition($instance->getPluginId(), $instance->getPluginDefinition());
    }
    $parameters = new MenuTreeParameters();
    $tree = $this->linkTree
      ->load('mock', $parameters);
    $count = function (array $tree) {
      $sum = function ($carry, MenuLinkTreeElement $item) {
        return $carry + $item->count();
      };
      return array_reduce($tree, $sum);
    };
    $this->assertEqual($count($tree), 8);
    $parameters = new MenuTreeParameters();
    $parameters->setRoot('test.example2');
    $tree = $this->linkTree
      ->load($instance->getMenuName(), $parameters);
    $top_link = reset($tree);
    $this->assertCount(1, $top_link->subtree);
    $child = reset($top_link->subtree);
    $this->assertEqual($child->link
      ->getPluginId(), $links[3]->getPluginId());
    $height = $this->linkTree
      ->getSubtreeHeight('test.example2');
    $this->assertEqual($height, 3);
  }

}