Update ImmutableAssociationService.php

lube · web-flow · commit 3b4a3740030f · 2016-10-11T18:05:01.000-03:00
diff --git a/Lube/GeneratorBundle/Services/ImmutableAssociationService.php b/Lube/GeneratorBundle/Services/ImmutableAssociationService.php
@@ -32,10 +32,11 @@ public function markAssociationsReadOnly($entity)
             if ($metadata->isSingleValuedAssociation($association))
             {
                 $getter = Inflector::camelize('get_' . $association);
-                $associated = $entity->$getter();
-                
-                if ($associated) {
-                    $this->em->getUnitOfWork()->markReadOnly($associated);
+                if (method_exists ($entity, $getter)) {
+                    $associated = $entity->$getter();
+                    if ($associated) {
+                        $this->em->getUnitOfWork()->markReadOnly($associated);
+                    }
                 }
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -32,10 +32,11 @@ public function markAssociationsReadOnly($entity)`
`32`	`32`	`if ($metadata->isSingleValuedAssociation($association))`
`33`	`33`	`{`
`34`	`34`	`$getter = Inflector::camelize('get_' . $association);`
`35`		`- $associated = $entity->$getter();`
`36`		`-`
`37`		`- if ($associated) {`
`38`		`- $this->em->getUnitOfWork()->markReadOnly($associated);`
	`35`	`+ if (method_exists ($entity, $getter)) {`
	`36`	`+ $associated = $entity->$getter();`
	`37`	`+ if ($associated) {`
	`38`	`+ $this->em->getUnitOfWork()->markReadOnly($associated);`
	`39`	`+ }`
`39`	`40`	`}`
`40`	`41`	`}`
`41`	`42`	`}`