symfony · epitre · Aug 19, 2020 · Aug 19, 2020 · Aug 19, 2020 · Aug 19, 2020
diff --git a/src/Symfony/Component/Workflow/CHANGELOG.md b/src/Symfony/Component/Workflow/CHANGELOG.md
@@ -9,6 +9,7 @@ CHANGELOG
  * Dispatch an event when the subject enters in the workflow for the very first time
  * Added a default context to the previous event
  * Added support for specifying which events should be dispatched when calling `workflow->apply()`
+ * It is now possible to retrieve a workflow by using a class name
 
 5.1.0
 -----

diff --git a/src/Symfony/Component/Workflow/Registry.php b/src/Symfony/Component/Workflow/Registry.php
@@ -41,7 +41,7 @@ public function has(object $subject, string $workflowName = null): bool
     /**
      * @return Workflow
      */
-    public function get(object $subject, string $workflowName = null)
+    public function get($subject, string $workflowName = null)
     {
         $matched = [];
 
@@ -51,16 +51,21 @@ public function get(object $subject, string $workflowName = null)
             }
         }
 
+        if (\is_string($subject)) {
+            $type = $subject;
+        } else {
+            $type = get_debug_type($subject);
+        }
         if (!$matched) {
-            throw new InvalidArgumentException(sprintf('Unable to find a workflow for class "%s".', get_debug_type($subject)));
+            throw new InvalidArgumentException(sprintf('Unable to find a workflow for class "%s".', $type));
         }
 
         if (2 <= \count($matched)) {
             $names = array_map(static function (WorkflowInterface $workflow): string {
                 return $workflow->getName();
             }, $matched);
 
-            throw new InvalidArgumentException(sprintf('Too many workflows (%s) match this subject (%s); set a different name on each and use the second (name) argument of this method.', implode(', ', $names), get_debug_type($subject)));
+            throw new InvalidArgumentException(sprintf('Too many workflows (%s) match this subject (%s); set a different name on each and use the second (name) argument of this method.', implode(', ', $names), $type));
         }
 
         return $matched[0];
@@ -81,12 +86,18 @@ public function all(object $subject): array
         return $matched;
     }
 
-    private function supports(WorkflowInterface $workflow, WorkflowSupportStrategyInterface $supportStrategy, object $subject, ?string $workflowName): bool
+    private function supports(WorkflowInterface $workflow, WorkflowSupportStrategyInterface $supportStrategy, $subject, ?string $workflowName): bool
     {
         if (null !== $workflowName && $workflowName !== $workflow->getName()) {
             return false;
         }
 
+        if (\is_string($subject)) {
+            $stdClass = new \stdClass();
+            $stdClass->class = $subject;
+            $subject = $stdClass;
+        }
+
         return $supportStrategy->supports($workflow, $subject);
     }
 }
diff --git a/src/Symfony/Component/Workflow/SupportStrategy/InstanceOfSupportStrategy.php b/src/Symfony/Component/Workflow/SupportStrategy/InstanceOfSupportStrategy.php
@@ -14,6 +14,7 @@
 use Symfony\Component\Workflow\WorkflowInterface;
 
 /**
+ * @author Carlos Pereira De Amorim <carlos@shauri.fr>
  * @author Andreas Kleemann <akleemann@inviqa.com>
  * @author Amrouche Hamza <hamza.simperfit@gmail.com>
  */
@@ -31,6 +32,10 @@ public function __construct(string $className)
      */
     public function supports(WorkflowInterface $workflow, object $subject): bool
     {
+        if ($subject instanceof \stdClass && property_exists($subject, 'class')) {
+            return $subject->class === $this->className;
+        }
+
         return $subject instanceof $this->className;
     }
 

diff --git a/src/Symfony/Component/Workflow/Tests/RegistryTest.php b/src/Symfony/Component/Workflow/Tests/RegistryTest.php
@@ -53,6 +53,13 @@ public function testGetWithSuccess()
         $this->assertSame('workflow2', $workflow->getName());
     }
 
+    public function testGetWithSuccessFromClassName()
+    {
+        $workflow = $this->registry->get(Subject1::class);
+        $this->assertInstanceOf(Workflow::class, $workflow);
+        $this->assertSame('workflow1', $workflow->getName());
+    }
+
     public function testGetWithMultipleMatch()
     {
         $this->expectException('Symfony\Component\Workflow\Exception\InvalidArgumentException');
@@ -102,7 +109,11 @@ private function createWorkflowSupportStrategy($supportedClassName)
     {
         $strategy = $this->getMockBuilder(WorkflowSupportStrategyInterface::class)->getMock();
         $strategy->expects($this->any())->method('supports')
-            ->willReturnCallback(function ($workflow, $subject) use ($supportedClassName) {
+            ->willReturnCallback(function ($workflow, object $subject) use ($supportedClassName) {
+                if ($subject instanceof \stdClass && property_exists($subject, 'class')) {
+                    return $subject->class === $supportedClassName;
+                }
+
                 return $subject instanceof $supportedClassName;
             });
 

diff --git a/src/Symfony/Component/Workflow/Tests/SupportStrategy/InstanceOfSupportStrategyTest.php b/src/Symfony/Component/Workflow/Tests/SupportStrategy/InstanceOfSupportStrategyTest.php
@@ -13,13 +13,21 @@ public function testSupportsIfClassInstance()
         $strategy = new InstanceOfSupportStrategy(Subject1::class);
 
         $this->assertTrue($strategy->supports($this->createWorkflow(), new Subject1()));
+
+        $stdClass = new \StdClass();
+        $stdClass->class = Subject1::class;
+        $this->assertTrue($strategy->supports($this->createWorkflow(), $stdClass));
     }
 
     public function testSupportsIfNotClassInstance()
     {
         $strategy = new InstanceOfSupportStrategy(Subject2::class);
 
         $this->assertFalse($strategy->supports($this->createWorkflow(), new Subject1()));
+
+        $stdClass = new \StdClass();
+        $stdClass->class = Subject1::class;
+        $this->assertFalse($strategy->supports($this->createWorkflow(), $stdClass));
     }
 
     private function createWorkflow()