using System;
using System.Threading.Tasks;
using Xunit.Sdk;
using Nito.AsyncEx;
using System.Threading;

namespace CefSharp.Test
{
    internal static class AssertEx
    {
        /// <summary>
		/// Verifies that a event with the exact event args (and not a derived type) is raised
        /// This method differs from <see cref="Xunit.Assert.RaisesAsync{T}(Action{EventHandler{T}}, Action{EventHandler{T}}, Func{Task})"/>
        /// in that it waits for the event to be raised before returning (or is cancelled).
		/// </summary>
		/// <typeparam name="T">The type of the event arguments to expect</typeparam>
        /// <param name="cancelAfter">number of miliseconds to wait before the timeout</param>
		/// <param name="attach">Code to attach the event handler</param>
		/// <param name="detach">Code to detach the event handler</param>
		/// <param name="testCode">A delegate to the code to be tested</param>
		/// <returns>The event sender and arguments wrapped in an object</returns>
		/// <exception cref="RaisesException">Thrown when the expected event was not raised.</exception>
		public static async Task<Xunit.Assert.RaisedEvent<T>> RaisesAsync<T>(
            int cancelAfter,
            Action<EventHandler<T>> attach,
            Action<EventHandler<T>> detach,
            Action testCode) where T : EventArgs
        {
            var raisedEvent = await RaisesAsyncInternal(cancelAfter, attach, detach, testCode);

            if (raisedEvent == null)
                throw new RaisesException(typeof(T));

            if (raisedEvent.Arguments != null && !raisedEvent.Arguments.GetType().Equals(typeof(T)))
                throw new RaisesException(typeof(T), raisedEvent.Arguments.GetType());

            return raisedEvent;
        }

		private static async Task<Xunit.Assert.RaisedEvent<T>> RaisesAsyncInternal<T>(
            int cancelAfter,
            Action<EventHandler<T>> attach,
            Action<EventHandler<T>> detach,
            Action testCode) where T : EventArgs
        {
            GuardArgumentNotNull(nameof(attach), attach);
            GuardArgumentNotNull(nameof(detach), detach);
            GuardArgumentNotNull(nameof(testCode), testCode);

            using var cts = new CancellationTokenSource();
            var manualResetEvent = new AsyncManualResetEvent();

            cts.CancelAfter(cancelAfter);

            Xunit.Assert.RaisedEvent<T> raisedEvent = null;
            
            attach(Handler);
            testCode();
            await manualResetEvent.WaitAsync(cts.Token);
            detach(Handler);

            return raisedEvent;

            void Handler(object s, T args)
            {
                raisedEvent = new Xunit.Assert.RaisedEvent<T>(s, args);
                manualResetEvent.Set();
            }
        }

        internal static void GuardArgumentNotNull(string argName, object argValue)
        {
            if (argValue == null)
            {
                throw new ArgumentNullException(argName);
            }
        }
    }
}