Merge remote-tracking branch 'origin/MAGETWO-52105' into MAGETWO-52835

Author: NadiyaS

lib/internal/Magento/Framework/EntityManager/MetadataPool.php

@@ -106,4 +106,15 @@ public function getHydrator($entityType)
         $objectManager = \Magento\Framework\App\ObjectManager::getInstance();
         return $objectManager->get(HydratorPool::class)->getHydrator($entityType);
     }
+
+    /**
+     * Check if entity type configuration was set to metadata
+     *
+     * @param string $entityType
+     * @return bool
+     */
+    public function hasConfiguration($entityType)
+    {
+        return isset($this->metadata[$entityType]);
+    }
 }

lib/internal/Magento/Framework/EntityManager/TypeResolver.php

@@ -10,6 +10,11 @@
  */
 class TypeResolver
 {
+    /**
+     * @var MetadataPool
+     */
+    private $metadataPool;
+
     /**
      * @var array
      */
@@ -18,6 +23,15 @@ class TypeResolver
         \Magento\SalesRule\Model\Rule\Interceptor::class => \Magento\SalesRule\Api\Data\RuleInterface::class
     ];
 
+    /**
+     * TypeResolver constructor.
+     * @param MetadataPool $metadataPool
+     */
+    public function __construct(MetadataPool $metadataPool)
+    {
+        $this->metadataPool = $metadataPool;
+    }
+
     /**
      * @param object $type
      * @return string
@@ -45,6 +59,12 @@ public function resolve($type)
             throw new \Exception('Unable to determine data interface for ' . $className);
         }
 
-        return reset($dataInterfaces);
+        foreach ($dataInterfaces as $dataInterface) {
+            if ($this->metadataPool->hasConfiguration($dataInterface)) {
+                $this->typeMapping[$className] = $dataInterface;
+            }
+        }
+
+        return $this->typeMapping[$className];
     }
 }