Update Entity.php

dracony · web-flow · commit 5f5d05761977 · 2017-05-15T16:56:42.000-07:00
diff --git a/src/PHPixie/ORM/Wrappers/Model/Entity.php b/src/PHPixie/ORM/Wrappers/Model/Entity.php
@@ -67,6 +67,11 @@ public function __set($name, $value)
     
     public function __call($method, $params)
     {
-        return $this->entity->__call($method, $params);
+        $result = $this->entity->__call($method, $params);
+        if($result === $this->entity) {
+            return $this;
+        }
+        
+        return $result;
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,11 @@ public function __set($name, $value)`
`67`	`67`
`68`	`68`	`public function __call($method, $params)`
`69`	`69`	`{`
`70`		`- return $this->entity->__call($method, $params);`
	`70`	`+ $result = $this->entity->__call($method, $params);`
	`71`	`+ if($result === $this->entity) {`
	`72`	`+ return $this;`
	`73`	`+ }`
	`74`	`+`
	`75`	`+ return $result;`
`71`	`76`	`}`
`72`	`77`	`}`