Merge pull request #87 from php-school/attach-to-multiple-events

Allow attaching to multiple events at same time

Author: AydinHassan

src/Event/EventDispatcher.php

@@ -65,10 +65,25 @@ public function dispatch(EventInterface $event)
     }
 
     /**
-     * @param string $eventName
+     * @param string $eventNames
+     * @param callable $callback
+     */
+    public function listen($eventNames, callable $callback)
+    {
+        if (!is_array($eventNames)) {
+            $eventNames = [$eventNames];
+        }
+
+        foreach ($eventNames as $eventName) {
+            $this->attachListener($eventName, $callback);
+        }
+    }
+
+    /**
+     * @param string|array $eventName
      * @param callable $callback
      */
-    public function listen($eventName, callable $callback)
+    private function attachListener($eventName, callable $callback)
     {
         if (!array_key_exists($eventName, $this->listeners)) {
             $this->listeners[$eventName] = [$callback];

test/Event/EventDispatcherTest.php

@@ -82,4 +82,19 @@ public function testVerifyReturnIsSkippedIfNotInstanceOfResult()
 
         $this->assertEquals([], iterator_to_array($this->results));
     }
+
+    public function testListenWithMultipleEvents()
+    {
+        $e1 = new Event('some-event', ['arg1' => 1, 'arg2' => 2]);
+        $e2 = new Event('some-event', ['arg1' => 1, 'arg2' => 2]);
+        $mockCallback1 = $this->getMock('stdClass', ['callback']);
+        $mockCallback1->expects($this->exactly(2))
+            ->method('callback')
+            ->withConsecutive([$e1], [$e2])
+            ->will($this->returnValue(true));
+
+        $this->eventDispatcher->listen(['some-event', 'second-event'], [$mockCallback1, 'callback']);
+        $this->eventDispatcher->dispatch($e1);
+        $this->eventDispatcher->dispatch($e2);
+    }
 }