test/integration/instrument_action_test.rb in librato-rails-0.11.0 vs test/integration/instrument_action_test.rb in librato-rails-0.11.1

@@ -6,16 +6,17 @@
     visit instrument_action_path
 
     # puts aggregate.instance_variable_get(:@cache).queued.inspect
     # puts counters.instance_variable_get(:@cache).inspect
 
-    base = 'rails.action.InstrumentActionController.inst.html'
+    source = 'InstrumentActionController.inst.html'
 
+    base = 'rails.action.request'
     timings = %w{time time.db time.view}
     timings.each do |t|
-      assert_equal 1, aggregate["#{base}.#{t}"][:count]
+      assert_equal 1, aggregate.fetch("#{base}.#{t}", source: source)[:count]
     end
 
-    assert_equal 1, counters["#{base}.total"]
+    assert_equal 1, counters.fetch("#{base}.total", source: source)
   end
 
 end