From 0d46c7a19b51e91bb0abd507c7883a615c1708fc Mon Sep 17 00:00:00 2001
From: Guo Xiang Tan <tgx_world@hotmail.com>
Date: Mon, 21 Aug 2017 22:29:14 +0900
Subject: [PATCH] Ensure we reset state for `Messagebus.track_publish`.

---
 spec/support/diagnostics_helper.rb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/spec/support/diagnostics_helper.rb b/spec/support/diagnostics_helper.rb
index 45a0f5146cc..bab725d8179 100644
--- a/spec/support/diagnostics_helper.rb
+++ b/spec/support/diagnostics_helper.rb
@@ -13,8 +13,9 @@ module MessageBus::DiagnosticsHelper
   def track_publish
     @tracking = tracking = []
     yield
-    @tracking = nil
     tracking
+  ensure
+    @tracking = nil
   end
 end