Merge pull request #430 from alcaeus/fix-embedded-document-proxy-generation

Exclude embedded documents from proxy generation

Author: alcaeus

CacheWarmer/ProxyCacheWarmer.php

@@ -15,6 +15,8 @@
 namespace Doctrine\Bundle\MongoDBBundle\CacheWarmer;
 
 use Doctrine\Common\Proxy\AbstractProxyFactory;
+use Doctrine\ODM\MongoDB\DocumentManager;
+use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 use Symfony\Component\HttpKernel\CacheWarmer\CacheWarmerInterface;
 
@@ -72,8 +74,20 @@ public function warmUp($cacheDir)
         $registry = $this->container->get('doctrine_mongodb');
         foreach ($registry->getManagers() as $dm) {
             /* @var $dm \Doctrine\ODM\MongoDB\DocumentManager */
-            $classes = $dm->getMetadataFactory()->getAllMetadata();
+            $classes = $this->getClassesForProxyGeneration($dm);
             $dm->getProxyFactory()->generateProxyClasses($classes);
         }
     }
+
+    /**
+     * @param DocumentManager $dm
+     *
+     * @return ClassMetadata[]
+     */
+    private function getClassesForProxyGeneration(DocumentManager $dm)
+    {
+        return array_filter($dm->getMetadataFactory()->getAllMetadata(), function (ClassMetadata $metadata) {
+            return !$metadata->isEmbeddedDocument && !$metadata->isMappedSuperclass;
+        });
+    }
 }

Tests/CacheWarmer/ProxyCacheWarmerTest.php

@@ -60,7 +60,11 @@ public function testWarmerNotOptional()
 
     public function testWarmerExecuted()
     {
-        $this->proxyMock->expects($this->once())->method('generateProxyClasses');
+        $this->proxyMock
+            ->expects($this->once())
+            ->method('generateProxyClasses')
+            ->with($this->countOf(1))
+        ;
         $this->warmer->warmUp('meh');
     }
 
@@ -82,4 +86,4 @@ public function provideWarmerNotExecuted()
             [ Configuration::AUTOGENERATE_FILE_NOT_EXISTS ],
         ];
     }
-}
+}