From d218b3e5a79cd538c2d3923616c66507e5a91ce6 Mon Sep 17 00:00:00 2001
From: Alan Poulain <contact@alanpoulain.eu>
Date: Mon, 15 Feb 2021 14:37:59 +0100
Subject: [PATCH] test: fix MongoDB tests for RPC

---
 .../TestBundle/DataTransformer/RPCOutputDataTransformer.php    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/Fixtures/TestBundle/DataTransformer/RPCOutputDataTransformer.php b/tests/Fixtures/TestBundle/DataTransformer/RPCOutputDataTransformer.php
index 7072137568c..54940515aa2 100644
--- a/tests/Fixtures/TestBundle/DataTransformer/RPCOutputDataTransformer.php
+++ b/tests/Fixtures/TestBundle/DataTransformer/RPCOutputDataTransformer.php
@@ -14,6 +14,7 @@
 namespace ApiPlatform\Core\Tests\Fixtures\TestBundle\DataTransformer;
 
 use ApiPlatform\Core\DataTransformer\DataTransformerInterface;
+use ApiPlatform\Core\Tests\Fixtures\TestBundle\Document\RPC as RPCDocument;
 use ApiPlatform\Core\Tests\Fixtures\TestBundle\Dto\RPCOutput;
 use ApiPlatform\Core\Tests\Fixtures\TestBundle\Entity\RPC;
 
@@ -32,6 +33,6 @@ public function transform($object, string $to, array $context = [])
      */
     public function supportsTransformation($object, string $to, array $context = []): bool
     {
-        return $object instanceof RPC && RPCOutput::class === $to;
+        return ($object instanceof RPC || $object instanceof RPCDocument) && RPCOutput::class === $to;
     }
 }