lib/snapdragon/cli_application.rb in snapdragon-0.1.2 vs lib/snapdragon/cli_application.rb in snapdragon-0.1.3

@@ -5,10 +5,16 @@
 require_relative './web_application'
 require_relative './suite'
 require_relative './spec_file'
 require_relative './spec_directory'
 
+# Set the default_wait_time to something reasonable for the entire length of
+# the test suite to run. This should probably eventually be something
+# configurable because these could break for people with long running test
+# suites.
+Capybara.default_wait_time = 120 # 2 mins
+
 module Snapdragon
   class CliApplication
     def initialize(arguements)
       @args = arguements
       @suite = Snapdragon::Suite.new
@@ -63,11 +69,13 @@
 
     def run_suite
       session = Capybara::Session.new(:poltergeist, Snapdragon::WebApplication.new(nil, @suite))
       if @suite.filtered?
         session.visit("/run?spec=#{@suite.spec_query_param}")
+        session.find("#testscomplete")
       else
         session.visit("/run")
+        session.find("#testscomplete")
       end
     end
   end
 end