]>
Commit | Line | Data |
---|---|---|
735068d1 JB |
1 | <?php |
2 | ||
3 | namespace Wallabag\CoreBundle\Tests\Subscriber; | |
4 | ||
735068d1 JB |
5 | use Doctrine\ORM\Mapping\ClassMetadata; |
6 | use Doctrine\ORM\Event\LoadClassMetadataEventArgs; | |
7 | use Doctrine\Common\EventManager; | |
8 | use Wallabag\CoreBundle\Subscriber\TablePrefixSubscriber; | |
9 | ||
8a493541 | 10 | class TablePrefixSubscriberTest extends \PHPUnit_Framework_TestCase |
735068d1 JB |
11 | { |
12 | public function dataForPrefix() | |
13 | { | |
14 | return array( | |
15 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'wallabag_user', '"wallabag_user"', new \Doctrine\DBAL\Platforms\PostgreSqlPlatform()), | |
16 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'wallabag_user', '`wallabag_user`', new \Doctrine\DBAL\Platforms\MySqlPlatform()), | |
17 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'wallabag_user', '"wallabag_user"', new \Doctrine\DBAL\Platforms\SqlitePlatform()), | |
18 | ||
19 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'wallabag_user', 'wallabag_user', new \Doctrine\DBAL\Platforms\PostgreSqlPlatform()), | |
20 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'wallabag_user', 'wallabag_user', new \Doctrine\DBAL\Platforms\MySqlPlatform()), | |
21 | array('wallabag_', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'wallabag_user', 'wallabag_user', new \Doctrine\DBAL\Platforms\SqlitePlatform()), | |
22 | ||
23 | array('', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'user', '"user"', new \Doctrine\DBAL\Platforms\PostgreSqlPlatform()), | |
24 | array('', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'user', '`user`', new \Doctrine\DBAL\Platforms\MySqlPlatform()), | |
25 | array('', 'Wallabag\UserBundle\Entity\User', '`user`', 'user', 'user', '"user"', new \Doctrine\DBAL\Platforms\SqlitePlatform()), | |
26 | ||
27 | array('', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'user', 'user', new \Doctrine\DBAL\Platforms\PostgreSqlPlatform()), | |
28 | array('', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'user', 'user', new \Doctrine\DBAL\Platforms\MySqlPlatform()), | |
29 | array('', 'Wallabag\UserBundle\Entity\User', 'user', 'user', 'user', 'user', new \Doctrine\DBAL\Platforms\SqlitePlatform()), | |
30 | ); | |
31 | } | |
32 | ||
33 | /** | |
34 | * @dataProvider dataForPrefix | |
35 | */ | |
36 | public function testPrefix($prefix, $entityName, $tableName, $tableNameExpected, $finalTableName, $finalTableNameQuoted, $platform) | |
37 | { | |
38 | $em = $this->getMockBuilder('Doctrine\ORM\EntityManager') | |
39 | ->disableOriginalConstructor() | |
40 | ->getMock(); | |
41 | ||
42 | $subscriber = new TablePrefixSubscriber($prefix); | |
43 | ||
44 | $metaClass = new ClassMetadata($entityName); | |
45 | $metaClass->setPrimaryTable(array('name' => $tableName)); | |
46 | ||
47 | $metaDataEvent = new LoadClassMetadataEventArgs($metaClass, $em); | |
48 | ||
49 | $this->assertEquals($tableNameExpected, $metaDataEvent->getClassMetadata()->getTableName()); | |
50 | ||
51 | $subscriber->loadClassMetadata($metaDataEvent); | |
52 | ||
53 | $this->assertEquals($finalTableName, $metaDataEvent->getClassMetadata()->getTableName()); | |
54 | $this->assertEquals($finalTableNameQuoted, $metaDataEvent->getClassMetadata()->getQuotedTableName($platform)); | |
55 | } | |
56 | ||
57 | /** | |
58 | * @dataProvider dataForPrefix | |
59 | */ | |
60 | public function testSubscribedEvents($prefix, $entityName, $tableName, $tableNameExpected, $finalTableName, $finalTableNameQuoted, $platform) | |
61 | { | |
62 | $em = $this->getMockBuilder('Doctrine\ORM\EntityManager') | |
63 | ->disableOriginalConstructor() | |
64 | ->getMock(); | |
65 | ||
66 | $metaClass = new ClassMetadata($entityName); | |
67 | $metaClass->setPrimaryTable(array('name' => $tableName)); | |
68 | ||
69 | $metaDataEvent = new LoadClassMetadataEventArgs($metaClass, $em); | |
70 | ||
71 | $subscriber = new TablePrefixSubscriber($prefix); | |
72 | ||
73 | $evm = new EventManager(); | |
74 | $evm->addEventSubscriber($subscriber); | |
75 | ||
76 | $evm->dispatchEvent('loadClassMetadata', $metaDataEvent); | |
77 | ||
78 | $this->assertEquals($finalTableName, $metaDataEvent->getClassMetadata()->getTableName()); | |
79 | $this->assertEquals($finalTableNameQuoted, $metaDataEvent->getClassMetadata()->getQuotedTableName($platform)); | |
80 | } | |
81 | ||
82 | public function testPrefixManyToMany() | |
83 | { | |
84 | $em = $this->getMockBuilder('Doctrine\ORM\EntityManager') | |
85 | ->disableOriginalConstructor() | |
86 | ->getMock(); | |
87 | ||
88 | $subscriber = new TablePrefixSubscriber('yo_'); | |
89 | ||
90 | $metaClass = new ClassMetadata('Wallabag\UserBundle\Entity\Entry'); | |
91 | $metaClass->setPrimaryTable(array('name' => 'entry')); | |
92 | $metaClass->mapManyToMany(array( | |
93 | 'fieldName' => 'tags', | |
94 | 'joinTable' => array('name' => null, 'schema' => null), | |
95 | 'targetEntity' => 'Tag', | |
96 | 'mappedBy' => null, | |
97 | 'inversedBy' => 'entries', | |
98 | 'cascade' => array('persist'), | |
99 | 'indexBy' => null, | |
100 | 'orphanRemoval' => false, | |
101 | 'fetch' => 2, | |
102 | )); | |
103 | ||
104 | $metaDataEvent = new LoadClassMetadataEventArgs($metaClass, $em); | |
105 | ||
106 | $this->assertEquals('entry', $metaDataEvent->getClassMetadata()->getTableName()); | |
107 | ||
108 | $subscriber->loadClassMetadata($metaDataEvent); | |
109 | ||
110 | $this->assertEquals('yo_entry', $metaDataEvent->getClassMetadata()->getTableName()); | |
111 | $this->assertEquals('yo_entry_tag', $metaDataEvent->getClassMetadata()->associationMappings['tags']['joinTable']['name']); | |
112 | $this->assertEquals('yo_entry', $metaDataEvent->getClassMetadata()->getQuotedTableName(new \Doctrine\DBAL\Platforms\MySqlPlatform())); | |
113 | } | |
114 | } |