cakephp/tests/TestCase/Model/Behavior/TreeBehaviorTest.php

<?php
/**
 * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
 * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 *
 * Licensed under The MIT License
 * For full copyright and license information, please see the LICENSE.txt
 * Redistributions of files must retain the above copyright notice.
 *
 * @copyright     Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 * @link          http://cakephp.org CakePHP(tm) Project
 * @since         CakePHP(tm) v 3.0.0
 * @license       MIT License (http://www.opensource.org/licenses/mit-license.php)
 */
namespace Cake\Test\TestCase\Model\Behavior;

use Cake\Collection\Collection;
use Cake\Event\Event;
use Cake\Model\Behavior\TranslateBehavior;
use Cake\ORM\Entity;
use Cake\ORM\TableRegistry;
use Cake\TestSuite\TestCase;

/**
 * Translate behavior test case
 */
class TreeBehaviorTest extends TestCase {

/**
 * fixtures
 *
 * @var array
 */
	public $fixtures = [
		'core.number_tree',
		'core.menu_link_tree'
	];

	public function setUp() {
		$this->table = TableRegistry::get('NumberTrees');
		$this->table->addBehavior('Tree');
	}

	public function tearDown() {
		parent::tearDown();
		TableRegistry::clear();
	}

/**
 * Tests the find('path') method
 *
 * @return void
 */
	public function testFindPath() {
		$nodes = $this->table->find('path', ['for' => 9]);
		$this->assertEquals([1, 6, 9], $nodes->extract('id')->toArray());

		$nodes = $this->table->find('path', ['for' => 10]);
		$this->assertEquals([1, 6, 10], $nodes->extract('id')->toArray());

		$nodes = $this->table->find('path', ['for' => 5]);
		$this->assertEquals([1, 2, 5], $nodes->extract('id')->toArray());

		$nodes = $this->table->find('path', ['for' => 1]);
		$this->assertEquals([1], $nodes->extract('id')->toArray());

		// find path with scope
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$nodes = $table->find('path', ['for' => 5]);
		$this->assertEquals([1, 3, 4, 5], $nodes->extract('id')->toArray());
	}

/**
 * Tests the childCount() method
 *
 * @return void
 */
	public function testChildCount() {
		// direct children for the root node
		$countDirect = $this->table->childCount(1, true);
		$this->assertEquals(2, $countDirect);

		// counts all the children of root
		$count = $this->table->childCount(1, false);
		$this->assertEquals(9, $count);

		// counts direct children
		$count = $this->table->childCount(2, false);
		$this->assertEquals(3, $count);

		// count children for a middle-node
		$count = $this->table->childCount(6, false);
		$this->assertEquals(4, $count);

		// count leaf children
		$count = $this->table->childCount(10, false);
		$this->assertEquals(0, $count);

		// test scoping
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$count = $table->childCount(3, false);
		$this->assertEquals(2, $count);
	}

/**
 * Tests the childCount() plus callable scoping
 *
 * @return void
 */
	public function testCallableScoping() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', [
			'scope' => function ($query) {
				return $query->where(['menu' => 'main-menu']);
			}
		]);
		$count = $table->childCount(1, false);
		$this->assertEquals(4, $count);
	}

/**
 * Tests the find('children') method
 *
 * @return void
 */
	public function testFindChildren() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);

		// root
		$nodeIds = [];
		$nodes = $table->find('children', ['for' => 1])->all();
		$this->assertEquals([2, 3, 4, 5], $nodes->extract('id')->toArray());

		// leaf
		$nodeIds = [];
		$nodes = $table->find('children', ['for' => 5])->all();
		$this->assertEquals(0, count($nodes->extract('id')->toArray()));

		// direct children
		$nodes = $table->find('children', ['for' => 1, 'direct' => true])->all();
		$this->assertEquals([2, 3], $nodes->extract('id')->toArray());
	}

/**
 * Tests that find('children') will throw an exception if the node was not found
 *
 * @expectedException \Cake\ORM\Error\RecordNotFoundException
 * @return void
 */
	public function testFindChildrenException() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$query = $table->find('children', ['for' => 500]);
	}

