Fixed generated factories

enumag · Sep 6, 2014 · 604d4f3 · 604d4f3
1 parent c7d046e
commit 604d4f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/DI/ContainerBuilder.php b/src/DI/ContainerBuilder.php
@@ -306,6 +306,9 @@ private function resolveImplement(ServiceDefinition $def, $name)
 
 		if (!$def->parameters) {
 			$ctorParams = array();
+			if ((!$def->factory || (!$def->factory->entity && !$def->factory->arguments)) && $def->class) {
+				$def->setFactory($def->class);
+			}
 			if ($def->factory && !$def->factory->arguments && ($class = $this->resolveEntityClass($def->factory, array($name => 1)))
 				&& ($ctor = Reflection\ClassType::from($class)->getConstructor())
 			) {