[tests] tree positioned node tests

tests/Gedmo/Timestampable/TimestampableTest.php

@@ -71,8 +71,6 @@ class TimestampableTest extends BaseTestCaseORM
         );
         $this->assertEquals(null, $sportComment->getClosed());
 
-        sleep(1);
-
         $sportComment->setStatus(1);
         $published = new Type();
         $published->setTitle('Published');

tests/Gedmo/Tree/Fixture/Category.php

@@ -83,4 +83,19 @@ class Category implements NodeInterface
     {
         return $this->parentId;
     }
+
+    public function getLeft()
+    {
+        return $this->lft;
+    }
+
+    public function getRight()
+    {
+        return $this->rgt;
+    }
+
+    public function getLevel()
+    {
+        return $this->level;
+    }
 }

tests/Gedmo/Tree/Fixture/RootCategory.php

@@ -43,7 +43,7 @@ class RootCategory
 
     /**
      * @gedmo:TreeRoot
-     * @Column(type="integer", nullable=true)
+     * @Column(type="integer")
      */
     private $root;
 

tests/Gedmo/Tree/NestedTreePositionTest.php

@@ -0,0 +1,245 @@
+<?php
+
+namespace Gedmo\Tree;
+
+use Doctrine\Common\EventManager;
+use Tool\BaseTestCaseORM;
+use Tree\Fixture\Category;
+use Tree\Fixture\RootCategory;
+
+/**
+ * These are tests for Tree behavior
+ *
+ * @author Gediminas Morkevicius <gediminas.morkevicius@gmail.com>
+ * @package Gedmo.Tree
+ * @link http://www.gediminasm.org
+ * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
+ */
+class NestedTreePositionTest extends BaseTestCaseORM
+{
+    const CATEGORY = "Tree\\Fixture\\Category";
+    const ROOT_CATEGORY = "Tree\\Fixture\\RootCategory";
+
+    protected function setUp()
+    {
+        parent::setUp();
+
+        $evm = new EventManager;
+        $evm->addEventSubscriber(new TreeListener);
+
+        $this->getMockSqliteEntityManager($evm);
+    }
+
+    public function testOnRootCategory()
+    {
+        // need to check if this does not produce errors
+        $repo = $this->em->getRepository(self::ROOT_CATEGORY);
+
+        $fruits = new RootCategory;
+        $fruits->setTitle('Fruits');
+
+        $vegitables = new RootCategory;
+        $vegitables->setTitle('Vegitables');
+
+        $milk = new RootCategory;
+        $milk->setTitle('Milk');
+
+        $meat = new RootCategory;
+        $meat->setTitle('Meat');
+
+        $repo
+            ->persistAsFirstChild($fruits)
+            ->persistAsFirstChild($vegitables)
+            ->persistAsLastChild($milk)
+            ->persistAsLastChild($meat);
+
+        $cookies = new RootCategory;
+        $cookies->setTitle('Cookies');
+
+        $drinks = new RootCategory;
+        $drinks->setTitle('Drinks');
+
+        $repo
+            ->persistAsNextSibling($cookies)
+            ->persistAsPrevSibling($drinks);
+
+        $this->em->flush();
+        $dql = 'SELECT COUNT(c) FROM ' . self::ROOT_CATEGORY . ' c';
+        $dql .= ' WHERE c.lft = 1 AND c.rgt = 2 AND c.parent IS NULL AND c.level = 0';
+        $count = $this->em->createQuery($dql)->getSingleScalarResult();
+        $this->assertEquals(6, $count);
+
+        $repo = $this->em->getRepository(self::CATEGORY);
+
+        $fruits = new Category;
+        $fruits->setTitle('Fruits');
+
+        $vegitables = new Category;
+        $vegitables->setTitle('Vegitables');
+
+        $milk = new Category;
+        $milk->setTitle('Milk');
+
+        $meat = new Category;
+        $meat->setTitle('Meat');
+
+        $repo
+            ->persistAsFirstChild($fruits)
+            ->persistAsFirstChild($vegitables)
+            ->persistAsLastChild($milk)
+            ->persistAsLastChild($meat);
+
+        $cookies = new Category;
+        $cookies->setTitle('Cookies');
+
+        $drinks = new Category;
+        $drinks->setTitle('Drinks');
+
+        $repo
+            ->persistAsNextSibling($cookies)
+            ->persistAsPrevSibling($drinks);
+
+        $this->em->flush();
+        $dql = 'SELECT COUNT(c) FROM ' . self::CATEGORY . ' c';
+        $dql .= ' WHERE c.parentId IS NULL AND c.level = 0';
+        $dql .= ' AND c.lft BETWEEN 1 AND 11';
+        $count = $this->em->createQuery($dql)->getSingleScalarResult();
+        $this->assertEquals(6, $count);
+    }
+
+    public function testRootTreePositionedInserts()
+    {
+        $repo = $this->em->getRepository(self::ROOT_CATEGORY);
+
+        // test child positioned inserts
+        $food = new RootCategory;
+        $food->setTitle('Food');
+
+        $fruits = new RootCategory;
+        $fruits->setTitle('Fruits');
+
+        $vegitables = new RootCategory;
+        $vegitables->setTitle('Vegitables');
+
+        $milk = new RootCategory;
+        $milk->setTitle('Milk');
+
+        $meat = new RootCategory;
+        $meat->setTitle('Meat');
+
+        $repo
+            ->persistAsFirstChild($food)
+            ->persistAsFirstChildOf($fruits, $food)
+            ->persistAsFirstChildOf($vegitables, $food)
+            ->persistAsLastChildOf($milk, $food)
+            ->persistAsLastChildOf($meat, $food);
+
+        $this->em->flush();
+
+        $this->assertEquals(4, $fruits->getLeft());
+        $this->assertEquals(5, $fruits->getRight());
+
+        $this->assertEquals(2, $vegitables->getLeft());
+        $this->assertEquals(3, $vegitables->getRight());
+
+        $this->assertEquals(6, $milk->getLeft());
+        $this->assertEquals(7, $milk->getRight());
+
+        $this->assertEquals(8, $meat->getLeft());
+        $this->assertEquals(9, $meat->getRight());
+
+        // test sibling positioned inserts
+        $cookies = new RootCategory;
+        $cookies->setTitle('Cookies');
+
+        $drinks = new RootCategory;
+        $drinks->setTitle('Drinks');
+
+        $repo
+            ->persistAsNextSiblingOf($cookies, $milk)
+            ->persistAsPrevSiblingOf($drinks, $milk);
+
+        $this->em->flush();
+
+        $this->assertEquals(6, $drinks->getLeft());
+        $this->assertEquals(7, $drinks->getRight());
+
+        $this->assertEquals(10, $cookies->getLeft());
+        $this->assertEquals(11, $cookies->getRight());
+
+        $this->assertTrue($repo->verify());
+    }
+
+    public function testSimpleTreePositionedInserts()
+    {
+        $repo = $this->em->getRepository(self::CATEGORY);
+
+        // test child positioned inserts
+        $food = new Category;
+        $food->setTitle('Food');
+        $repo->persistAsFirstChild($food);
+
+        $fruits = new Category;
+        $fruits->setTitle('Fruits');
+        $fruits->setParent($food);
+        $repo->persistAsFirstChild($fruits);
+
+        $vegitables = new Category;
+        $vegitables->setTitle('Vegitables');
+        $vegitables->setParent($food);
+        $repo->persistAsFirstChild($vegitables);
+
+        $milk = new Category;
+        $milk->setTitle('Milk');
+        $milk->setParent($food);
+        $repo->persistAsLastChild($milk);
+
+        $meat = new Category;
+        $meat->setTitle('Meat');
+        $meat->setParent($food);
+        $repo->persistAsLastChild($meat);
+
+        $this->em->flush();
+
+        $this->assertEquals(4, $fruits->getLeft());
+        $this->assertEquals(5, $fruits->getRight());
+
+        $this->assertEquals(2, $vegitables->getLeft());
+        $this->assertEquals(3, $vegitables->getRight());
+
+        $this->assertEquals(6, $milk->getLeft());
+        $this->assertEquals(7, $milk->getRight());
+
+        $this->assertEquals(8, $meat->getLeft());
+        $this->assertEquals(9, $meat->getRight());
+
+        // test sibling positioned inserts
+        $cookies = new Category;
+        $cookies->setTitle('Cookies');
+        $cookies->setParent($milk);
+        $repo->persistAsNextSibling($cookies);
+
+        $drinks = new Category;
+        $drinks->setTitle('Drinks');
+        $drinks->setParent($milk);
+        $repo->persistAsPrevSibling($drinks);
+
+        $this->em->flush();
+
+        $this->assertEquals(6, $drinks->getLeft());
+        $this->assertEquals(7, $drinks->getRight());
+
+        $this->assertEquals(10, $cookies->getLeft());
+        $this->assertEquals(11, $cookies->getRight());
+
+        $this->assertTrue($repo->verify());
+    }
+
+    protected function getUsedEntityFixtures()
+    {
+        return array(
+            self::CATEGORY,
+            self::ROOT_CATEGORY
+        );
+    }
+}

tests/phpunit.dist.xml

@@ -17,6 +17,7 @@
             <file>./Gedmo/Tree/TranslatableSluggableTreeTest.php</file>
             <file>./Gedmo/Tree/TreeTest.php</file>
             <file>./Gedmo/Tree/InMemoryUpdatesTest.php</file>
+            <file>./Gedmo/Tree/NestedTreePositionTest.php</file>
         </testsuite>
         <testsuite name="Timestampable Extension">
             <directory suffix=".php">./Gedmo/Timestampable/</directory>