public static async Task TimeoutAfter()

in e2etest/GuestProxyAgentTest/Extensions/TaskExtensions.cs [8:23]

• 15 lines of code
• 3 McCabe index (conditional complexity)

        public static async Task TimeoutAfter(this Task task, int timeoutMilliSeconds, CancellationTokenSource cancellationTokenSource = null!)
        {
            if (task == await Task.WhenAny(task, Task.Delay(timeoutMilliSeconds)))
            {
                await task;
            }
            else
            {
                if (cancellationTokenSource != null)
                {
                    // Cancel the task
                    cancellationTokenSource.Cancel();
                }
                throw new TimeoutException("task time out.");
            }
        }