1 files changed, 10 insertions, 1 deletions
diff --git a/cucumber/features/step_definitions/common_steps.rb b/cucumber/features/step_definitions/common_steps.rb
index 0cebc1a3..feec90e2 100644
--- a/cucumber/features/step_definitions/common_steps.rb
+++ b/cucumber/features/step_definitions/common_steps.rb
@@ -669,7 +669,7 @@ Given /^I install GRUB$/ do
 end
 
 Given /^I allow reboot after the install is complete$/ do
-  @screen.wait(diui_png("InstallComplete"), 4 * 60 * PATIENCE)
+  step('I see the "InstallComplete" screen, after at most 240 seconds')
   @screen.type(Sikuli::Key.ENTER)
 end
 
@@ -880,6 +880,15 @@ Then /^I (do not )?see "([^"]*)" after at most (\d+) seconds$/ do |negation, ima
   end
 end
 
+Then /^I (do not )?see the "([^"]*)" screen, after at most (\d+) seconds$/ do |negation, image, time|
+  begin
+    @screen.wait(diui_png(image), time.to_i)
+    raise "found '" + diui_png(image) + "' while expecting not to" if negation
+  rescue FindFailed => e
+    raise e if not(negation)
+  end
+end
+
 Then /^all Internet traffic has only flowed through Tor$/ do
   leaks = FirewallLeakCheck.new(@sniffer.pcap_file,
                                 :accepted_hosts => get_all_tor_nodes)