test/test_stat.rb in pork-1.4.2 vs test/test_stat.rb in pork-1.4.3

@@ -19,13 +19,13 @@
       skip if line.include?('}')
     end
   end
 
   def run check=:expect_one_error
-    @stat = @executor.execute(
-      Pork.execute_mode,
-      Pork::Stat.new(Pork.report_class.new(StringIO.new)))
+    stat = Pork::Stat.new(Pork.report_class.new(StringIO.new))
+    stat.protected_exceptions = pork_stat.protected_exceptions
+    @stat = @executor.execute(Pork.execute_mode, stat)
     send(check)
   end
 
   def expect_one_error
     expect(@stat.reporter.io.string).eq "\e[31mE\e[0m"
@@ -37,9 +37,14 @@
   def expect_one_failure
     expect(@stat.reporter.io.string).eq "\e[35mF\e[0m"
     expect(@stat.tests)     .eq 1
     expect(@stat.assertions).eq 0
     expect(@stat.failures)  .eq 1
+  end
+
+  would 'rescue custom errors' do
+    @executor.would{ raise WebMockError }
+    run
   end
 
   would 'always have backtrace' do
     @executor.would
     run