Merge pull request #13 from benSlaughter/Issue7

added tests for error handling during record event

lib/allotment.rb

@@ -12,9 +12,12 @@ class Allotment
 	class << self
 		def record_event name = 'unnamed', &block
 			start_recording name
-			yield
-		ensure
-			return stop_recording name
+			begin
+				yield
+			ensure
+				time = stop_recording name
+			end
+			return time
 		end
 
 		def start_recording name = 'unnamed'

spec/allotment_spec.rb

@@ -25,6 +25,11 @@
       result = Allotment.record_event('my_recording 0.4') { sleep 0.01 }
       result.round(2).should eq 0.01
     end
+
+    it "returns still records the performance upon error" do
+      expect { Allotment.record_event('failed_recording') { sleep 0.01; raise 'error' } }.to raise_error RuntimeError, "error"
+      Allotment.results['failed_recording'].first.round(2).should eq 0.01
+    end
   end
 
   describe ".start_recording" do