protected function MongodbMenuTreeStorageTest::assertMenuLink in MongoDB 8

Tests that a link's stored representation matches the expected values.

Parameters

string $id: The ID of the menu link to test

array $expected_properties: A keyed array of column names and values like has_children and depth.

array $parents: An ordered array of the IDs of the menu links that are the parents.

array $children: Array of child IDs that are visible (enabled == 1).

File

src/Tests/Menu/MongodbMenuTreeStorageTest.php, line 54

Contains \Drupal\mongodb\Tests\Menu\MenuTreeStorageTest.

Class

MongodbMenuTreeStorageTest

Tests the menu tree storage.

Namespace

Drupal\mongodb\Tests\Menu

Code

protected function assertMenuLink($id, array $expected_properties, array $parents = array(), array $children = array()) {
  $query = [];
  $query['id'] = $id;
  foreach ($expected_properties as $field => $value) {
    if (preg_match('/^p(\\d+)$/', $field, $matches)) {
      $field = 'p.' . $matches[1];
    }
    $query[$field] = $value;
  }
  $collection = $this->mongo
    ->get('menu_tree');
  $all = iterator_to_array($collection
    ->find($query));
  $this
    ->assertEqual(count($all), 1, "Found link {$id} matching all the expected properties");
  $raw = reset($all);

  // Put the current link onto the front.
  array_unshift($parents, $raw['id']);
  $query = $this->connection
    ->select('menu_tree');
  $query
    ->fields('menu_tree', array(
    'id',
    'mlid',
  ));
  $query
    ->condition('id', $parents, 'IN');
  $found_parents = [];
  $query = [];
  $query['id']['$in'] = $parents;
  foreach ($collection
    ->find($query, [
    'id',
    'mlid',
  ]) as $link) {
    $found_parents[$link['id']] = $link['mlid'];
  }
  $this
    ->assertEqual(count($parents), count($found_parents), 'Found expected number of parents');
  $this
    ->assertEqual($raw['depth'], count($found_parents), 'Number of parents is the same as the depth');
  $materialized_path = $this->treeStorage
    ->getRootPathIds($id);
  $this
    ->assertEqual(array_values($materialized_path), array_values($parents), 'Parents match the materialized path');

  // Check that the selected mlid values of the parents are in the correct
  // column, including the link's own.
  for ($i = $raw['depth']; $i >= 1; $i--) {
    $parent_id = array_shift($parents);
    $this
      ->assertEqual($raw["p{$i}"], $found_parents[$parent_id], "mlid of parent matches at column p{$i}");
  }
  for ($i = $raw['depth'] + 1; $i <= $this->treeStorage
    ->maxDepth(); $i++) {
    $this
      ->assertEqual($raw["p{$i}"], 0, "parent is 0 at column p{$i} greater than depth");
  }
  if ($parents) {
    $this
      ->assertEqual($raw['parent'], end($parents), 'Ensure that the parent field is set properly');
  }
  $found_children = array_keys($this->treeStorage
    ->loadAllChildren($id));

  // We need both these checks since the 2nd will pass if there are extra
  // IDs loaded in $found_children.
  $this
    ->assertEqual(count($children), count($found_children), "Found expected number of children for {$id}");
  $this
    ->assertEqual(array_intersect($children, $found_children), $children, 'Child IDs match');
}