Add getter for ActivityPrototype::$factory (#492)

temporalio · Sep 3, 2024 · 28538c1 · 28538c1
1 parent 7a11616
commit 28538c1
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/Internal/Declaration/Prototype/ActivityPrototype.php b/src/Internal/Declaration/Prototype/ActivityPrototype.php
@@ -112,4 +112,9 @@ public function isLocalActivity(): bool
     {
         return $this->isLocalActivity;
     }
+
+    public function getFactory(): ?Closure
+    {
+        return $this->factory;
+    }
 }
diff --git a/tests/Unit/Activity/ActivityPrototypeTestCase.php b/tests/Unit/Activity/ActivityPrototypeTestCase.php
@@ -111,4 +111,13 @@ public function testFactoryAcceptsReflectionClassOfActivity(): void
         $this->assertEquals($protoWithFactory->getInstance()->getContext(), $protoWithFactory->getInstance()->getContext());
         $this->assertNotSame($protoWithFactory->getInstance()->getContext(), $protoWithFactory->getInstance()->getContext());
     }
+
+    public function testGetFactory(): void
+    {
+        $proto = $this->activityReader->fromClass(DummyActivity::class)[0];
+        $protoWithFactory = $proto->withFactory(fn (\ReflectionClass $reflectionClass) => $reflectionClass->newInstance());
+
+        $this->assertNull($proto->getFactory());
+        $this->assertNotNull($protoWithFactory->getFactory());
+    }
 }