|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace Ddeboer\DataImport\Tests\Reader; |
|
4
|
|
|
|
|
5
|
|
|
use Ddeboer\DataImport\Reader\DoctrineReader; |
|
6
|
|
|
use Doctrine\ORM\EntityManager; |
|
7
|
|
|
use Doctrine\ORM\Tools\Setup; |
|
8
|
|
|
use Ddeboer\DataImport\Tests\Fixtures\Entity\User; |
|
9
|
|
|
|
|
10
|
|
|
class DoctrineReaderTest extends \PHPUnit_Framework_TestCase |
|
11
|
|
|
{ |
|
12
|
|
|
public function testGetFields() |
|
13
|
|
|
{ |
|
14
|
|
|
$fields = $this->getReader()->getFields(); |
|
15
|
|
|
$this->assertInternalType('array', $fields); |
|
16
|
|
|
$this->assertEquals(array('id', 'username'), $fields); |
|
17
|
|
|
} |
|
18
|
|
|
|
|
19
|
|
|
public function testCount() |
|
20
|
|
|
{ |
|
21
|
|
|
$this->assertEquals(100, $this->getReader()->count()); |
|
22
|
|
|
} |
|
23
|
|
|
|
|
24
|
|
|
public function testIterate() |
|
25
|
|
|
{ |
|
26
|
|
|
$i = 1; |
|
27
|
|
|
foreach ($this->getReader() as $data) { |
|
28
|
|
|
$this->assertInternalType('array', $data); |
|
29
|
|
|
$this->assertEquals('user' . $i, $data['username']); |
|
30
|
|
|
$i++; |
|
31
|
|
|
} |
|
32
|
|
|
} |
|
33
|
|
|
|
|
34
|
|
|
protected function getReader() |
|
35
|
|
|
{ |
|
36
|
|
|
$em = $this->getEntityManager(); |
|
37
|
|
|
|
|
38
|
|
|
for ($i = 1; $i <= 100; $i++) { |
|
39
|
|
|
$user = new User(); |
|
40
|
|
|
$user->setUsername('user'. $i); |
|
41
|
|
|
$em->persist($user); |
|
42
|
|
|
} |
|
43
|
|
|
|
|
44
|
|
|
$em->flush(); |
|
45
|
|
|
|
|
46
|
|
|
return new DoctrineReader($em, 'Ddeboer\DataImport\Tests\Fixtures\Entity\User'); |
|
47
|
|
|
} |
|
48
|
|
|
|
|
49
|
|
|
protected function getEntityManager() |
|
50
|
|
|
{ |
|
51
|
|
|
$dbParams = array( |
|
52
|
|
|
'driver' => 'pdo_sqlite', |
|
53
|
|
|
); |
|
54
|
|
|
|
|
55
|
|
|
$paths = array( |
|
56
|
|
|
__DIR__.'/../Fixtures/Entity' |
|
57
|
|
|
); |
|
58
|
|
|
|
|
59
|
|
|
$config = Setup::createAnnotationMetadataConfiguration($paths, true); |
|
60
|
|
|
$em = EntityManager::create($dbParams, $config); |
|
61
|
|
|
|
|
62
|
|
|
$schemaTool = new \Doctrine\ORM\Tools\SchemaTool($em); |
|
63
|
|
|
$schemaTool->createSchema( |
|
64
|
|
|
array( |
|
65
|
|
|
$em->getMetadataFactory()->getMetadataFor('Ddeboer\DataImport\Tests\Fixtures\Entity\User') |
|
66
|
|
|
) |
|
67
|
|
|
); |
|
68
|
|
|
|
|
69
|
|
|
return $em; |
|
70
|
|
|
} |
|
71
|
|
|
} |
|
72
|
|
|
|
ddeboer /
data-import
