// Copyright © 2010-2017 The CefSharp Authors. All rights reserved.

//

// Use of this source code is governed by a BSD-style license that can be found in the LICENSE file.

using System;

using System.Threading.Tasks;

using CefSharp.Internals;

namespace CefSharp.Test

{

public static class WebBrowserTestExtensions

{

public static Task LoadPageAsync(this IWebBrowser browser, string address = null)

{

//If using .Net 4.6 then use TaskCreationOptions.RunContinuationsAsynchronously

//and switch to tcs.TrySetResult below - no need for the custom extension method

var tcs = new TaskCompletionSource<bool>();

EventHandler<LoadingStateChangedEventArgs> handler = null;

handler = (sender, args) =>

{

//Wait for while page to finish loading not just the first frame

if (!args.IsLoading)

{

browser.LoadingStateChanged -= handler;

//This is required when using a standard TaskCompletionSource

//Extension method found in the CefSharp.Internals namespace

tcs.TrySetResultAsync(true);

}

};

browser.LoadingStateChanged += handler;

if (!string.IsNullOrEmpty(address))

{

browser.Load(address);

}

return tcs.Task;

}

}

}