tick 0.1.0 → 0.1.1

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    tick (0.0.1)
+    tick (0.1.0)
       rainbow
 
 GEM

data/README.md

@@ -4,7 +4,7 @@ Tick
 About 
 ------
 
-Tick benchmark your method and print benchmark in color  
+Tick benchmark your method and print it in color  
 
 
 Installation 
@@ -19,6 +19,9 @@ Usge
     def foo 
     end 
     tick :foo  
+    tick :foo, :message => "benchmark for foo"
+    tick :foo, :message => lambda {|class_name, method_name| " #{class_name}-#{method_name}"}
+
 
 
 Configuration 
@@ -51,6 +54,7 @@ Customize messages:
     Tick.time_message = lambda { |sec| "COST (#{sec})" }
 
 Default: 
+
 * desc_message: "TICK: method '#{method_name.to_s}' in class '#{self.class.name}'" 
 * time_message: "(#{sec.to_s} ms)"
 
@@ -61,6 +65,7 @@ Set 256 color:
     Tick.time_color = "#FFC482"
 
 Default:
+
 *  desc_color: yellow 
 *  time_color: cyan
 

data/lib/tick/version.rb

@@ -1,3 +1,3 @@
 module Tick
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/tick.rb

@@ -40,7 +40,7 @@ module Tick
   end
 
   def self.desc_message
-    @desc_message
+    @desc_message ||= lambda {|class_name, method_name| "TICK: method '#{method_name}' in class '#{class_name}'"}
   end
 
   def self.desc_message=(block)
@@ -58,7 +58,7 @@ module Tick
   end
 
   def self.time_message
-    @time_message
+    @time_message ||= lambda { |sec| "(#{sec.to_s} s)" }
   end
 
   def self.time_message=(block)
@@ -86,13 +86,22 @@ module Tick
 
 
   module ClassMethods 
-    def tick(method_name)
+    def tick(method_name, options = {})
       alias_method "#{method_name}_without_tick", method_name  
       define_method method_name do
         result = nil 
         if Tick.enabled 
           sec = Benchmark.realtime  { result = self.send("#{method_name}_without_tick") } 
-          _log_benchmark(method_name, sec)
+
+          desc = nil 
+          if options[:message].kind_of?(Proc) 
+            desc = options[:message].call(self.class.name, method_name)
+          else
+            desc = options[:message] || Tick.desc_message.call(self.class.name, method_name)
+          end
+
+          time = Tick.time_message.call(sec)
+          _log_benchmark(desc, time)
         else  
           result = self.send("#{method_name}_without_tick")
         end
@@ -102,9 +111,7 @@ module Tick
   end
 
   module InstanceMethods 
-    def _log_benchmark(method_name, sec)
-      desc = Tick.desc_message.nil? ? "TICK: method '#{method_name.to_s}' in class '#{self.class.name}'" : Tick.desc_message.call(self.class.name, method_name)
-      time = Tick.time_message.nil? ? "(#{sec.to_s} ms)" : Tick.time_message.call(sec)
+    def _log_benchmark(desc, time)
       message = self._colorize_desc(desc)
       message << "  "
       message << self._colorize_time(time)

data/spec/tick_spec.rb

@@ -78,11 +78,16 @@ describe Tick do
 end
 
 describe "A class include Tick" do
-  before do 
-    @klass = Class.new 
+  before(:each) do 
+    class TestClass;end
+    @klass = TestClass
     @klass.send(:include, Tick)
   end
 
+  after(:each) do 
+    Object.send(:remove_const, :TestClass)
+  end
+
   it "should have class method tick" do 
     @klass.should respond_to :tick
   end
@@ -96,6 +101,32 @@ describe "A class include Tick" do
     lambda {@klass.send(:tick, :xxx)}.should raise_error
   end
 
+  describe "tick with options" do 
+    before(:each) do 
+      @klass.class_eval do  
+        def default
+          sleep 0.5
+        end
+      end
+      @instance = @klass.new 
+      @instance.should respond_to :default
+    end
+
+    it "support customize message for each method" do 
+      message = "method default"
+      @klass.send(:tick, :default, :message => message) 
+      mock(@instance)._log_benchmark(message, anything)
+      @instance.default
+    end
+
+    it "support customize lambda message for each method" do 
+      message = "method default"
+      @klass.send(:tick, :default, :message => lambda {|class_name, method_name| "m:#{method_name}" }) 
+      mock(@instance)._log_benchmark("m:default", anything)
+      @instance.default
+    end
+  end
+
   describe "tick on :default method" do 
     before do 
       @klass.class_eval do  
@@ -110,7 +141,7 @@ describe "A class include Tick" do
     end
 
     it "should log time when default method be called" do 
-      mock(@instance)._log_benchmark(:default, numeric)
+      mock(@instance)._log_benchmark(Tick.desc_message.call("TestClass","default"), anything)
       @instance.default
     end
 
@@ -121,6 +152,7 @@ describe "A class include Tick" do
       @instance.default
       Tick.enabled = old_value
     end
+
   end
 
   describe "#_log_benchmark" do 
@@ -142,7 +174,7 @@ describe "A class include Tick" do
       message << "  "
       message << time
       mock(Tick.logger).debug(message)
-      @instance._log_benchmark(:default, 0.1)    
+      @instance._log_benchmark(desc, time)    
     end
 
     it "should print custom message if set custom message" do 
@@ -155,7 +187,7 @@ describe "A class include Tick" do
       message << "  "
       message << time
       mock(Tick.logger).debug(message)
-      @instance._log_benchmark(:default, 0.1)    
+      @instance._log_benchmark(desc, time)    
     end
   end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: tick
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: false
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - allenwei