Merge pull request #401 from Charlie-Lucas/feature/1.5-allow-to-use-directly-the-entity-id-to-find-a-business-page

allow to pass directly the entity id in find BP method in BusinessPag…

Author: Leny BERNARD

Bundle/BusinessPageBundle/Repository/BusinessPageRepository.php

@@ -12,14 +12,19 @@
 class BusinessPageRepository extends EntityRepository
 {
     /**
-     * Find the pagePatterns of the business entity.
+     *  Find the pagePatterns of the business entity.
      *
-     * @param BusinessEntity $businessEntity
+     * @param BusinessTemplate $pattern
+     * @param object           $entity         | int $entityId
+     * @param BusinessEntity   $businessEntity
      *
-     * @return array The list of pagePatterns
+     * @return mixed
      */
     public function findPageByBusinessEntityAndPattern(BusinessTemplate $pattern, $entity, BusinessEntity $businessEntity)
     {
+        if (is_object($entity)) {
+            $entity = $entity->getId();
+        }
         $qb = $this->createQueryBuilder('BusinessPage');
         $qb->join('BusinessPage.entityProxy', 'proxy');
         $qb->join('BusinessPage.template', 'template');
@@ -30,7 +35,7 @@ public function findPageByBusinessEntityAndPattern(BusinessTemplate $pattern, $e
         $qb->andWhere('entity.id = :entityId');
 
         $qb->setParameter(':templateId', $pattern);
-        $qb->setParameter(':entityId', $entity->getId());
+        $qb->setParameter(':entityId', $entity);
         $result = $qb->getQuery()->getOneOrNullResult();
 
         return $result;