Add MockInstrumentationService for tests

mtodd · mtodd · commit 090561cda1b9 · 2014-08-29T19:56:52.000-07:00
diff --git a/test/test_helper.rb b/test/test_helper.rb
@@ -46,3 +46,21 @@ def options
     end
   end
 end
+
+class MockInstrumentationService
+  def initialize
+    @events = {}
+  end
+
+  def instrument(event, payload)
+    result = yield(payload)
+    @events[event] ||= []
+    @events[event] << [payload, result]
+    result
+  end
+
+  def subscribe(event)
+    @events[event] ||= []
+    @events[event]
+  end
+end
