Single fixture

README.md

@@ -53,3 +53,11 @@ Access the Doctrine command line as following
 ```sh
 ./vendor/bin/doctrine-module data-fixture:import 
 ```
+
+##Options
+
+--append
+
+--purge-with-truncate  
+
+--fixtures=./path/to/fixtures/or/directory

src/DoctrineDataFixtureModule/Command/ImportCommand.php

@@ -51,12 +51,21 @@ class ImportCommand extends Command
      * @var Zend\ServiceManager\ServiceLocatorInterface
      */
     protected $serviceLocator;
+    /**
+     * ServiceLocatorAwareLoader
+     * @var DoctrineDataFixtureModule\Loader\ServiceLocatorAwareLoader
+     */
+    protected $loader;
+
+    protected $purger;
 
     const PURGE_MODE_TRUNCATE = 2;
 
     public function __construct(ServiceLocatorInterface $serviceLocator)
     {
         $this->serviceLocator = $serviceLocator;
+        $this->loader = new ServiceLocatorAwareLoader($this->serviceLocator);
+
         parent::__construct();
     }
 
@@ -72,24 +81,44 @@ protected function configure()
 EOT
             )
             ->addOption('append', null, InputOption::VALUE_NONE, 'Append data to existing data.')
-            ->addOption('purge-with-truncate', null, InputOption::VALUE_NONE, 'Truncate tables before inserting data');
+            ->addOption('purge-with-truncate', null, InputOption::VALUE_NONE, 'Truncate tables before inserting data')
+            ->addOption(
+                'fixtures',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'Set path to Fixture Class or Directory to be added'
+            );
     }
 
     public function execute(InputInterface $input, OutputInterface $output)
     {
-        $loader = new ServiceLocatorAwareLoader($this->serviceLocator);
-        $purger = new ORMPurger();
-
         if ($input->getOption('purge-with-truncate')) {
-            $purger->setPurgeMode(self::PURGE_MODE_TRUNCATE);
+            $this->purger->setPurgeMode(self::PURGE_MODE_TRUNCATE);
         }
 
-        $executor = new ORMExecutor($this->em, $purger);
+        if ($input->getOption('fixtures') != null) {
+            $fixtures = $input->getOption('fixtures');
+            if (is_dir($fixtures)) {
+                $this->loader->loadFromDirectory($fixtures);
+            } elseif (file_exists($fixtures)) {
+                $classes = get_declared_classes();
+                include($fixtures);
+                $newClasses = get_declared_classes();
 
-        foreach ($this->paths as $key => $value) {
-            $loader->loadFromDirectory($value);
+                $diff = array_diff($newClasses, $classes);
+                $class = array_pop($diff);
+                $this->loader->addFixture(new $class);
+            } else {
+                throw new \RuntimeException('Cannot find File or Directory.');
+            }
+        } else {
+            foreach ($this->paths as $key => $value) {
+                $this->loader->loadFromDirectory($value);
+            }
         }
-        $executor->execute($loader->getFixtures(), $input->getOption('append'));
+
+        $executor = new ORMExecutor($this->em, $this->purger);
+        $executor->execute($this->loader->getFixtures(), $input->getOption('append'));
     }
 
     public function setPath($paths)
@@ -101,4 +130,14 @@ public function setEntityManager($em)
     {
         $this->em = $em;
     }
+
+    public function getLoader()
+    {
+        return $this->loader;
+    }
+
+    public function setPurger(ORMPurger $purger)
+    {
+        $this->purger = $purger;
+    }
 }

src/DoctrineDataFixtureModule/Module.php

@@ -27,6 +27,7 @@
 use Doctrine\ORM\Tools\Console\ConsoleRunner;
 use DoctrineDataFixtureModule\Command\ImportCommand;
 use DoctrineDataFixtureModule\Service\FixtureFactory;
