Always specify context parameter to AsyncToSyncAdapter.Await

nunit · Dec 2, 2024 · 18f9331 · 18f9331
1 parent 5a4f35c
commit 18f9331
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 26 deletions.
diff --git a/src/NUnitFramework/framework/Assert.Exceptions.Async.cs b/src/NUnitFramework/framework/Assert.Exceptions.Async.cs
@@ -23,7 +23,7 @@ public partial class Assert
             Exception? caughtException = null;
             try
             {
-                AsyncToSyncAdapter.Await(code.Invoke);
+                AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, code.Invoke);
             }
             catch (Exception e)
             {

diff --git a/src/NUnitFramework/framework/Assert.cs b/src/NUnitFramework/framework/Assert.cs
@@ -230,7 +230,7 @@ public static void Multiple(AsyncTestDelegate testDelegate)
         {
             using (EnterMultipleScope())
             {
-                AsyncToSyncAdapter.Await(testDelegate.Invoke);
+                AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, testDelegate.Invoke);
             }
         }
 

diff --git a/src/NUnitFramework/framework/Constraints/Constraint.cs b/src/NUnitFramework/framework/Constraints/Constraint.cs
@@ -93,7 +93,7 @@ protected Constraint(params object?[] args)
         public virtual ConstraintResult ApplyTo<TActual>(ActualValueDelegate<TActual> del)
         {
             if (AsyncToSyncAdapter.IsAsyncOperation(del))
-                return ApplyTo(AsyncToSyncAdapter.Await(() => del.Invoke()));
+                return ApplyTo(AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, () => del.Invoke()));
 
             return ApplyTo(GetTestObject(del));
         }

diff --git a/src/NUnitFramework/framework/Constraints/DelayedConstraint.cs b/src/NUnitFramework/framework/Constraints/DelayedConstraint.cs
@@ -361,14 +361,6 @@ public override async Task<ConstraintResult> ApplyToAsync<TActual>(Func<Task<TAc
             return new DelegatingConstraintResult(this, await BaseConstraint.ApplyToAsync(taskDel));
         }
 
-        private static object? InvokeDelegate<T>(ActualValueDelegate<T> del)
-        {
-            if (AsyncToSyncAdapter.IsAsyncOperation(del))
-                return AsyncToSyncAdapter.Await(() => del.Invoke());
-
-            return del();
-        }
-
         /// <summary>
         /// Returns the string representation of the constraint.
         /// </summary>

diff --git a/src/NUnitFramework/framework/Internal/AsyncEnumerableAdapter.NetCore.cs b/src/NUnitFramework/framework/Internal/AsyncEnumerableAdapter.NetCore.cs
@@ -50,10 +50,10 @@ public AsyncEnumeratorWrapper(IAsyncEnumerator<object?> asyncEnumerator)
             public object? Current => _asyncEnumerator.Current;
 
             public void Dispose()
-                => AsyncToSyncAdapter.Await(() => _asyncEnumerator.DisposeAsync());
+                => AsyncToSyncAdapter.Await(context: null, () => _asyncEnumerator.DisposeAsync());
 
             public bool MoveNext()
-                => AsyncToSyncAdapter.Await<bool>(() => _asyncEnumerator.MoveNextAsync());
+                => AsyncToSyncAdapter.Await<bool>(context: null, () => _asyncEnumerator.MoveNextAsync());
 
             public void Reset()
                 => throw new InvalidOperationException("Can not reset an async enumerable.");

diff --git a/src/NUnitFramework/framework/Internal/AsyncEnumerableAdapter.NetFramework.cs b/src/NUnitFramework/framework/Internal/AsyncEnumerableAdapter.NetFramework.cs
@@ -95,10 +95,10 @@ public AsyncEnumeratorWrapper(AsyncEnumerableShapeInfo shape, object asyncEnumer
             public object? Current => _getCurrentMethod.Invoke();
 
             public void Dispose()
-                => AsyncToSyncAdapter.Await(() => _disposeAsyncMethod.Invoke(_asyncEnumerator, null));
+                => AsyncToSyncAdapter.Await(context: null, () => _disposeAsyncMethod.Invoke(_asyncEnumerator, null));
 
             public bool MoveNext()
-                => AsyncToSyncAdapter.Await<bool>(() => _moveNextAsyncMethod.Invoke());
+                => AsyncToSyncAdapter.Await<bool>(context: null, () => _moveNextAsyncMethod.Invoke());
 
             public void Reset()
                 => throw new InvalidOperationException("Can not reset an async enumerable.");

diff --git a/src/NUnitFramework/framework/Internal/AsyncToSyncAdapter.cs b/src/NUnitFramework/framework/Internal/AsyncToSyncAdapter.cs
@@ -21,12 +21,6 @@ public static bool IsAsyncOperation(Delegate @delegate)
             return IsAsyncOperation(@delegate.GetMethodInfo());
         }
 
