From c68887e1e803ede711417e07008bd4976a629063 Mon Sep 17 00:00:00 2001
From: Maxime Steinhausser <maxime.steinhausser@gmail.com>
Date: Tue, 1 May 2018 14:54:27 +0200
Subject: [PATCH] [Messenger] Reset traceable buses

---
 .../Messenger/DataCollector/MessengerDataCollector.php       | 3 +++
 src/Symfony/Component/Messenger/TraceableMessageBus.php      | 5 +++++
 2 files changed, 8 insertions(+)

diff --git a/src/Symfony/Component/Messenger/DataCollector/MessengerDataCollector.php b/src/Symfony/Component/Messenger/DataCollector/MessengerDataCollector.php
index 5e30bcf281..0fe44d62fe 100644
--- a/src/Symfony/Component/Messenger/DataCollector/MessengerDataCollector.php
+++ b/src/Symfony/Component/Messenger/DataCollector/MessengerDataCollector.php
@@ -67,6 +67,9 @@ class MessengerDataCollector extends DataCollector implements LateDataCollectorI
     public function reset()
     {
         $this->data = array();
+        foreach ($this->traceableBuses as $traceableBus) {
+            $traceableBus->reset();
+        }
     }
 
     private function collectMessage(string $busName, array $tracedMessage)
diff --git a/src/Symfony/Component/Messenger/TraceableMessageBus.php b/src/Symfony/Component/Messenger/TraceableMessageBus.php
index b8b151f32a..ecf0c5658c 100644
--- a/src/Symfony/Component/Messenger/TraceableMessageBus.php
+++ b/src/Symfony/Component/Messenger/TraceableMessageBus.php
@@ -52,4 +52,9 @@ class TraceableMessageBus implements MessageBusInterface
     {
         return $this->dispatchedMessages;
     }
+
+    public function reset()
+    {
+        $this->dispatchedMessages = array();
+    }
 }