[#14349] - Correcting tests

phalcon · Sep 7, 2019 · 6000b5a · 6000b5a
1 parent a185740
commit 6000b5a
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/unit/Flash/Direct/Helper/FlashBase.php b/tests/unit/Flash/Direct/Helper/FlashBase.php
@@ -23,6 +23,7 @@ class FlashBase
     private $notImplicit = false;
     private $notHtml     = false;
     private $classes     = null;
+    private $escaper     = null;
 
     private $default = [
         'success' => 'successMessage',
@@ -33,8 +34,7 @@ class FlashBase
 
     public function _before(UnitTester $I)
     {
-        $this->newDi();
-        $this->setDiEscaper();
+        $this->escaper = $this->newEscaper();
     }
 
     /**
@@ -65,7 +65,9 @@ public function testFlashDirectImplicitFlushHtml(UnitTester $I)
      */
     private function stringTest(UnitTester $I, string $function)
     {
-        $flash   = new Direct($this->classes);
+        $flash   = new Direct($this->escaper);
+        $flash->setClasses($this->classes);
+
         $message = 'sample message';
 
         if ($this->notHtml) {
@@ -214,7 +216,8 @@ public function testFlashDirectNoImplicitFlushNoHtml(UnitTester $I)
      */
     public function testFlashDirectWithAutoEscaping(UnitTester $I)
     {
-        $flash = new Direct($this->classes);
+        $flash   = new Direct($this->escaper);
+        $flash->setClasses($this->classes);
 
         $flash->setAutomaticHtml(false);
         $flash->setImplicitFlush(false);