Added WaitForFile to the Waiter object

unknown · unknown · commit e4cb90022d2a · 2016-01-06T15:01:54.000Z
diff --git a/Selenium/ComInterfaces/_Waiter.cs b/Selenium/ComInterfaces/_Waiter.cs
@@ -21,6 +21,8 @@ public interface _Waiter {
         [DispId(830), Description("Waits for the given function to return true. Ex: Waiter.Until addressOf MyFunction")]
         object Until(object function, object argument = null, int timeout = -1, string timeoutMessage = null);
 
+        [DispId(876), Description("Waits for a file to be ready.")]
+        void WaitForFile(string path, int timeout = -1);
     }
 
 }
diff --git a/Selenium/Waiter.cs b/Selenium/Waiter.cs
@@ -2,6 +2,7 @@
 using Selenium.Internal;
 using System;
 using System.ComponentModel;
+using System.IO;
 using System.Runtime.InteropServices;
 
 namespace Selenium {
@@ -265,6 +266,25 @@ private static bool IsNotNullOrFalse<T>(T value) {
             return !(value == null || value is bool && (bool)(object)value == false);
         }
 
+
+
+        /// <summary>
+        /// Waits for a file to be ready.
+        /// </summary>
+        /// <param name="path">File path</param>
+        /// <param name="timeout">Timeout in milliseonds</param>
+        public void WaitForFile(string path, int timeout = -1) {
+            Waiter.WaitUntil(() => {
+                if (File.Exists(path)) {
+                    try {
+                        File.Move(path, path);
+                        return true;
+                    } catch (System.IO.IOException) { }
+                }
+                return false;
+            }, timeout, 100);
+        }
+
     }
 
 }

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,8 @@ public interface _Waiter {`
`21`	`21`	`[DispId(830), Description("Waits for the given function to return true. Ex: Waiter.Until addressOf MyFunction")]`
`22`	`22`	`object Until(object function, object argument = null, int timeout = -1, string timeoutMessage = null);`
`23`	`23`
	`24`	`+ [DispId(876), Description("Waits for a file to be ready.")]`
	`25`	`+ void WaitForFile(string path, int timeout = -1);`
`24`	`26`	`}`
`25`	`27`
`26`	`28`	`}`