Patch FactoryMuffin::generate() method to support dynamic setter and getter

with the OCP\AppFramework\Db\Entity class

Signed-off-by: Tortue Torche <tortuetorche@users.noreply.github.com>

2 files changed, 38 insertions, 3 deletions

diff --git a/tests/Unit/FactoryMuffin.php b/tests/Unit/FactoryMuffin.php
new file mode 100644
index 00000000..a8fb7d59
--- /dev/null
+++ b/tests/Unit/FactoryMuffin.php
@@ -0,0 +1,36 @@
+<?php
+
+namespace OCA\Polls\Tests\Unit;
+
+use League\FactoryMuffin\FactoryMuffin as OriginalFactoryMuffin;
+use OCP\AppFramework\Db\Entity;
+
+class FactoryMuffin extends OriginalFactoryMuffin
+{
+    /**
+     * Generate and set the model attributes.
+     * NOTE: Patch the original method to support dynamic setter and getter
+    *        of the OCP\AppFramework\Db\Entity class
+     *
+     * @param object $model The model instance.
+     * @param array  $attr  The model attributes.
+     *
+     * @return void
+     */
+    protected function generate($model, array $attr = [])
+    {
+        foreach ($attr as $key => $kind) {
+            $value = $this->factory->generate($kind, $model, $this);
+
+            $setter = 'set'.ucfirst(static::camelize($key));
+            // check if there is a setter and use it instead
+            if ($model instanceof Entity && is_callable([$model, $setter])) {
+                $model->$setter($value);
+            } elseif (method_exists($model, $setter) && is_callable([$model, $setter])) {
+                $model->$setter($value);
+            } else {
+                $model->$key = $value;
+            }
+        }
+    }
+}
diff --git a/tests/Unit/UnitTestCase.php b/tests/Unit/UnitTestCase.php
index 6be39d40..2edf4af1 100644
--- a/tests/Unit/UnitTestCase.php
+++ b/tests/Unit/UnitTestCase.php
@@ -23,10 +23,9 @@
 
 namespace OCA\Polls\Tests\Unit;
 
-use League\FactoryMuffin\FactoryMuffin;
-use PHPUnit\Framework\TestCase as PHPUnit_Framework_TestCase;
+use PHPUnit\Framework\TestCase;
 
-abstract class UnitTestCase extends PHPUnit_Framework_TestCase {
+abstract class UnitTestCase extends TestCase {
 
 	/** @var FactoryMuffin */
 	protected $fm;