MenuLinkTreeTest.php in Drupal 8

Namespace

Drupal\KernelTests\Core\Menu

File

core/tests/Drupal/KernelTests/Core/Menu/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);
  }

}

Classes

Name	Description
MenuLinkTreeTest	Tests the menu link tree.