fix repository create model

alphasoft-fr · Apr 17, 2024 · 5733501 · 5733501
1 parent 4c28b42
commit 5733501
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 28 deletions.
diff --git a/src/Repository/Repository.php b/src/Repository/Repository.php
@@ -215,9 +215,6 @@ final protected function mapPropertiesToColumn(array $arguments): array
 
     final protected function createModel(array $data): object
     {
-        /**
-         * @var class-string<AsEntity> $entityName
-         */
         $entityName = $this->getEntityName();
         $primaryKeyValue = $data[$entityName::getPrimaryKeyColumn()];
         $cacheKey = $entityName.$primaryKeyValue;
@@ -228,14 +225,14 @@ final protected function createModel(array $data): object
             $this->cache->set($entity->_getKey(), $entity);
         }
 
-        if ($entity instanceof AsEntity) {
+        if (method_exists($entity, 'hydrate')) {
             $entity->hydrate($data);
         }
 
-        /**
-         * @var object$entity
-         */
-        $entity->setEntityManager($this->manager);
+        if (method_exists($entity, 'setEntityManager')) {
+            $entity->setEntityManager($this->manager);
+        }
+
         return $entity;
     }
 

diff --git a/tests/Repository/CommentRepository.php b/tests/Repository/CommentRepository.php