+use Doctrine\Common\DataFixtures\Purger\ORMPurger;
 
 /**
  * Base module for Doctrine Data Fixture.
@@ -74,6 +75,7 @@ public function init(ModuleManager $e)
             $importCommand = new ImportCommand($sm);
             $importCommand->setEntityManager($em);
             $importCommand->setPath($paths);
+            $importCommand->setORMPurger(new ORMPurger);
             ConsoleRunner::addCommands($cli);
             $cli->addCommands(array(
                 $importCommand

tests/DoctrineDataFixtureTest/Command/ImportCommandTest.php

@@ -0,0 +1,129 @@
+<?php
+/*
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * This software consists of voluntary contributions made by many individuals
+ * and is licensed under the LGPL. For more information, see
+ * <http://www.doctrine-project.org>.
+ */
+namespace DoctrineDataFixtureTest\Command;
+
+
+use DoctrineDataFixtureModule\Command\ImportCommand;
+use Doctrine\ORM\EntityManager;
+use Symfony\Component\Console\Tester\CommandTester;
+
+use Zend\ServiceManager\ServiceManager;
+use Zend\Mvc\Service\ServiceManagerConfig;
+use PHPUnit_Framework_TestCase;
+use Doctrine\ORM\Tools\Setup;
+
+/**
+ * Test Import commands for fixtures
+ *
+ * @license http://www.opensource.org/licenses/lgpl-license.php LGPL
+ * @link    www.doctrine-project.org
+ * @author  Martin Shwalbe <martin.shwalbe@gmail.com>
+ */
+class ImportCommandTest extends PHPUnit_Framework_TestCase
+{
+   /**
+     * @dataProvider provider
+     */
+    public function testExecute($option, $value, $assert)
+    {
+        $serviceLocator = new ServiceManager(new ServiceManagerConfig());
+
+        $command = new ImportCommand($serviceLocator);
+
+        $command->setentityManager($this->getMockSqliteEntityManager());
+        $command->setPurger($this->getMockPurger());
+        $paths = array(
+            'DoctrineDataFixture_Test_Paths' => __DIR__ . '/../TestAsset/Fixtures/NoSL',
+        );
+        $command->setPath($paths);
+
+        $commandTester = new CommandTester($command);
+        $commandTester->execute(
+            array(
+                $option=> $value,
+            )
+        );
+
+        $loader= $command->getLoader();        
+        $fixtures = $loader->getFixtures();
+
+        $this->assertArrayHasKey($assert, $fixtures);
+    }
+
+    public function provider() {
+        return array(
+            array('--append', true, 'DoctrineDataFixtureTest\TestAsset\Fixtures\NoSL\FixtureA'),
+            array('--fixtures', __DIR__ . '/../TestAsset/Fixtures/NoSL', 'DoctrineDataFixtureTest\TestAsset\Fixtures\NoSL\FixtureA'),
+            array('--fixtures', __DIR__ . '/../TestAsset/Fixtures/HasSL/FixtureA.php', 'DoctrineDataFixtureTest\TestAsset\Fixtures\HasSL\FixtureA')
+        );
+    }
+
+    private function getMockFixture($em)
+    {
+        return $this->getMock('Doctrine\Common\DataFixtures\FixtureInterface');
+    }
+
+    private function getMockPurger()
+    {
+        return $this->getMock('Doctrine\Common\DataFixtures\Purger\ORMPurger');
+    }
+
+    /**
+     * EntityManager mock object together with
+     * annotation mapping driver and pdo_sqlite
+     * database in memory
+     *
+     * @return EntityManager
+     */
+    protected function getMockSqliteEntityManager()
+    {
+        $dbParams = array('driver' => 'pdo_sqlite', 'memory' => true);
+        $config = Setup::createAnnotationMetadataConfiguration(array(__DIR__ . '/../TestAsset/Entity'), true);
+        return EntityManager::create($dbParams, $config);
+    }
+
+    protected function getMockEntityManager()
+    {
+        $driver = $this->getMock('Doctrine\DBAL\Driver');
+        $driver->expects($this->once())
+            ->method('getDatabasePlatform')
+            ->will($this->returnValue($this->getMock('Doctrine\DBAL\Platforms\MySqlPlatform')));
+
+        $conn = $this->getMock('Doctrine\DBAL\Connection', array(), array(array(), $driver));
+        $conn->expects($this->once())
+            ->method('getEventManager')
+            ->will($this->returnValue($this->getMock('Doctrine\Common\EventManager')));
+
+        $config = $this->getMock('Doctrine\ORM\Configuration');
+        $config->expects($this->once())
+            ->method('getProxyDir')
+            ->will($this->returnValue('test'));
+
+        $config->expects($this->once())
+            ->method('getProxyNamespace')
+            ->will($this->returnValue('Proxies'));
+
+        $config->expects($this->once())
+            ->method('getMetadataDriverImpl')
+            ->will($this->returnValue($this->getMock('Doctrine\ORM\Mapping\Driver\DriverChain')));
+
+        $em = EntityManager::create($conn, $config);
+        return $em;
+    }
+}

tests/DoctrineDataFixtureTest/TestAsset/Entity/Role.php

@@ -0,0 +1,36 @@
+<?php
+
+namespace Doctrine\Tests\Common\DataFixtures\TestEntity;
+
+/**
+ * @Entity
+ */
+class Role
+{
+    /**
+     * @Column(type="integer")
+     * @Id
+     * @GeneratedValue(strategy="IDENTITY")
+     */
+    private $id;
+
+    /**
+     * @Column(length=50)
+     */
+    private $name;
+
+    public function getId()
+    {
+        return $this->id;
+    }
+
+    public function setName($name)
+    {
+        $this->name = $name;
+    }
+
+    public function getName()
+    {
+        return $this->name;
+    }
+}

tests/DoctrineDataFixtureTest/TestAsset/Entity/User.php

@@ -0,0 +1,76 @@
+<?php
+
+namespace DoctrineDataFixtureTest\TestAsset\Entity;
+
+/**
+ * @Entity
+ */
+class User
+{
+    /**
+     * @Column(type="integer")
+     * @Id
+     */
+    private $id;
+
+    /**
+     * @Column(length=32)
+     * @Id
+     */
+    private $code;
+
+    /**
+     * @Column(length=32)
+     */
+    private $password;
+
+    /**
+     * @Column(length=255)
+     */
+    private $email;
+
+    /**
+     * @ManyToOne(targetEntity="Role", cascade={"persist"})
+     */
+    private $role;
+
+    public function setId($id)
+    {
+        $this->id = $id;
+    }
+
+    public function setCode($code)
+    {
+        $this->code = $code;
+    }
+
+    public function setPassword($password)
+    {
+        $this->password = md5($password);
+    }
+
+    public function getPassword()
+    {
+        return $this->password;
+    }
+
+    public function setEmail($email)
+    {
+        $this->email = $email;
+    }
+
+    public function getEmail()
+    {
+        return $this->email;
+    }
+
+    public function setRole(Role $role)
+    {
+        $this->role = $role;
+    }
+
+    public function getRole()
+    {
+        return $this->role;
+    }
+}