Respect parent class contract in ContainerAwareDoctrineEventManager

Koc · Koc · commit a5076ce0b3bd · 2019-05-01T00:27:11.000+03:00
diff --git a/src/Symfony/Bridge/Doctrine/ContainerAwareEventManager.php b/src/Symfony/Bridge/Doctrine/ContainerAwareEventManager.php
@@ -44,23 +44,22 @@ public function __construct(ContainerInterface $container)
      * @param EventArgs $eventArgs The event arguments to pass to the event handlers/listeners.
      *                             If not supplied, the single empty EventArgs instance is used.
      *
-     * @return bool
+     * @return void
      */
     public function dispatchEvent($eventName, EventArgs $eventArgs = null)
     {
-        if (isset($this->listeners[$eventName])) {
-            $eventArgs = null === $eventArgs ? EventArgs::getEmptyInstance() : $eventArgs;
+        if (!isset($this->listeners[$eventName])) {
+            return;
+        }
 
-            $initialized = isset($this->initialized[$eventName]);
+        $eventArgs = null === $eventArgs ? EventArgs::getEmptyInstance() : $eventArgs;
 
-            foreach ($this->listeners[$eventName] as $hash => $listener) {
-                if (!$initialized && \is_string($listener)) {
-                    $this->listeners[$eventName][$hash] = $listener = $this->container->get($listener);
-                }
+        if (!isset($this->initialized[$eventName])) {
+            $this->initializeListeners($eventName);
+        }
 
-                $listener->$eventName($eventArgs);
-            }
-            $this->initialized[$eventName] = true;
+        foreach ($this->listeners[$eventName] as $hash => $listener) {
+            $listener->$eventName($eventArgs);
         }
     }
 
@@ -73,7 +72,21 @@ public function dispatchEvent($eventName, EventArgs $eventArgs = null)
      */
     public function getListeners($event = null)
     {
-        return $event ? $this->listeners[$event] : $this->listeners;
+        if (null !== $event) {
+            if (!isset($this->initialized[$event])) {
+                $this->initializeListeners($event);
+            }
+
+            return $this->listeners[$event];
+        }
+
+        foreach (array_keys($this->listeners) as $event) {
+            if (!isset($this->initialized[$event])) {
+                $this->initializeListeners($event);
+            }
+        }
+
+        return $this->listeners;
     }
 
     /**
@@ -138,4 +151,18 @@ public function removeEventListener($events, $listener)
             }
         }
     }
+
+    /**
+     * @param string $eventName
+     */
+    private function initializeListeners($eventName)
+    {
+        foreach ($this->listeners[$eventName] as $hash => $listener) {
+            if (\is_string($listener)) {
+                $this->listeners[$eventName][$hash] = $this->container->get($listener);
+            }
+        }
+
+        $this->initialized[$eventName] = true;
+    }
 }
diff --git a/src/Symfony/Bridge/Doctrine/Tests/ContainerAwareEventManagerTest.php b/src/Symfony/Bridge/Doctrine/Tests/ContainerAwareEventManagerTest.php
@@ -38,6 +38,24 @@ public function testDispatchEvent()
         $this->assertTrue($listener2->called);
     }
 
+    public function testGetListenersForEvent()
+    {
+        $this->container->set('foobar', $listener1 = new MyListener());
+        $this->evm->addEventListener('foo', 'foobar');
+        $this->evm->addEventListener('foo', $listener2 = new MyListener());
+
+        $this->assertSame([$listener1, $listener2], $this->evm->getListeners('foo'));
+    }
+
+    public function testGetListeners()
+    {
+        $this->container->set('foobar', $listener1 = new MyListener());
+        $this->evm->addEventListener('foo', 'foobar');
+        $this->evm->addEventListener('foo', $listener2 = new MyListener());
+
+        $this->assertSame(['foo' => [$listener1, $listener2]], $this->evm->getListeners());
+    }
+
     public function testRemoveEventListener()
     {
         $this->evm->addEventListener('foo', 'bar');
