%PDF-
%PDF-
Mini Shell
Mini Shell
<?php
namespace Ddeboer\DataImport\Tests\Reader;
use Ddeboer\DataImport\Reader\DoctrineReader;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\Tools\Setup;
use Ddeboer\DataImport\Tests\Fixtures\Entity\User;
class DoctrineReaderTest extends \PHPUnit_Framework_TestCase
{
public function testGetFields()
{
$fields = $this->getReader()->getFields();
$this->assertInternalType('array', $fields);
$this->assertEquals(array('id', 'username'), $fields);
}
public function testCount()
{
$this->assertEquals(100, $this->getReader()->count());
}
public function testIterate()
{
$i = 1;
foreach ($this->getReader() as $data) {
$this->assertInternalType('array', $data);
$this->assertEquals('user' . $i, $data['username']);
$i++;
}
}
protected function getReader()
{
$em = $this->getEntityManager();
for ($i = 1; $i <= 100; $i++) {
$user = new User();
$user->setUsername('user'. $i);
$em->persist($user);
}
$em->flush();
return new DoctrineReader($em, 'Ddeboer\DataImport\Tests\Fixtures\Entity\User');
}
protected function getEntityManager()
{
$dbParams = array(
'driver' => 'pdo_sqlite',
);
$paths = array(
__DIR__.'/../Fixtures/Entity'
);
$config = Setup::createAnnotationMetadataConfiguration($paths, true);
$em = EntityManager::create($dbParams, $config);
$schemaTool = new \Doctrine\ORM\Tools\SchemaTool($em);
$schemaTool->createSchema(
array(
$em->getMetadataFactory()->getMetadataFor('Ddeboer\DataImport\Tests\Fixtures\Entity\User')
)
);
return $em;
}
}
Zerion Mini Shell 1.0