spec/runner: catch timeouts in the specs

meh · meh · commit 9c4f61e6f6cd · 2014-01-30T20:43:47.000+01:00
diff --git a/spec/runner.rb b/spec/runner.rb
@@ -82,33 +82,37 @@
 
 print "\rRunning specs..."
 
-# wait until the specs have finished, thus changing the content of #totals
-Selenium::WebDriver::Wait.new(timeout: 1200, interval: 30).until {
-  print '.'
+begin
+  # wait until the specs have finished, thus changing the content of #totals
+  Selenium::WebDriver::Wait.new(timeout: 1200, interval: 30).until {
+    print '.'
 
-  not browser['totals'].text.strip.empty?
-}
+    not browser['totals'].text.strip.empty?
+  }
 
-totals   = browser['totals'].text
-duration = browser[css: '#duration strong'].text
+  totals   = browser['totals'].text
+  duration = browser[css: '#duration strong'].text
 
-print "\r#{totals} in #{duration}"
+  print "\r#{totals} in #{duration}"
 
-# take a screenshot and upload it to imgur
-begin
-  browser.save_screenshot('screenshot.png')
-  response = RestClient.post('https://api.imgur.com/3/upload',
-    { image: File.open('screenshot.png') },
-    { 'Authorization' => 'Client-ID 1979876fe2a097e' })
-
-  print " ("
-  print JSON.parse(response.to_str)['data']['link']
-  puts  ")"
-rescue Exception
-  puts
-end
-
-# no failures, happy times
-unless totals =~ / 0 failures/
-  exit 1
+  # no failures, happy times
+  unless totals =~ / 0 failures/
+    exit 1
+  end
+rescue Selenium::WebDriver::Error::TimeOutError
+  print "\rThe specs have timed out."
+ensure
+  # take a screenshot and upload it to imgur
+  begin
+    browser.save_screenshot('screenshot.png')
+    response = RestClient.post('https://api.imgur.com/3/upload',
+      { image: File.open('screenshot.png') },
+      { 'Authorization' => 'Client-ID 1979876fe2a097e' })
+
+    print " ("
+    print JSON.parse(response.to_str)['data']['link']
+    puts  ")"
+  rescue Exception
+    puts
+  end
 end
