From 246e8dc4cc920417ab5f75045c3ebbbe6abe4fc7 Mon Sep 17 00:00:00 2001
From: Cyrille Bourgois <cyrille.bourgois@gmail.com>
Date: Tue, 30 Dec 2014 14:50:48 +0100
Subject: [PATCH] Fix Doctrine ODM Support

Changes Faker\ORM\Doctrine\EntityPopulator to manage Doctrine ODM Documents.

Using the Class Doctrine\Common\Persistence\Mapping\ClassMetadata instead of Doctrine\ORM\Mapping\ClassMetadata

Managing ODM ReferenceOne as ORM One-To-One relations
(\Doctrine\ODM\MongoDB\Mapping\ClassMetadata does not have the getAssociationMappings function, I retrieve associationMappings using the exposed public var)

Changes generateId's signature to avoid using Doctrine\ORM\EntityManagerInterface; uses Doctrine\Common\Persistence\ObjectManager instead.
---
 src/Faker/ORM/Doctrine/EntityPopulator.php | 40 ++++++++++++++--------
 1 file changed, 26 insertions(+), 14 deletions(-)

diff --git a/src/Faker/ORM/Doctrine/EntityPopulator.php b/src/Faker/ORM/Doctrine/EntityPopulator.php
index 5ed8acbb01..e3bff11a17 100644
--- a/src/Faker/ORM/Doctrine/EntityPopulator.php
+++ b/src/Faker/ORM/Doctrine/EntityPopulator.php
@@ -3,8 +3,7 @@
 namespace Faker\ORM\Doctrine;
 
 use Doctrine\Common\Persistence\ObjectManager;
-use Doctrine\ORM\EntityManagerInterface;
-use Doctrine\ORM\Mapping\ClassMetadata;
+use Doctrine\Common\Persistence\Mapping\ClassMetadata;
 
 /**
  * Service class for populating a table through a Doctrine Entity class.
@@ -100,13 +99,26 @@ public function guessColumnFormatters(\Faker\Generator $generator)
             $relatedClass = $this->class->getAssociationTargetClass($assocName);
 
             $unique = $optional = false;
-            $mappings = $this->class->getAssociationMappings();
-            foreach ($mappings as $mapping) {
-                if ($mapping['targetEntity'] == $relatedClass) {
-                    if ($mapping['type'] == ClassMetadata::ONE_TO_ONE) {
-                        $unique = true;
-                        $optional = isset($mapping['joinColumns'][0]['nullable']) ? $mapping['joinColumns'][0]['nullable'] : false;
-                        break;
+            if ($this->class instanceof \Doctrine\ORM\Mapping\ClassMetadata) {
+                $mappings = $this->class->getAssociationMappings();
+                foreach ($mappings as $mapping) {
+                    if ($mapping['targetEntity'] == $relatedClass) {
+                        if ($mapping['type'] == \Doctrine\ORM\Mapping\ClassMetadata::ONE_TO_ONE) {
+                            $unique = true;
+                            $optional = isset($mapping['joinColumns'][0]['nullable']) ? $mapping['joinColumns'][0]['nullable'] : false;
+                            break;
+                        }
+                    }
+                }
+            }  elseif ( $this->class instanceof \Doctrine\ODM\MongoDB\Mapping\ClassMetadata) {
+                $mappings = $this->class->associationMappings;
+                foreach ($mappings as $mapping) {
+                    if ($mapping['targetDocument'] == $relatedClass) {
+                        if ($mapping['type'] == \Doctrine\ODM\MongoDB\Mapping\ClassMetadata::ONE && $mapping['association'] == \Doctrine\ODM\MongoDB\Mapping\ClassMetadata::REFERENCE_ONE ) {
+                            $unique = true;
+                            $optional = isset($mapping['nullable']) ? $mapping['nullable'] : false;
+                            break;
+                        }
                     }
                 }
             }
@@ -173,16 +185,16 @@ private function callMethods($obj, $insertedEntities)
         }
     }
 
-    private function generateId($obj, $column, EntityManagerInterface $manager)
+    private function generateId($obj, $column, ObjectManager $manager)
     {
-        /* @var $repository \Doctrine\ORM\EntityRepository */
+        /* @var $repository \Doctrine\Common\Persistence\ObjectRepository */
         $repository = $manager->getRepository(get_class($obj));
         $result = $repository->createQueryBuilder('e')
                 ->select(sprintf('e.%s', $column))
                 ->getQuery()
-                ->getResult();
-        $ids = array_map('current', $result);
-
+                ->execute();
+        $ids = array_map('current', $result->toArray());
+        
         $id = null;
         do {
             $id = rand();