/**
 * Tests the moveUp() method
 *
 * @return void
 */
	public function testMoveUp() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);

		// top level, wont move
		$this->assertFalse($this->table->moveUp(1, 10));

		// edge cases
		$this->assertFalse($this->table->moveUp(1, 0));
		$this->assertFalse($this->table->moveUp(1, -10));

		// move inner node
		$result = $table->moveUp(3, 1);
		$nodes = $table->find('children', ['for' => 1])->all();
		$this->assertEquals([3, 4, 5, 2], $nodes->extract('id')->toArray());
		$this->assertTrue($result);

		// move leaf
		$this->assertFalse($table->moveUp(5, 1));

		// move to first position
		$table->moveUp(8, true);
		$nodes = $table->find()
			->select(['id'])
			->where(function($exp) {
				return $exp->isNull('parent_id');
			})
			->where(['menu' => 'main-menu'])
			->order(['lft' => 'ASC'])
			->all();
		$this->assertEquals([8, 1, 6], $nodes->extract('id')->toArray());
	}

/**
 * Tests that moveUp() will throw an exception if the node was not found
 *
 * @expectedException \Cake\ORM\Error\RecordNotFoundException
 * @return void
 */
	public function testMoveUpException() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$table->moveUp(500, 1);
	}

/**
 * Tests the moveDown() method
 *
 * @return void
 */
	public function testMoveDown() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);

		// latest node, wont move
		$this->assertFalse($this->table->moveDown(8, 10));

		// edge cases
		$this->assertFalse($this->table->moveDown(8, 0));
		$this->assertFalse($this->table->moveUp(8, -10));

		// move inner node
		$result = $table->moveDown(2, 1);
		$nodes = $table->find('children', ['for' => 1])->all();
		$this->assertEquals([3, 4, 5, 2], $nodes->extract('id')->toArray());
		$this->assertTrue($result);

		// move leaf
		$this->assertFalse( $table->moveDown(5, 1));

		// move to last position
		$table->moveDown(1, true);
		$nodes = $table->find()
			->select(['id'])
			->where(function($exp) {
				return $exp->isNull('parent_id');
			})
			->where(['menu' => 'main-menu'])
			->order(['lft' => 'ASC'])
			->all();
		$this->assertEquals([6, 8, 1], $nodes->extract('id')->toArray());
	}

/**
 * Tests that moveDown() will throw an exception if the node was not found
 *
 * @expectedException \Cake\ORM\Error\RecordNotFoundException
 * @return void
 */
	public function testMoveDownException() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$table->moveDown(500, 1);
	}

/**
 * Tests the recover function
 *
 * @return void
 */
	public function testRecover() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->updateAll(['lft' => null, 'rght' => null], []);
		$table->recover();
		$result = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $result);
	}

/**
 * Tests the recover function with a custom scope
 *
 * @return void
 */
	public function testRecoverScoped() {
		$table = TableRegistry::get('MenuLinkTrees');
		$table->addBehavior('Tree', ['scope' => ['menu' => 'main-menu']]);
		$expected = $table->find()
			->where(['menu' => 'main-menu'])
			->order('lft')
			->hydrate(false)
			->toArray();

		$expected2 = $table->find()
			->where(['menu' => 'categories'])
			->order('lft')
			->hydrate(false)
			->toArray();

		$table->updateAll(['lft' => null, 'rght' => null], ['menu' => 'main-menu']);
		$table->recover();
		$result = $table->find()
			->where(['menu' => 'main-menu'])
			->order('lft')
			->hydrate(false)
			->toArray();
		$this->assertEquals($expected, $result);

		$result2 = $table->find()
			->where(['menu' => 'categories'])
			->order('lft')
			->hydrate(false)
			->toArray();
		$this->assertEquals($expected2, $result2);
	}

/**
 * Tests adding a new orphan node
 *
 * @return void
 */
	public function testAddOrphan() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = new Entity(
			['name' => 'New Orphan', 'parent_id' => null],
			['markNew' => true]
		);
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(23, $entity->lft);
		$this->assertEquals(24, $entity->rght);

		$expected[] = $entity->toArray();
		$results = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $results);
	}

