Merge pull request #75 from symfony-cmf/orm_purge_database

Added purgeDatabase method

Author: wouterj

CHANGELOG.md

@@ -1,6 +1,11 @@
 Changelog
 =========
 
+dev-master
+----------
+
+* **2014-10-06**: Added purgeDatabase method to ORM manager
+
 1.2.0-RC1
 ---------
 

src/Functional/DbManager/ORM.php

@@ -32,6 +32,11 @@
  */
 class ORM
 {
+    /**
+     * @var ORMExecutor
+     */
+    private $executor;
+
     /**
      * @var ContainerInterface
      */
@@ -77,27 +82,31 @@ public function getOm($managerName = null)
         return $this->om;
     }
 
+    /**
+     * Purge the database
+     */
+    public function purgeDatabase()
+    {
+        $referenceRepository = new ProxyReferenceRepository($this->getOm());
+        $this->getExecutor()->setReferenceRepository($referenceRepository);
+        $this->getExecutor()->purge();
+    }
+
     /**
      * Loads fixture classes.
      * 
      * @param string[] $classNames
      */
     public function loadFixtures(array $classNames)
     {
+        $this->purgeDatabase();
         $loader = new ContainerAwareLoader($this->container);;
-        $purger = new ORMPurger();
-        $executor = new ORMExecutor($this->getOm(), $purger);
-
-        $referenceRepository = new ProxyReferenceRepository($this->getOm());
-
-        $executor->setReferenceRepository($referenceRepository);
-        $executor->purge();
 
         foreach ($classNames as $className) {
             $this->loadFixtureClass($loader, $className);
         }
 
-        $executor->execute($loader->getFixtures(), true);
+        $this->getExecutor()->execute($loader->getFixtures(), true);
     }
 
     /**
@@ -130,4 +139,21 @@ protected function loadFixtureClass(Loader $loader, $className)
             }
         }
     }
+
+    /**
+     * Return the ORM Executor class
+     *
+     * @return ORMExecutor
+     */
+    private function getExecutor()
+    {
+        if ($this->executor) {
+            return $this->executor;
+        }
+
+        $purger = new ORMPurger();
+        $this->executor = new ORMExecutor($this->getOm(), $purger);
+
+        return $this->executor;
+    }
 }