Merge pull request #47 from cabello/patch-3

Add support to dynamic helpers
instaclick · Jan 8, 2014 · 7717a13 · 7717a13
2 parents a3f37a7 + 95487c1
commit 7717a13
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/Test/Helper/AbstractHelper.php b/Test/Helper/AbstractHelper.php
@@ -24,7 +24,12 @@ abstract class AbstractHelper
      *
      * @param \IC\Bundle\Base\TestBundle\Test\Functional\WebTestCase $testCase
      */
-    public function __construct(WebTestCase $testCase)
+    public function __construct(WebTestCase $testCase = null)
+    {
+        $this->testCase = $testCase;
+    }
+
+    public function setTestCase(WebTestCase $testCase)
     {
         $this->testCase = $testCase;
     }

diff --git a/Test/WebTestCase.php b/Test/WebTestCase.php
@@ -75,7 +75,18 @@ public function getHelper($name)
     {
         $helperClass = $this->helperList->get($name);
 
-        return new $helperClass($this);
+        if ($helperClass) {
+            return new $helperClass($this);
+        }
+
+        $container     = $this->getClient()->getContainer();
+        $helperService = $container->get($name);
+
+        if ($helperService) {
+            $helperService->setTestCase($this);
+
+            return $helperService;
+        }
     }
 
     /**