allow to mock local activities

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 - Allow to mock workflows without a running temporal server and worker
+- Allow to mock local activities
 
 ## v0.4
 - Added support for `Eloquent` models serialization/deserialization

src/Builder/LocalActivityBuilder.php

@@ -3,10 +3,10 @@
 namespace Keepsuit\LaravelTemporal\Builder;
 
 use InvalidArgumentException;
+use Keepsuit\LaravelTemporal\Facade\Temporal;
 use Temporal\Activity\LocalActivityOptions;
 use Temporal\Common\RetryOptions;
 use Temporal\Internal\Workflow\ActivityProxy;
-use Temporal\Workflow;
 use Temporal\Workflow\ActivityStubInterface;
 
 /**
@@ -39,12 +39,12 @@ public static function new(): LocalActivityBuilder
      */
     public function build(string $class)
     {
-        return Workflow::newActivityStub($class, $this->activityOptions);
+        return Temporal::getTemporalContext()->newActivityStub($class, $this->activityOptions);
     }
 
     public function buildUntyped(): ActivityStubInterface
     {
-        return Workflow::newUntypedActivityStub($this->activityOptions);
+        return Temporal::getTemporalContext()->newUntypedActivityStub($this->activityOptions);
     }
 
     public function __call(string $name, array $arguments): self

src/Testing/Fakes/FakeActivityStub.php

@@ -7,6 +7,7 @@
 use React\Promise\PromiseInterface;
 use Temporal\Activity\ActivityOptions;
 use Temporal\Activity\ActivityOptionsInterface;
+use Temporal\Activity\LocalActivityOptions;
 use Temporal\DataConverter\EncodedValues;
 use Temporal\Workflow\ActivityStubInterface;
 
@@ -25,16 +26,18 @@ public function getOptions(): ActivityOptionsInterface
 
     public function execute(string $name, array $args = [], $returnType = null, bool $isLocalActivity = false): PromiseInterface
     {
-        /** @var ActivityOptions $options */
+        /** @var ActivityOptions|LocalActivityOptions $options */
         $options = $this->getOptions();
 
-        $mock = $this->getTemporalMocker()->getActivityResult($name, $options->taskQueue);
+        $taskQueue = $isLocalActivity ? null : $options->taskQueue;
+
+        $mock = $this->getTemporalMocker()->getActivityResult($name, $taskQueue);
 
         if (! $mock instanceof \Closure) {
             return $this->stub->execute($name, $args, $returnType, $isLocalActivity);
         }
 
-        $this->getTemporalMocker()->recordActivityDispatch($name, $options->taskQueue, $args);
+        $this->getTemporalMocker()->recordActivityDispatch($name, $taskQueue, $args);
 
         $this->result = $mock->__invoke(...$args);
 

src/Testing/TemporalMocker.php

@@ -34,7 +34,7 @@ public function mockActivityResult(string $activityName, mixed $activityResult,
         $this->cache->saveActivityMock($activityName, $result, $taskQueue);
     }
 
-    public function getActivityResult(string $activityName, string $taskQueue): ?Closure
+    public function getActivityResult(string $activityName, ?string $taskQueue): ?Closure
     {
         return $this->cache->getActivityMock($activityName, $taskQueue);
     }
@@ -49,7 +49,7 @@ public function getWorkflowDispatches(string $workflowName): array
         return $this->cache->getWorkflowDispatches($workflowName);
     }
 
-    public function recordActivityDispatch(string $activityName, string $taskQueue, array $args): void
+    public function recordActivityDispatch(string $activityName, ?string $taskQueue, array $args): void
     {
         $this->cache->recordActivityDispatch($activityName, $taskQueue, $args);
     }

src/Testing/TemporalMockerCache.php

@@ -93,9 +93,10 @@ public function saveActivityMock(string $activityName, mixed $value, ?string $ta
         ]);
     }
 
-    public function getActivityMock(string $activityName, string $taskQueue): ?Closure
+    public function getActivityMock(string $activityName, ?string $taskQueue): ?Closure
     {
         $value = $this->cache->get(sprintf('activity::%s', $activityName));
+
         if (! is_array($value)) {
             return null;
         }
@@ -146,7 +147,7 @@ public function getWorkflowDispatches(string $workflowName): array
         );
     }
 
-    public function recordActivityDispatch(string $activityName, string $taskQueue, array $args): void
+    public function recordActivityDispatch(string $activityName, ?string $taskQueue, array $args): void
     {
         $cacheKey = sprintf('activity_dispatch::%s', $activityName);