csharp/tests/KubernetesClient.Tests/TaskAssert.cs

using System;
using System.Threading;
using System.Threading.Tasks;
using Xunit;

namespace k8s.Tests
{
    static class TaskAssert
    {
        public static void NotCompleted(Task task, string message = "Task should not be completed")
        {
            Assert.False(task.IsCompleted, message);
        }

        public static async Task Completed(Task task, TimeSpan timeout, string message = "Task timed out")
        {
            var timeoutTask = Task.Delay(
                TimeSpan.FromMilliseconds(1000));

            var completedTask = await Task.WhenAny(task, timeoutTask);
            Assert.True(ReferenceEquals(task, completedTask), message);

            await completedTask;
        }

        public static async Task<T> Completed<T>(Task<T> task, TimeSpan timeout, string message = "Task timed out")
        {
            var timeoutTask =
                Task.Delay(
                        TimeSpan.FromMilliseconds(1000))
                    .ContinueWith(
                        completedTimeoutTask =>
                            default(T)); // Value is never returned, but we need a task of the same result type in order to use Task.WhenAny.

            var completedTask = await Task.WhenAny(task, timeoutTask);
            Assert.True(ReferenceEquals(task, completedTask), message);

            return await completedTask;
        }
    }
}