Merge pull request #25 from davedevelopment/better-wrapping

Wrap individual function calls
hamcrest · May 11, 2015 · b37020a · b37020a
2 parents 18ec485 + d2be20f
commit b37020a
Show file tree

Hide file tree

Showing 4 changed files with 164 additions and 78 deletions.
diff --git a/generator/GlobalFunctionFile.php b/generator/GlobalFunctionFile.php
@@ -30,4 +30,13 @@ public function build()
         $this->addCode($this->functions);
         $this->addPart('functions_footer');
     }
+
+    public function generateFactoryCall(FactoryCall $call)
+    {
+        $code = "if (!function_exists('{$call->getName()}')) {";
+        $code.= parent::generateFactoryCall($call);
+        $code.= "}\n";
+
+        return $code;
+    }
 }
diff --git a/generator/parts/functions_footer.txt b/generator/parts/functions_footer.txt
@@ -1 +0,0 @@
-}

diff --git a/generator/parts/functions_header.txt b/generator/parts/functions_header.txt
@@ -21,3 +21,4 @@ if (!function_exists('assertThat')) {
             $args
         );
     }
+}
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,3 +21,4 @@ if (!function_exists('assertThat')) { @@
                 $args
             );
         }
+    }