Messenger persist using context resource_class

Author: soyuka

src/Bridge/Symfony/Messenger/DataPersister.php

@@ -15,6 +15,7 @@
 
 use ApiPlatform\Core\Api\OperationType;
 use ApiPlatform\Core\DataPersister\ContextAwareDataPersisterInterface;
+use ApiPlatform\Core\Exception\ResourceClassNotFoundException;
 use ApiPlatform\Core\Metadata\Resource\Factory\ResourceMetadataFactoryInterface;
 use ApiPlatform\Core\Util\ClassInfoTrait;
 use Symfony\Component\Messenger\Envelope;
@@ -46,7 +47,12 @@ public function __construct(ResourceMetadataFactoryInterface $resourceMetadataFa
      */
     public function supports($data, array $context = []): bool
     {
-        $resourceMetadata = $this->resourceMetadataFactory->create($this->getObjectClass($data));
+        try {
+            $resourceMetadata = $this->resourceMetadataFactory->create($context['resource_class'] ?? $this->getObjectClass($data));
+        } catch (ResourceClassNotFoundException $e) {
+            return false;
+        }
+
         if (null !== $operationName = $context['collection_operation_name'] ?? $context['item_operation_name'] ?? null) {
             return true === $resourceMetadata->getTypedOperationAttribute(
                 $context['collection_operation_name'] ?? false ? OperationType::COLLECTION : OperationType::ITEM,

tests/Bridge/Symfony/Messenger/DataPersisterTest.php

@@ -15,9 +15,11 @@
 
 use ApiPlatform\Core\Bridge\Symfony\Messenger\DataPersister;
 use ApiPlatform\Core\Bridge\Symfony\Messenger\RemoveStamp;
+use ApiPlatform\Core\Exception\ResourceClassNotFoundException;
 use ApiPlatform\Core\Metadata\Resource\Factory\ResourceMetadataFactoryInterface;
 use ApiPlatform\Core\Metadata\Resource\ResourceMetadata;
 use ApiPlatform\Core\Tests\Fixtures\TestBundle\Entity\Dummy;
+use ApiPlatform\Core\Tests\Fixtures\TestBundle\Entity\DummyCar;
 use PHPUnit\Framework\TestCase;
 use Prophecy\Argument;
 use Symfony\Component\Messenger\Envelope;
@@ -38,6 +40,17 @@ public function testSupport()
         $this->assertTrue($dataPersister->supports(new Dummy()));
     }
 
+    public function testSupportWithContext()
+    {
+        $metadataFactoryProphecy = $this->prophesize(ResourceMetadataFactoryInterface::class);
+        $metadataFactoryProphecy->create(Dummy::class)->willReturn(new ResourceMetadata(null, null, null, null, null, ['messenger' => true]));
+        $metadataFactoryProphecy->create(DummyCar::class)->willThrow(new ResourceClassNotFoundException());
+
+        $dataPersister = new DataPersister($metadataFactoryProphecy->reveal(), $this->prophesize(MessageBusInterface::class)->reveal());
+        $this->assertTrue($dataPersister->supports(new DummyCar(), ['resource_class' => Dummy::class]));
+        $this->assertFalse($dataPersister->supports(new DummyCar()));
+    }
+
     public function testPersist()
     {
         $dummy = new Dummy();