-        public static object? Await(Func<object?> invoke)
-            => Await<object?>(default(TestExecutionContext), invoke);
-
-        public static TResult? Await<TResult>(Func<object?> invoke)
-            => Await<TResult>(default(TestExecutionContext), invoke);
-
         public static object? Await(TestExecutionContext? context, Func<object?> invoke)
             => Await<object?>(context, invoke);
 

diff --git a/src/NUnitFramework/framework/Internal/DisposeHelper.cs b/src/NUnitFramework/framework/Internal/DisposeHelper.cs
@@ -30,12 +30,12 @@ public static void EnsureDisposed(object? value)
 #if NETFRAMEWORK
                 if (TryGetAsyncDispose(value.GetType(), out var method))
                 {
-                    AsyncToSyncAdapter.Await(() => method.Invoke(value, null));
+                    AsyncToSyncAdapter.Await(context: null, () => method.Invoke(value, null));
                 }
 #else
                 if (value is IAsyncDisposable asyncDisposable)
                 {
-                    AsyncToSyncAdapter.Await(() => asyncDisposable.DisposeAsync());
+                    AsyncToSyncAdapter.Await(context: null, () => asyncDisposable.DisposeAsync());
                 }
 #endif
                 else if (value is IDisposable disposable)

diff --git a/src/NUnitFramework/framework/Internal/ExceptionHelper.cs b/src/NUnitFramework/framework/Internal/ExceptionHelper.cs
@@ -172,7 +172,7 @@ private static List<Exception> FlattenExceptionHierarchy(Exception exception)
                 {
                     try
                     {
-                        AsyncToSyncAdapter.Await(parameterlessDelegate.DynamicInvokeWithTransparentExceptions);
+                        AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, parameterlessDelegate.DynamicInvokeWithTransparentExceptions);
                     }
                     catch (Exception ex)
                     {

diff --git a/src/NUnitFramework/framework/Internal/Extensions/MethodInfoExtensions.cs b/src/NUnitFramework/framework/Internal/Extensions/MethodInfoExtensions.cs
@@ -12,7 +12,7 @@ internal static class MethodInfoExtensions
         public static TReturn? InvokeMaybeAwait<TReturn>(this MethodInfo m, object?[]? methodArgs)
         {
             if (AsyncToSyncAdapter.IsAsyncOperation(m))
-                return (TReturn?)AsyncToSyncAdapter.Await(() => m.Invoke(null, methodArgs));
+                return (TReturn?)AsyncToSyncAdapter.Await(context: null, () => m.Invoke(null, methodArgs));
             return (TReturn?)m.Invoke(null, methodArgs);
         }
 

diff --git a/src/NUnitFramework/tests/HelperConstraints.cs b/src/NUnitFramework/tests/HelperConstraints.cs
@@ -43,7 +43,7 @@ public override ConstraintResult ApplyTo<TActual>(TActual actual)
                 if (AsyncToSyncAdapter.IsAsyncOperation(@delegate))
                 {
                     stopwatch.Start();
-                    AsyncToSyncAdapter.Await(() => @delegate.DynamicInvoke());
+                    AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, () => @delegate.DynamicInvoke());
                     stopwatch.Stop();
                 }
                 else
-Original file line number
+Diff line change
@@ Expand Up / @@ -230,7 +230,7 @@ public static void Multiple(AsyncTestDelegate testDelegate) @@
             {
                 using (EnterMultipleScope())
                 {
-                    AsyncToSyncAdapter.Await(testDelegate.Invoke);
+                    AsyncToSyncAdapter.Await(TestExecutionContext.CurrentContext, testDelegate.Invoke);
                 }
             }
@@ Expand Down @@