Fluent interface for fillFromFields.

AthensFramework · Oct 4, 2016 · 88272fe · 88272fe
1 parent e610c20
commit 88272fe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/orm-wrapper/PropelObjectWrapper.php b/src/orm-wrapper/PropelObjectWrapper.php
@@ -171,7 +171,7 @@ protected function addBehaviorConstraintsToFields(array $fields)
      * Expects that $fields contains a set of $fieldName => $field pairs.
      *
      * @param FieldInterface[] $fields
-     * @return void
+     * @return ObjectWrapperInterface
      */
     public function fillFromFields(array $fields)
     {
@@ -199,6 +199,8 @@ public function fillFromFields(array $fields)
                 }
             }
         }
+
+        return $this;
     }
 
     /**

diff --git a/test/PropelObjectWrapperTest.php b/test/PropelObjectWrapperTest.php
@@ -2,6 +2,7 @@
 
 namespace Athens\Core\Test;
 
+use Athens\Core\ORMWrapper\ObjectWrapperInterface;
 use PHPUnit_Framework_TestCase;
 
 use Athens\Propel\ORMWrapper\PropelObjectWrapper;
@@ -391,9 +392,10 @@ public function testFillFromFields()
             $field->setValidatedData($values[$fieldName]);
         }
 
-        $wrappedTestClass->fillFromFields($fields);
+        $result = $wrappedTestClass->fillFromFields($fields);
 
         $this->assertEquals($values, $wrappedTestClass->getValues());
+        $this->assertInstanceOf(ObjectWrapperInterface::class, $result);
     }
 
     public function testSave()