test/test_helper.rb in vegas-0.1.8 vs test/test_helper.rb in vegas-0.1.9

@@ -18,10 +18,12 @@
 
 module TestHelper
 
   def vegas(*args, &block)
     Vegas::Runner.any_instance.stubs(:daemonize!).once
-    Rack::Handler::Thin.stubs(:run).once
+
+    Vegas::JRUBY ? Rack::Handler::WEBrick.stubs(:run).once : Rack::Handler::Thin.stubs(:run).once
+
     @vegas = Vegas::Runner.new(*args, &block)
   end
 
   def body
     last_response.body.to_s