trait MenuUiTrait in Drupal 9

trait MenuUiTrait {

  /**
   * Asserts that a menu fetched from the database matches an expected one.
   *
   * @param array $expected_item
   *   Array containing properties to check.
   * @param int $menu_plugin_id
   *   Menu item id.
   */
  protected function assertMenuLink(array $expected_item, $menu_plugin_id) {

    // Retrieve the menu link.

    /** @var \Drupal\Core\Menu\MenuLinkManagerInterface $menu_link_manager */
    $menu_link_manager = \Drupal::service('plugin.manager.menu.link');
    $menu_link_manager
      ->resetDefinitions();

    // Reset the static load cache.
    \Drupal::entityTypeManager()
      ->getStorage('menu_link_content')
      ->resetCache();
    $definition = $menu_link_manager
      ->getDefinition($menu_plugin_id);
    $entity = NULL;

    // Pull the path from the menu link content.
    if (strpos($menu_plugin_id, 'menu_link_content') === 0) {
      list(, $uuid) = explode(':', $menu_plugin_id, 2);

      /** @var \Drupal\menu_link_content\Entity\MenuLinkContent $entity */
      $entity = \Drupal::service('entity.repository')
        ->loadEntityByUuid('menu_link_content', $uuid);
    }
    if (isset($expected_item['children'])) {
      $child_ids = array_values($menu_link_manager
        ->getChildIds($menu_plugin_id));
      sort($expected_item['children']);
      if ($child_ids) {
        sort($child_ids);
      }
      $this
        ->assertSame($expected_item['children'], $child_ids);
      unset($expected_item['children']);
    }
    if (isset($expected_item['parents'])) {
      $parent_ids = array_values($menu_link_manager
        ->getParentIds($menu_plugin_id));
      $this
        ->assertSame($expected_item['parents'], $parent_ids);
      unset($expected_item['parents']);
    }
    if (isset($expected_item['langcode']) && $entity) {
      $this
        ->assertEquals($expected_item['langcode'], $entity->langcode->value);
      unset($expected_item['langcode']);
    }
    if (isset($expected_item['enabled']) && $entity) {
      $this
        ->assertEquals($expected_item['enabled'], $entity->enabled->value);
      unset($expected_item['enabled']);
    }
    foreach ($expected_item as $key => $value) {
      $this
        ->assertNotNull($definition[$key]);
      $this
        ->assertSame($value, $definition[$key]);
    }
  }

}