benchmarks duration even if it fails (#23)

Co-authored-by: fsainz <[email protected]>

lib/worker_tools/benchmark.rb

@@ -6,9 +6,14 @@ module Benchmark
       attr_accessor :benchmark
 
       def with_wrapper_benchmark(&block)
-        @benchmark = ::Benchmark.measure(&block)
-      ensure
-        model.meta['duration'] = @benchmark.real.round if model.respond_to?(:meta)
+        benchmark = ::Benchmark.measure do
+          block.call
+        rescue StandardError => e
+          @benchmark_error = e
+        end
+
+        model.meta['duration'] = benchmark.real.round if model.respond_to?(:meta)
+        raise @benchmark_error if @benchmark_error
       end
     end
   end

test/worker_tools/benchmark_test.rb

@@ -33,5 +33,17 @@ def run; end
       @importer.perform(@import)
       expect(@importer.model.meta['duration']).must_equal 2
     end
+
+    it 'assigns the duration even if the block fails' do
+      def @importer.run
+        sleep 1
+        raise StandardError
+      end
+
+      ::Benchmark.unstub(:measure)
+
+      assert_raises(StandardError) { @importer.perform(@import) }
+      expect(@importer.model.meta['duration'] >= 1).must_equal true
+    end
   end
 end