symfony
diff --git a/‎src/Symfony/Bridge/Twig/Tests/Extension/WorkflowExtensionTest.php
Lines changed: 38 additions & 8 deletions b/‎src/Symfony/Bridge/Twig/Tests/Extension/WorkflowExtensionTest.php
Lines changed: 38 additions & 8 deletions
@@ -15,32 +15,62 @@
 use Symfony\Component\Workflow\Definition;
 use Symfony\Component\Workflow\Marking;
 use Symfony\Component\Workflow\Registry;
+use Symfony\Component\Workflow\Transition;
 use Symfony\Component\Workflow\Workflow;
 
 class WorkflowExtensionTest extends \PHPUnit_Framework_TestCase
 {
+    private $extension;
+
     protected function setUp()
     {
         if (!class_exists(Workflow::class)) {
             $this->markTestSkipped('The Workflow component is needed to run tests for this extension.');
         }
-    }
 
-    public function testHasMarkedPlace()
-    {
-        $definition = new Definition(['ordered', 'waiting_for_payment', 'processed'], []);
+        $places = array('ordered', 'waiting_for_payment', 'processed');
+        $transitions = array(
+            new Transition('t1', 'ordered', 'waiting_for_payment'),
+            new Transition('t2', 'waiting_for_payment', 'processed'),
+        );
+        $definition = new Definition($places, $transitions);
         $workflow = new Workflow($definition);
 
         $registry = new Registry();
         $registry->add($workflow, \stdClass::class);
 
-        $extension = new WorkflowExtension($registry);
+        $this->extension = new WorkflowExtension($registry);
+    }
+
+    public function testCanTransition()
+    {
+        $subject = new \stdClass();
+        $subject->marking = array();
+
+        $this->assertTrue($this->extension->canTransition($subject, 't1'));
+        $this->assertFalse($this->extension->canTransition($subject, 't2'));
+    }
+
+    public function testGetEnabledTransitions()
+    {
+        $subject = new \stdClass();
+        $subject->marking = array();
+
+        $transitions = $this->extension->getEnabledTransitions($subject);
 
+        $this->assertCount(1, $transitions);
+        $this->assertInstanceOf(Transition::class, $transitions[0]);
+        $this->assertSame('t1', $transitions[0]->getName());
+    }
+
+    public function testHasMarkedPlace()
+    {
         $subject = new \stdClass();
+        $subject->marking = array();
         $subject->marking = array('ordered' => 1, 'waiting_for_payment' => 1);
 
-        $this->assertTrue($extension->hasMarkedPlace($subject, 'ordered'));
-        $this->assertTrue($extension->hasMarkedPlace($subject, 'waiting_for_payment'));
-        $this->assertFalse($extension->hasMarkedPlace($subject, 'processed'));
+        $this->assertTrue($this->extension->hasMarkedPlace($subject, 'ordered'));
+        $this->assertTrue($this->extension->hasMarkedPlace($subject, 'waiting_for_payment'));
+        $this->assertFalse($this->extension->hasMarkedPlace($subject, 'processed'));
     }
 }