mock function has been refactored

Author: diloabininyeri

src/MockFunction.php

@@ -4,6 +4,7 @@
 
 use Closure;
 use ReflectionObject;
+use function call_user_func;
 
 /**
  *
@@ -53,22 +54,57 @@ public function __construct()
     public function add(string $name, mixed $returnValue): void
     {
         if (!($returnValue instanceof Closure)) {
-            $returnValue=static fn() => $returnValue;
+            $returnValue = static fn() => $returnValue;
         }
         $this->functions[$name] = $returnValue;
     }
 
+    /**
+     * @noinspection PhpUnused
+     * @param string $name
+     * @param mixed $returnValue
+     * @return void
+     */
+    public function addIfNotDefined(string $name, mixed $returnValue): void
+    {
+        if (!$this->has($name)) {
+            $this->add($name, $returnValue);
+        }
+    }
+
+    /**
+     * @param string $name
+     * @return bool
+     */
+    public function has(string $name): bool
+    {
+        return isset($this->functions[$name]);
+    }
+
     /**
      * @param string $name
      * @param array $args
      * @return mixed
      */
     private function call(string $name, array $args): mixed
     {
-        $func = $this->functions[$name];
-        $returnValue = $func(...$args);
-        $this->incrementCount($name);
-        return $returnValue;
+        return $this->executeWithEffect(
+            call_user_func($this->functions[$name], ...$args),
+            fn() => $this->incrementCount($name)
+        );
+    }
+
+    /**
+     * @param string $name
+     * @param mixed ...$args
+     * @return mixed
+     */
+    private function callGlobalFunction(string $name, mixed ...$args): mixed
+    {
+        return $this->executeWithEffect(
+            $name(...$args),
+            fn() => $this->incrementCount($name)
+        );
     }
 
     /**
@@ -112,10 +148,14 @@ private function generate(string $namespace): string
         return $code;
     }
 
-    private function callGlobalFunction(string $name, mixed ...$args): mixed
+    /**
+     * @param mixed $returnValue
+     * @param Closure $effect
+     * @return mixed
+     */
+    private function executeWithEffect(mixed $returnValue, Closure $effect): mixed
     {
-        $returnValue = $name(...$args);
-        $this->incrementCountOutOfScope($name);
+        $effect();
         return $returnValue;
     }
 
@@ -139,8 +179,7 @@ public function scope(?string $namespace = null): void
      */
     private function getNamespaceFromTrace(array $trace): string
     {
-        $file = $trace[0]['file'];
-        if (preg_match('/\bnamespace\s+([^;]+);/', file_get_contents($file), $matches)) {
+        if (preg_match('/\bnamespace\s+([^;]+);/', file_get_contents($trace[0]['file']), $matches)) {
             return trim($matches[1]);
         }
         throw new NamespaceNotFound('the namespace keyword could not find');
@@ -183,6 +222,10 @@ public function getCalledCountOutScope(string $functionName): int
         return $this->getCalledCount($functionName)['out_scope'];
     }
 
+    /**
+     * @param string $functionName
+     * @return int
+     */
     public function getTotalCount(string $functionName): int
     {
         $calledCount = $this->getCalledCount($functionName);
@@ -222,16 +265,6 @@ public function runWithMock(object $object, Closure $closure): mixed
         return $result;
     }
 
-    /**
-     * @param string $name
-     * @return void
-     */
-    private function incrementCountOutOfScope(string $name): void
-    {
-        $this->calledCount[$name][$name] ??= 0;
-        ++$this->calledCount[$name]['out_scope'];
-    }
-
     /**
      * @param string $method
      * @param array $arguments