From 56f3d05f4746c091827d7607f0afe2857959c73c Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Tue, 8 Jan 2013 10:45:44 -0800
Subject: adding start / finish on the instrumenter, adding tests for the class

---
 .../test/notifications/instrumenter_test.rb        | 50 ++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 activesupport/test/notifications/instrumenter_test.rb

(limited to 'activesupport/test/notifications')

diff --git a/activesupport/test/notifications/instrumenter_test.rb b/activesupport/test/notifications/instrumenter_test.rb
new file mode 100644
index 0000000000..62a9b61464
--- /dev/null
+++ b/activesupport/test/notifications/instrumenter_test.rb
@@ -0,0 +1,50 @@
+require 'abstract_unit'
+require 'active_support/notifications/instrumenter'
+
+module ActiveSupport
+  module Notifications
+    class InstrumenterTest < ActiveSupport::TestCase
+      class TestNotifier
+        attr_reader :starts, :finishes
+
+        def initialize
+          @starts   = []
+          @finishes = []
+        end
+
+        def start(*args);  @starts << args; end
+        def finish(*args); @finishes << args; end
+      end
+
+      attr_reader :instrumenter, :notifier, :payload
+
+      def setup
+        super
+        @notifier     = TestNotifier.new
+        @instrumenter = Instrumenter.new @notifier
+        @payload      =  { :foo => Object.new }
+      end
+
+      def test_instrument
+        called  = false
+        instrumenter.instrument("foo", payload) {
+          called = true
+        }
+
+        assert called
+      end
+
+      def test_start
+        instrumenter.start("foo", payload)
+        assert_equal [["foo", instrumenter.id, payload]], notifier.starts
+        assert_predicate notifier.finishes, :empty?
+      end
+
+      def test_finish
+        instrumenter.finish("foo", payload)
+        assert_equal [["foo", instrumenter.id, payload]], notifier.finishes
+        assert_predicate notifier.starts, :empty?
+      end
+    end
+  end
+end
-- 
cgit v1.2.3