Allow setting a decorator on a form

packaged · May 10, 2019 · e86459d · e86459d
1 parent 9e2dace
commit e86459d
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 13 deletions.
diff --git a/src/DataHandlers/AbstractDataHandler.php b/src/DataHandlers/AbstractDataHandler.php
@@ -294,9 +294,8 @@ public function setDefaultValue($defaultValue)
     return $this;
   }
 
-  public function __toString()
+  public function render(): string
   {
-    return $this->getDecorator()->produceSafeHTML()->getContent();
+    return $this->getDecorator()->render();
   }
-
 }
diff --git a/src/DataHandlers/Interfaces/DataHandler.php b/src/DataHandlers/Interfaces/DataHandler.php
@@ -97,4 +97,6 @@ public function addError(ValidationException ...$errors);
    * @return DataHandler
    */
   public function clearErrors();
+
+  public function render(): string;
 }
diff --git a/src/Form/Form.php b/src/Form/Form.php
@@ -207,7 +207,7 @@ public function getSubmitDecorator(): ?Decorator
 
   public function render(): string
   {
-    return (string)$this->produceSafeHTML();
+    return $this->produceSafeHTML()->getContent();
   }
 
   public function produceSafeHTML(): SafeHtml
@@ -224,6 +224,12 @@ public function getDecorator(): FormDecorator
     return $this->_decorator->setForm($this);
   }
 
+  public function setDecorator(FormDecorator $decorator)
+  {
+    $this->_decorator = $decorator;
+    return $this;
+  }
+
   protected function _defaultDecorator(): FormDecorator
   {
     return new DefaultFormDecorator();

diff --git a/tests/Decorators/FormCustomDecoratorTest.php b/tests/Decorators/FormCustomDecoratorTest.php
@@ -3,13 +3,15 @@
 namespace Packaged\Tests\Form\Decorators;
 
 use Packaged\Tests\Form\Supporting\CustomDecoratorForm;
+use Packaged\Tests\Form\Supporting\CustomFormDecorator;
 use PHPUnit\Framework\TestCase;
 
 class FormCustomDecoratorTest extends TestCase
 {
   public function testCustomDecorator()
   {
-    $form = new CustomDecoratorForm('abc');
+    $form = new CustomDecoratorForm('csrf');
+    $form->setDecorator(new CustomFormDecorator());
     $output = $form->produceSafeHTML()->getContent();
     $this->assertContains('name="email"', $output);
     $this->assertContains('method="post"><div class="hidden"><input type="hidden"', $output);

diff --git a/tests/Supporting/CustomDecoratorForm.php b/tests/Supporting/CustomDecoratorForm.php
@@ -3,7 +3,6 @@
 
 use Packaged\Form\Csrf\CsrfForm;
 use Packaged\Form\DataHandlers\TextDataHandler;
-use Packaged\Form\Form\Interfaces\FormDecorator;
 
 class CustomDecoratorForm extends CsrfForm
 {
@@ -16,9 +15,4 @@ protected function _initDataHandlers()
     $this->name = new TextDataHandler();
     $this->email = new TextDataHandler();
   }
-
-  protected function _defaultDecorator(): FormDecorator
-  {
-    return new CustomFormDecorator();
-  }
 }
diff --git a/tests/Supporting/CustomFormDecorator.phtml b/tests/Supporting/CustomFormDecorator.phtml
@@ -10,6 +10,6 @@ use Packaged\Tests\Form\Supporting\CustomFormDecorator;
 $form = $this->getForm();
 ?>
 <div class="hidden"><?= implode('', $this->_getHiddenDecorators()); ?></div>
-<?= $form->name; ?>
-<?= $form->email; ?>
+<?= $form->name->render(); ?>
+<?= $form->email->render(); ?>
 <?= $this->getForm()->getSubmitDecorator(); ?>