File: PrePersistEventTest.php

package info (click to toggle)
doctrine 3.5.2%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 11,556 kB
  • sloc: php: 108,620; xml: 1,340; makefile: 35; sh: 14
file content (93 lines) | stat: -rw-r--r-- 2,590 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
<?php

declare(strict_types=1);

namespace Doctrine\Tests\ORM\Functional;

use Doctrine\ORM\Event\PrePersistEventArgs;
use Doctrine\ORM\Events;
use Doctrine\ORM\Mapping\Column;
use Doctrine\ORM\Mapping\Entity;
use Doctrine\ORM\Mapping\GeneratedValue;
use Doctrine\ORM\Mapping\Id;
use Doctrine\ORM\Mapping\ManyToOne;
use Doctrine\Tests\OrmFunctionalTestCase;

use function uniqid;

class PrePersistEventTest extends OrmFunctionalTestCase
{
    protected function setUp(): void
    {
        parent::setUp();

        $this->createSchemaForModels(
            EntityWithUnmappedEntity::class,
            EntityWithCascadeAssociation::class,
        );
    }

    public function testCallingPersistInPrePersistHook(): void
    {
        $entityWithUnmapped = new EntityWithUnmappedEntity();
        $entityWithCascade  = new EntityWithCascadeAssociation();

        $entityWithUnmapped->unmapped = $entityWithCascade;
        $entityWithCascade->cascaded  = $entityWithUnmapped;

        $this->_em->getEventManager()->addEventListener(Events::prePersist, new PrePersistUnmappedPersistListener());
        $this->_em->persist($entityWithUnmapped);

        $this->assertTrue($this->_em->getUnitOfWork()->isScheduledForInsert($entityWithCascade));
        $this->assertTrue($this->_em->getUnitOfWork()->isScheduledForInsert($entityWithUnmapped));
    }
}

class PrePersistUnmappedPersistListener
{
    public function prePersist(PrePersistEventArgs $args): void
    {
        $object = $args->getObject();

        if ($object instanceof EntityWithUnmappedEntity) {
            $uow = $args->getObjectManager()->getUnitOfWork();

            if ($object->unmapped && ! $uow->isInIdentityMap($object->unmapped) && ! $uow->isScheduledForInsert($object->unmapped)) {
                $args->getObjectManager()->persist($object->unmapped);
            }
        }
    }
}

#[Entity]
class EntityWithUnmappedEntity
{
    #[Id]
    #[Column(type: 'string', length: 255)]
    #[GeneratedValue(strategy: 'NONE')]
    public string $id;

    public EntityWithCascadeAssociation|null $unmapped = null;

    public function __construct()
    {
        $this->id = uniqid(self::class, true);
    }
}

#[Entity]
class EntityWithCascadeAssociation
{
    #[Id]
    #[Column(type: 'string', length: 255)]
    #[GeneratedValue(strategy: 'NONE')]
    public string $id;

    #[ManyToOne(targetEntity: EntityWithUnmappedEntity::class, cascade: ['persist'])]
    public EntityWithUnmappedEntity|null $cascaded = null;

    public function __construct()
    {
        $this->id = uniqid(self::class, true);
    }
}