WaitUntil.cs

WaitUntil.cs

private void WaitUntil(Action assertion, int timeoutMillis)
{
    var cancelTokenSource = new CancellationTokenSource(timeoutMillis);
    var token = cancelTokenSource.Token;

    var t = Task.Factory.StartNew(() =>
    {
        while (true)
        {
            try
            {
                token.ThrowIfCancellationRequested();
                assertion();
                break;
            }
            catch (OperationCanceledException e)
            {
                throw new OperationCanceledException("Timed out while waiting on assertion.");
            }
            catch (Exception)
            {
                // Allow retry attempts to fire
            }
        }
    }, token);
    Task.WaitAll(t);
}