spec_helper.rb in watir-webdriver-0.3.5

- old
+ new

@@ -18,12 +18,11 @@
 include Watir::Exception
 
 WEBDRIVER_SELECTORS = [:class, :class_name, :css, :id, :name, :tag_name, :xpath]
 
 if ENV['TRAVIS']
-  require 'headless'
-  display = Headless.new
-  display.start
+  ENV['DISPLAY'] = ":99.0"
+end
 
-  pid = Process.pid
-  at_exit { display.stop if Process.pid == pid }
+if Selenium::WebDriver::Platform.linux? && ENV['DISPLAY'].nil?
+  raise "DISPLAY not set"
 end
\ No newline at end of file