/**
 * Tests that adding a child node as a decendant of one of the roots works
 *
 * @return void
 */
	public function testAddMiddle() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = new Entity(
			['name' => 'laptops', 'parent_id' => 1],
			['markNew' => true]
		);
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(20, $entity->lft);
		$this->assertEquals(21, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->recover();
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $result);
	}

/**
 * Tests adding a leaf to the tree
 *
 * @return void
 */
	public function testAddLeaf() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = new Entity(
			['name' => 'laptops', 'parent_id' => 2],
			['markNew' => true]
		);
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(9, $entity->lft);
		$this->assertEquals(10, $entity->rght);

		$results = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->recover();
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $results);
	}

/**
 * Tests moving a subtree to the right
 *
 * @return void
 */
	public function testReParentSubTreeRight() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(2);
		$entity->parent_id = 6;
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(11, $entity->lft);
		$this->assertEquals(18, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false);
		$expected = [1, 6, 7, 8, 9, 10, 2, 3, 4, 5, 11];
		$this->assertEquals($expected, $result->extract('id')->toArray());
		$numbers = [];
		$result->each(function($v) use (&$numbers) {
			$numbers[] = $v['lft'];
			$numbers[] = $v['rght'];
		});
		sort($numbers);
		$this->assertEquals(range(1, 22), $numbers);
	}

/**
 * Tests moving a subtree to the left
 *
 * @return void
 */
	public function testReParentSubTreeLeft() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(6);
		$entity->parent_id = 2;
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(9, $entity->lft);
		$this->assertEquals(18, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->recover();
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $result);
	}

/**
 * Test moving a leaft to the left
 *
 * @return void
 */
	public function testReParentLeafLeft() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(10);
		$entity->parent_id = 2;
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(9, $entity->lft);
		$this->assertEquals(10, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->recover();
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $result);
	}

/**
 * Test moving a leaft to the left
 *
 * @return void
 */
	public function testReParentLeafRight() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(5);
		$entity->parent_id = 6;
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(17, $entity->lft);
		$this->assertEquals(18, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false);
		$expected = [1, 2, 3, 4, 6, 7, 8, 9, 10, 5, 11];
		$this->assertEquals($expected, $result->extract('id')->toArray());
		$numbers = [];
		$result->each(function($v) use (&$numbers) {
			$numbers[] = $v['lft'];
			$numbers[] = $v['rght'];
		});
		sort($numbers);
		$this->assertEquals(range(1, 22), $numbers);
	}

/**
 * Tests moving a subtree as a new root
 *
 * @return void
 */
	public function testRootingSubTree() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(2);
		$entity->parent_id = null;
		$this->assertSame($entity, $table->save($entity));
		$this->assertEquals(15, $entity->lft);
		$this->assertEquals(22, $entity->rght);

		$result = $table->find()->order('lft')->hydrate(false);
		$expected = [1, 6, 7, 8, 9, 10, 11, 2, 3, 4, 5];
		$this->assertEquals($expected, $result->extract('id')->toArray());
		$numbers = [];
		$result->each(function($v) use (&$numbers) {
			$numbers[] = $v['lft'];
			$numbers[] = $v['rght'];
		});
		sort($numbers);
		$this->assertEquals(range(1, 22), $numbers);
	}

/**
 * Tests that trying to create a cycle throws an exception
 *
 * @expectedException RuntimeException
 * @expectedExceptionMessage Cannot use node "5" as parent for entity "2"
 * @return void
 */
	public function testReparentCycle() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(2);
		$entity->parent_id = 5;
		$table->save($entity);
	}

/**
 * Tests deleting a leaf in the tree
 *
 * @return void
 */
	public function testDeleteLeaf() {
		$table = TableRegistry::get('NumberTrees');
		$table->addBehavior('Tree');
		$entity = $table->get(4);
		$this->assertTrue($table->delete($entity));
		$result = $table->find()->order('lft')->hydrate(false)->toArray();
		$table->recover();
		$expected = $table->find()->order('lft')->hydrate(false)->toArray();
		$this->assertEquals($expected, $result);
	}
}