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
  );
  }
+}