ClosureTreeRepositoryTest.php 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. <?php
  2. namespace Gedmo\Tree;
  3. use Doctrine\Common\EventManager;
  4. use Tool\BaseTestCaseORM;
  5. use Tree\Fixture\Closure\Category;
  6. /**
  7. * These are tests for Tree behavior
  8. *
  9. * @author Gustavo Adrian <comfortablynumb84@gmail.com>
  10. * @author Gediminas Morkevicius <gediminas.morkevicius@gmail.com>
  11. * @package Gedmo.Tree
  12. * @link http://www.gediminasm.org
  13. * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
  14. */
  15. class ClosureTreeRepositoryTest extends BaseTestCaseORM
  16. {
  17. const CATEGORY = "Tree\\Fixture\\Closure\\Category";
  18. const CLOSURE = "Tree\\Fixture\\Closure\\CategoryClosure";
  19. protected function setUp()
  20. {
  21. parent::setUp();
  22. $evm = new EventManager;
  23. $evm->addEventSubscriber(new TreeListener);
  24. $this->getMockSqliteEntityManager($evm);
  25. $this->populate();
  26. }
  27. public function testChildCount()
  28. {
  29. $repo = $this->em->getRepository(self::CATEGORY);
  30. $food = $repo->findOneByTitle('Food');
  31. $directCount = $repo->childCount($food, true);
  32. $this->assertEquals(3, $directCount);
  33. $fruits = $repo->findOneByTitle('Fruits');
  34. $count = $repo->childCount($fruits);
  35. $this->assertEquals(4, $count);
  36. $rootCount = $repo->childCount(null, true);
  37. $this->assertEquals(1, $rootCount);
  38. }
  39. public function testPath()
  40. {
  41. $repo = $this->em->getRepository(self::CATEGORY);
  42. $fruits = $repo->findOneByTitle('Fruits');
  43. $path = $repo->getPath($fruits);
  44. $this->assertCount(2, $path);
  45. $this->assertEquals('Food', $path[0]->getTitle());
  46. $this->assertEquals('Fruits', $path[1]->getTitle());
  47. $strawberries = $repo->findOneByTitle('Strawberries');
  48. $path = $repo->getPath($strawberries);
  49. $this->assertCount(4, $path);
  50. $this->assertEquals('Food', $path[0]->getTitle());
  51. $this->assertEquals('Fruits', $path[1]->getTitle());
  52. $this->assertEquals('Berries', $path[2]->getTitle());
  53. $this->assertEquals('Strawberries', $path[3]->getTitle());
  54. }
  55. public function testChildren()
  56. {
  57. $repo = $this->em->getRepository(self::CATEGORY);
  58. $fruits = $repo->findOneByTitle('Fruits');
  59. // direct children of node, sorted by title ascending order
  60. $children = $repo->children($fruits, true, 'title');
  61. $this->assertCount(3, $children);
  62. $this->assertEquals('Berries', $children[0]->getTitle());
  63. $this->assertEquals('Lemons', $children[1]->getTitle());
  64. $this->assertEquals('Oranges', $children[2]->getTitle());
  65. // all children of node
  66. $children = $repo->children($fruits);
  67. $this->assertCount(4, $children);
  68. $this->assertEquals('Oranges', $children[0]->getTitle());
  69. $this->assertEquals('Lemons', $children[1]->getTitle());
  70. $this->assertEquals('Berries', $children[2]->getTitle());
  71. $this->assertEquals('Strawberries', $children[3]->getTitle());
  72. // direct root nodes
  73. $children = $repo->children(null, true, 'title');
  74. $this->assertCount(1, $children);
  75. $this->assertEquals('Food', $children[0]->getTitle());
  76. // all tree
  77. $children = $repo->children();
  78. $this->assertCount(12, $children);
  79. }
  80. public function testSingleNodeRemoval()
  81. {
  82. $repo = $this->em->getRepository(self::CATEGORY);
  83. $fruits = $repo->findOneByTitle('Fruits');
  84. $repo->removeFromTree($fruits);
  85. // ensure in memory node integrity
  86. $this->em->flush();
  87. $food = $repo->findOneByTitle('Food');
  88. $children = $repo->children($food, true);
  89. $this->assertCount(5, $children);
  90. $berries = $repo->findOneByTitle('Berries');
  91. $this->assertEquals(1, $repo->childCount($berries, true));
  92. $lemons = $repo->findOneByTitle('Lemons');
  93. $this->assertEquals(0, $repo->childCount($lemons, true));
  94. $repo->removeFromTree($food);
  95. $vegitables = $repo->findOneByTitle('Vegitables');
  96. $this->assertEquals(2, $repo->childCount($vegitables, true));
  97. $this->assertNull($vegitables->getParent());
  98. $repo->removeFromTree($lemons);
  99. $this->assertCount(4, $repo->children(null, true));
  100. }
  101. public function testBuildTree()
  102. {
  103. $repo = $this->em->getRepository(self::CATEGORY);
  104. $roots = $repo->getRootNodes();
  105. $tree = $repo->childrenHierarchy(
  106. $roots[0],
  107. false,
  108. array('childSort' => array('field' => 'title', 'dir' => 'asc'))
  109. );
  110. $fruits = $tree[0]['__children'][0];
  111. $milk = $tree[0]['__children'][1];
  112. $vegitables = $tree[0]['__children'][2];
  113. $this->assertEquals('Food', $tree[0]['title']);
  114. $this->assertEquals('Fruits', $fruits['title']);
  115. $this->assertEquals('Berries', $fruits['__children'][0]['title']);
  116. $this->assertEquals('Strawberries', $fruits['__children'][0]['__children'][0]['title']);
  117. $this->assertEquals('Milk', $milk['title']);
  118. $this->assertEquals('Cheese', $milk['__children'][0]['title']);
  119. $this->assertEquals('Mould cheese', $milk['__children'][0]['__children'][0]['title']);
  120. $this->assertEquals('Vegitables', $vegitables['title']);
  121. $this->assertEquals('Cabbages', $vegitables['__children'][0]['title']);
  122. $this->assertEquals('Carrots', $vegitables['__children'][1]['title']);
  123. $food = $repo->findOneByTitle('Food');
  124. $vegitables = $repo->findOneByTitle('Vegitables');
  125. $boringFood = new Category();
  126. $boringFood->setTitle('Boring Food');
  127. $boringFood->setParent($food);
  128. $vegitables->setParent($boringFood);
  129. $this->em->persist($boringFood);
  130. $this->em->flush();
  131. $tree = $repo->childrenHierarchy(
  132. $roots[0],
  133. false,
  134. array('childSort' => array('field' => 'title', 'dir' => 'asc'))
  135. );
  136. $boringFood = $tree[0]['__children'][0];
  137. $fruits = $tree[0]['__children'][1];
  138. $milk = $tree[0]['__children'][2];
  139. $vegitables = $boringFood['__children'][0];
  140. $this->assertEquals('Food', $tree[0]['title']);
  141. $this->assertEquals('Fruits', $fruits['title']);
  142. $this->assertEquals('Berries', $fruits['__children'][0]['title']);
  143. $this->assertEquals('Strawberries', $fruits['__children'][0]['__children'][0]['title']);
  144. $this->assertEquals('Milk', $milk['title']);
  145. $this->assertEquals('Cheese', $milk['__children'][0]['title']);
  146. $this->assertEquals('Mould cheese', $milk['__children'][0]['__children'][0]['title']);
  147. $this->assertEquals('Boring Food', $boringFood['title']);
  148. $this->assertEquals('Vegitables', $boringFood['__children'][0]['title']);
  149. $this->assertEquals('Cabbages', $vegitables['__children'][0]['title']);
  150. $this->assertEquals('Carrots', $vegitables['__children'][1]['title']);
  151. }
  152. protected function getUsedEntityFixtures()
  153. {
  154. return array(
  155. self::CATEGORY,
  156. self::CLOSURE
  157. );
  158. }
  159. private function populate()
  160. {
  161. $food = new Category;
  162. $food->setTitle("Food");
  163. $this->em->persist($food);
  164. $vegitables = new Category;
  165. $vegitables->setTitle('Vegitables');
  166. $vegitables->setParent($food);
  167. $this->em->persist($vegitables);
  168. $fruits = new Category;
  169. $fruits->setTitle('Fruits');
  170. $fruits->setParent($food);
  171. $this->em->persist($fruits);
  172. $oranges = new Category;
  173. $oranges->setTitle('Oranges');
  174. $oranges->setParent($fruits);
  175. $this->em->persist($oranges);
  176. $lemons = new Category;
  177. $lemons->setTitle('Lemons');
  178. $lemons->setParent($fruits);
  179. $this->em->persist($lemons);
  180. $berries = new Category;
  181. $berries->setTitle('Berries');
  182. $berries->setParent($fruits);
  183. $this->em->persist($berries);
  184. $strawberries = new Category;
  185. $strawberries->setTitle('Strawberries');
  186. $strawberries->setParent($berries);
  187. $this->em->persist($strawberries);
  188. $cabbages = new Category;
  189. $cabbages->setTitle('Cabbages');
  190. $cabbages->setParent($vegitables);
  191. $this->em->persist($cabbages);
  192. $carrots = new Category;
  193. $carrots->setTitle('Carrots');
  194. $carrots->setParent($vegitables);
  195. $this->em->persist($carrots);
  196. $milk = new Category;
  197. $milk->setTitle('Milk');
  198. $milk->setParent($food);
  199. $this->em->persist($milk);
  200. $cheese = new Category;
  201. $cheese->setTitle('Cheese');
  202. $cheese->setParent($milk);
  203. $this->em->persist($cheese);
  204. $mouldCheese = new Category;
  205. $mouldCheese->setTitle('Mould cheese');
  206. $mouldCheese->setParent($cheese);
  207. $this->em->persist($mouldCheese);
  208. $this->em->flush();
  209. $this->em->clear();
  210. }
  211. }