Skip to content

Commit

Permalink
Fix OneToManyPersister::deleteEntityCollection missing discriminator …
Browse files Browse the repository at this point in the history
…column/value. (doctrineGH-11500)
  • Loading branch information
gitbugr committed Jun 14, 2024
1 parent 858a1ad commit bceb75d
Show file tree
Hide file tree
Showing 2 changed files with 122 additions and 1 deletion.
14 changes: 13 additions & 1 deletion src/Persisters/Collection/OneToManyPersister.php
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
use Doctrine\Common\Collections\Criteria;
use Doctrine\DBAL\Exception as DBALException;
use Doctrine\DBAL\Types\Type;
use Doctrine\ORM\EntityNotFoundException;
use Doctrine\ORM\Mapping\MappingException;
use Doctrine\ORM\PersistentCollection;
use Doctrine\ORM\Utility\PersisterHelper;

Expand Down Expand Up @@ -166,7 +168,10 @@ public function loadCriteria(PersistentCollection $collection, Criteria $criteri
throw new BadMethodCallException('Filtering a collection by Criteria is not supported by this CollectionPersister.');
}

/** @throws DBALException */
/**
* @throws EntityNotFoundException
* @throws MappingException
*/
private function deleteEntityCollection(PersistentCollection $collection): int
{
$mapping = $collection->getMapping();
Expand All @@ -186,6 +191,13 @@ private function deleteEntityCollection(PersistentCollection $collection): int
$statement = 'DELETE FROM ' . $this->quoteStrategy->getTableName($targetClass, $this->platform)
. ' WHERE ' . implode(' = ? AND ', $columns) . ' = ?';

if ($targetClass->isInheritanceTypeSingleTable()) {
$discriminatorColumn = $targetClass->getDiscriminatorColumn();
$statement .= ' AND ' . $discriminatorColumn['name'] . ' = ?';
$parameters[] = $targetClass->discriminatorValue;
$types[] = $discriminatorColumn['type'];
}

$numAffected = $this->conn->executeStatement($statement, $parameters, $types);

assert(is_int($numAffected));
Expand Down
109 changes: 109 additions & 0 deletions tests/Tests/ORM/Functional/Ticket/GH11500Test.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,109 @@
<?php

declare(strict_types=1);

namespace Doctrine\Tests\ORM\Functional\Ticket;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\Common\Collections\Collection;
use Doctrine\ORM\Mapping as ORM;
use Doctrine\ORM\PersistentCollection;
use Doctrine\ORM\Persisters\Collection\OneToManyPersister;
use Doctrine\Tests\Mocks\ConnectionMock;
use Doctrine\Tests\OrmFunctionalTestCase;

use function array_pop;
use function assert;

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

$this->setUpEntitySchema([
GH11500AbstractTestEntity::class,
GH11500TestEntity::class,
GH11500TestEntityHolder::class,
]);
}

public function testDeleteOneToManyCollectionWithSingleTableInheritance(): void
{
$em = $this->getTestEntityManager();
$conn = $em->getConnection();
assert($conn instanceof ConnectionMock);

$testEntity = new GH11500TestEntity();
$testEntityHolder = new GH11500TestEntityHolder();

$testEntity->testEntityHolder = $testEntityHolder;
$testEntityHolder->testEntities->add($testEntity);

$em->persist($testEntity);
$em->persist($testEntityHolder);
$em->flush();

assert($testEntityHolder->testEntities instanceof PersistentCollection);

$persister = new OneToManyPersister($em);
$persister->delete($testEntityHolder->testEntities);

$updates = $conn->getExecuteStatements();
$lastStatement = array_pop($updates);

self::assertEquals('DELETE FROM one_to_many_single_table_inheritance_test_entities WHERE test_entity_holder_id = ? AND type = ?', $lastStatement['sql']);
self::assertEquals('test_entity', $lastStatement['params'][1] ?? null);
}
}



/**
* @ORM\Entity
* @ORM\Table(name="one_to_many_single_table_inheritance_test_entities")
* @ORM\InheritanceType("SINGLE_TABLE")
* @ORM\DiscriminatorColumn(name="type", type="string")
* @ORM\DiscriminatorMap({"test_entity"="GH11500TestEntity"})
*/
class GH11500AbstractTestEntity
{
/**
* @ORM\Id
* @ORM\Column(type="integer")
* @ORM\GeneratedValue
*/
public int $id;
}


/** @ORM\Entity */
class GH11500TestEntity extends GH11500AbstractTestEntity
{
/**
* @ORM\ManyToOne(targetEntity="GH11500TestEntityHolder", inversedBy="testEntities")
* @ORM\JoinColumn(name="test_entity_holder_id", referencedColumnName="id")
*/
public GH11500TestEntityHolder $testEntityHolder;
}

/** @ORM\Entity */
class GH11500TestEntityHolder
{
/**
* @ORM\Id
* @ORM\Column(type="integer")
* @ORM\GeneratedValue
*/
public int $id;

/**
* @ORM\OneToMany(targetEntity="GH11500TestEntity", mappedBy="testEntityHolder", orphanRemoval=true)
*/
public Collection $testEntities;

public function __construct()
{
$this->testEntities = new ArrayCollection();
}
}

0 comments on commit bceb75d

Please sign in to comment.