benchmark_harness 0.0.0 → 0.0.1

data/VERSION

@@ -1 +1 @@
-0.0.0
+0.0.1

data/benchmark_harness.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{benchmark_harness}
-  s.version = "0.0.0"
+  s.version = "0.0.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Daniel Higginbotham"]
@@ -30,6 +30,7 @@ Gem::Specification.new do |s|
     "lib/benchmark_harness.rb",
     "lib/benchmark_harness/collection.rb",
     "spec/benchmark_harness_spec.rb",
+    "spec/fixtures/nested_example.rb",
     "spec/fixtures/string_example.rb",
     "spec/spec_helper.rb"
   ]
@@ -40,6 +41,7 @@ Gem::Specification.new do |s|
   s.summary = %q{harness for running benchmark multiple times and running stats on the results}
   s.test_files = [
     "spec/benchmark_harness_spec.rb",
+    "spec/fixtures/nested_example.rb",
     "spec/fixtures/string_example.rb",
     "spec/spec_helper.rb"
   ]

data/lib/benchmark_harness.rb

@@ -10,7 +10,7 @@ class BenchmarkHarness
   
   class << self
     extend Forwardable
-    def_delegators :instance, :collections, :measure, :flush!
+    def_delegators :instance, :collections, :measure, :flush!, :wrap_method
   end
   
   def collections
@@ -22,9 +22,56 @@ class BenchmarkHarness
   end
   
   def measure(collection_name)
+    result = nil
     m = Benchmark.measure{
-      yield
+      result = yield
     }
     collections[collection_name] << m.to_a[1..-1]
+    result
+  end
+  
+  # @param [String] signature
+  # @example "BenchmarkHarness#measure" instance method
+  # @example "BenchmarkHarness.measure" class method
+  def wrap_method(signature, collection_name)
+    if signature =~ /#/
+      split_on = "#"
+      class_method = false
+    elsif signature =~ /\./
+      split_on = "."
+      class_method = true
+    else
+      raise ArgumentError, "your signature must contain a method"
+    end
+    
+    constant_name, method_name = signature.split(split_on)
+    constant = get_constant(constant_name)
+    
+    method_definition = <<-END
+      alias :#{method_name}_pre_benchmark_harness :#{method_name}
+      def #{method_name}(*args, &block)
+        BenchmarkHarness.measure(:#{collection_name}){result = #{method_name}_pre_benchmark_harness(*args, &block)}
+      end
+    END
+    
+    if class_method
+      constant = class << constant; self; end
+    end
+    
+    constant.class_eval method_definition
+  end
+  
+  # @param [String] name
+  def get_constant(name)
+    pieces = name.split("::")
+    parent_constant = Object
+    pieces.each do |piece|
+      if parent_constant.const_defined?(piece)
+        parent_constant = parent_constant.const_get(piece)
+      else
+        raise NameError, "#{name} contains an undefined constant"
+      end
+    end
+    parent_constant
   end
 end

data/spec/benchmark_harness_spec.rb

@@ -83,6 +83,20 @@ describe "BenchmarkHarness" do
     end
   end
   
+  describe "#wrap" do
+    it "should wrap a class method with benchmark harness measure" do
+      BenchmarkHarness.wrap_method("NestedExample::OneLevelDown.add", :add)
+      NestedExample::OneLevelDown.add
+      BenchmarkHarness.collections[:add].should_not be_empty      
+    end
+    
+    it "should wrap an instance method with benchmark harness measure" do
+      BenchmarkHarness.wrap_method("NestedExample::OneLevelDown#subtract", :subtract)
+      NestedExample::OneLevelDown.new.subtract
+      BenchmarkHarness.collections[:subtract].should_not be_empty
+    end
+  end
+  
   describe "#flush" do
     it "should remove all previous samples" do
       2.times{

data/spec/fixtures/nested_example.rb

@@ -0,0 +1,13 @@
+module NestedExample
+  class OneLevelDown
+    class << self
+      def add
+        1 + 1
+      end
+    end
+  
+    def subtract
+      1 - 1
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -12,5 +12,5 @@ Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
 Dir["#{File.dirname(__FILE__)}/fixtures/**/*.rb"].each {|f| require f}
 
 RSpec.configure do |config|
-  
+  config.mock_with :mocha
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: benchmark_harness
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: false
   segments: 
   - 0
   - 0
-  - 0
-  version: 0.0.0
+  - 1
+  version: 0.0.1
 platform: ruby
 authors: 
 - Daniel Higginbotham
@@ -145,6 +145,7 @@ files:
 - lib/benchmark_harness.rb
 - lib/benchmark_harness/collection.rb
 - spec/benchmark_harness_spec.rb
+- spec/fixtures/nested_example.rb
 - spec/fixtures/string_example.rb
 - spec/spec_helper.rb
 has_rdoc: true
@@ -183,5 +184,6 @@ specification_version: 3
 summary: harness for running benchmark multiple times and running stats on the results
 test_files: 
 - spec/benchmark_harness_spec.rb
+- spec/fixtures/nested_example.rb
 - spec/fixtures/string_example.rb
 - spec/spec_helper.rb