lib/watir/window_collection.rb in watir-7.0.0.beta1 vs lib/watir/window_collection.rb in watir-7.0.0.beta2

@@ -37,9 +37,18 @@
     def ==(other)
       window_list == other.send(:window_list)
     end
     alias eql? ==
 
+    def restore!
+      return if @browser.closed?
+
+      window_list.reject { |win| win.handle == @browser.original_window.handle }.each(&:close)
+      @browser.original_window.use
+    rescue StandardError
+      @browser.close
+    end
+
     def reset!
       @window_list = nil
     end
 
     private