|
11 | 11 | use Doctrine\ORM\Mapping\ClassMetadata; |
12 | 12 | use Doctrine\ORM\Mapping\ClassMetadataFactory; |
13 | 13 | use Doctrine\ORM\Mapping\Column; |
| 14 | +use Doctrine\ORM\Mapping\DiscriminatorColumn; |
14 | 15 | use Doctrine\ORM\Mapping\DiscriminatorMap; |
15 | 16 | use Doctrine\ORM\Mapping\Driver\AnnotationDriver; |
16 | 17 | use Doctrine\ORM\Mapping\Entity; |
|
33 | 34 | use Doctrine\Tests\Models\DirectoryTree\Directory; |
34 | 35 | use Doctrine\Tests\Models\DirectoryTree\File; |
35 | 36 | use Doctrine\Tests\Models\ECommerce\ECommerceCart; |
| 37 | +use Generator; |
36 | 38 |
|
37 | 39 | class AnnotationDriverTest extends AbstractMappingDriverTest |
38 | 40 | { |
@@ -274,6 +276,30 @@ public function testAttributeOverridesMappingWithTrait(): void |
274 | 276 | self::assertArrayHasKey('example_trait_bar_id', $metadataWithoutOverride->associationMappings['bar']['joinColumnFieldNames']); |
275 | 277 | self::assertArrayHasKey('example_entity_overridden_bar_id', $metadataWithOverride->associationMappings['bar']['joinColumnFieldNames']); |
276 | 278 | } |
| 279 | + |
| 280 | + /** |
| 281 | + * @psalm-param class-string $class |
| 282 | + * |
| 283 | + * @dataProvider provideDiscriminatorColumnTestcases |
| 284 | + */ |
| 285 | + public function testLengthForDiscriminatorColumn(string $class, int $expectedLength): void |
| 286 | + { |
| 287 | + $factory = $this->createClassMetadataFactory(); |
| 288 | + |
| 289 | + $metadata = $factory->getMetadataFor($class); |
| 290 | + |
| 291 | + self::assertNotNull($metadata->discriminatorColumn); |
| 292 | + self::assertArrayHasKey('length', $metadata->discriminatorColumn); |
| 293 | + self::assertSame($expectedLength, $metadata->discriminatorColumn['length']); |
| 294 | + } |
| 295 | + |
| 296 | + public function provideDiscriminatorColumnTestcases(): Generator |
| 297 | + { |
| 298 | + yield [DiscriminatorColumnWithNullLength::class, 255]; |
| 299 | + yield [DiscriminatorColumnWithNoLength::class, 255]; |
| 300 | + yield [DiscriminatorColumnWithZeroLength::class, 0]; |
| 301 | + yield [DiscriminatorColumnWithNonZeroLength::class, 60]; |
| 302 | + } |
277 | 303 | } |
278 | 304 |
|
279 | 305 | /** |
@@ -438,3 +464,83 @@ class AnnotationSLCFoo |
438 | 464 | */ |
439 | 465 | public $id; |
440 | 466 | } |
| 467 | + |
| 468 | +/** |
| 469 | + * @Entity |
| 470 | + * @InheritanceType("SINGLE_TABLE") |
| 471 | + * @DiscriminatorColumn( |
| 472 | + * name="type", |
| 473 | + * type="string", |
| 474 | + * length=0, |
| 475 | + * columnDefinition="enum('region','airport','station','poi') NOT NULL", |
| 476 | + * ), |
| 477 | + * @DiscriminatorMap({"s"="SuperEntity", "c"="ChildEntity"}) |
| 478 | + */ |
| 479 | +class DiscriminatorColumnWithZeroLength |
| 480 | +{ |
| 481 | + /** |
| 482 | + * @var int |
| 483 | + * @Id |
| 484 | + * @Column |
| 485 | + */ |
| 486 | + public $id; |
| 487 | +} |
| 488 | + |
| 489 | +/** |
| 490 | + * @Entity |
| 491 | + * @InheritanceType("SINGLE_TABLE") |
| 492 | + * @DiscriminatorColumn( |
| 493 | + * name="type", |
| 494 | + * type="string", |
| 495 | + * ), |
| 496 | + * @DiscriminatorMap({"s"="SuperEntity", "c"="ChildEntity"}) |
| 497 | + */ |
| 498 | +class DiscriminatorColumnWithNoLength |
| 499 | +{ |
| 500 | + /** |
| 501 | + * @var int |
| 502 | + * @Id |
| 503 | + * @Column |
| 504 | + */ |
| 505 | + public $id; |
| 506 | +} |
| 507 | + |
| 508 | +/** |
| 509 | + * @Entity |
| 510 | + * @InheritanceType("SINGLE_TABLE") |
| 511 | + * @DiscriminatorColumn( |
| 512 | + * name="type", |
| 513 | + * type="string", |
| 514 | + * length=60, |
| 515 | + * ), |
| 516 | + * @DiscriminatorMap({"s"="SuperEntity", "c"="ChildEntity"}) |
| 517 | + */ |
| 518 | +class DiscriminatorColumnWithNonZeroLength |
| 519 | +{ |
| 520 | + /** |
| 521 | + * @var int |
| 522 | + * @Id |
| 523 | + * @Column |
| 524 | + */ |
| 525 | + public $id; |
| 526 | +} |
| 527 | + |
| 528 | +/** |
| 529 | + * @Entity |
| 530 | + * @InheritanceType("SINGLE_TABLE") |
| 531 | + * @DiscriminatorColumn( |
| 532 | + * name="type", |
| 533 | + * type="string", |
| 534 | + * length=null, |
| 535 | + * ), |
| 536 | + * @DiscriminatorMap({"s"="SuperEntity", "c"="ChildEntity"}) |
| 537 | + */ |
| 538 | +class DiscriminatorColumnWithNullLength |
| 539 | +{ |
| 540 | + /** |
| 541 | + * @var int |
| 542 | + * @Id |
| 543 | + * @Column |
| 544 | + */ |
| 545 | + public $id; |
| 546 | +} |
0 commit comments