TUnit configureawait false

Author: distantcam

src/AutoCtor.Tests/ExampleTests.cs

@@ -46,7 +46,8 @@ public async Task CodeCompilesProperly(CodeFileTheoryData theoryData)
 
         await Assert.That(outputCompilation.GetDiagnostics(TestHelper.CancellationToken)
             .Where(d => !theoryData.IgnoredCompileDiagnostics.Contains(d.Id)))
-            .IsEmpty();
+            .IsEmpty()
+            .ConfigureAwait(false);
     }
 
 #if ROSLYN_4_4

src/AutoCtor.Tests/GeneratedAttributeTests.cs

@@ -39,8 +39,10 @@ public async Task AttributeCompilesProperly()
         var outputCompilationDiagnostics = outputCompilation
             .GetDiagnostics(TestHelper.CancellationToken);
 
-        await Assert.That(diagnostics).IsEmpty();
-        await Assert.That(outputCompilationDiagnostics).IsEmpty();
+        await Assert.That(diagnostics).IsEmpty()
+            .ConfigureAwait(false);
+        await Assert.That(outputCompilationDiagnostics).IsEmpty()
+            .ConfigureAwait(false);
     }
 
     [Test]
@@ -66,8 +68,10 @@ public async Task PreserveAttributesTest()
         var outputCompilationDiagnostics = outputCompilation
             .GetDiagnostics(TestHelper.CancellationToken);
 
-        await Assert.That(diagnostics).IsEmpty();
-        await Assert.That(outputCompilationDiagnostics).IsEmpty();
+        await Assert.That(diagnostics).IsEmpty()
+            .ConfigureAwait(false);
+        await Assert.That(outputCompilationDiagnostics).IsEmpty()
+            .ConfigureAwait(false);
     }
 
     [Test]
@@ -102,7 +106,9 @@ public async Task EnsureGeneratedAttributesAreNotExternallyVisible()
         var outputCompilationDiagnostics = outputCompilation
             .GetDiagnostics(TestHelper.CancellationToken);
 
-        await Assert.That(diagnostics).IsEmpty();
-        await Assert.That(outputCompilationDiagnostics).IsEmpty();
+        await Assert.That(diagnostics).IsEmpty()
+            .ConfigureAwait(false);
+        await Assert.That(outputCompilationDiagnostics).IsEmpty()
+            .ConfigureAwait(false);
     }
 }

src/AutoCtor.Tests/Issue73.cs

@@ -41,8 +41,10 @@ public async Task CodeCompilesWithoutErrors()
             .GetDiagnostics(TestHelper.CancellationToken)
             .Where(d => !ignoredWarnings.Contains(d.Id));
 
-        await Assert.That(diagnostics).IsEmpty();
-        await Assert.That(outputCompilationDiagnostics).IsEmpty();
+        await Assert.That(diagnostics).IsEmpty()
+            .ConfigureAwait(false);
+        await Assert.That(outputCompilationDiagnostics).IsEmpty()
+            .ConfigureAwait(false);
     }
 
     private static CompilationBuilder Common()

src/AutoCtor.Tests/Utilities/ExampleTestsHelper.cs

@@ -54,8 +54,10 @@ public static async Task AssertRunsEqual(
         using var _ = Assert.Multiple();
 
         // Both runs should have the same tracked steps
-        await Assert.That(trackedSteps1).IsNotEmpty();
-        await Assert.That(trackedSteps2.Keys).IsEquivalentTo(trackedSteps1.Keys);
+        await Assert.That(trackedSteps1).IsNotEmpty()
+            .ConfigureAwait(false);
+        await Assert.That(trackedSteps2.Keys).IsEquivalentTo(trackedSteps1.Keys)
+            .ConfigureAwait(false);
 
         // Get the IncrementalGeneratorRunStep collection for each run
         foreach (var (trackingName, runSteps1) in trackedSteps1)
@@ -72,17 +74,18 @@ await AssertStepsEqual(runSteps1, runSteps2)
         static Dictionary<string, ImmutableArray<IncrementalGeneratorRunStep>> GetTrackedSteps(
             GeneratorDriverRunResult runResult, IEnumerable<string> trackingNames)
             => runResult
-                    .Results[0] // We're only running a single generator, so this is safe
-                    .TrackedSteps // Get the pipeline outputs
-                    .Where(step => trackingNames.Contains(step.Key)) // filter to known steps
-                    .ToDictionary(x => x.Key, x => x.Value); // Convert to a dictionary
+                .Results[0] // We're only running a single generator, so this is safe
+                .TrackedSteps // Get the pipeline outputs
+                .Where(step => trackingNames.Contains(step.Key)) // filter to known steps
+                .ToDictionary(x => x.Key, x => x.Value); // Convert to a dictionary
     }
 
     private static async Task AssertStepsEqual(
         ImmutableArray<IncrementalGeneratorRunStep> runSteps1,
         ImmutableArray<IncrementalGeneratorRunStep> runSteps2)
     {
-        await Assert.That(runSteps2.Length).IsEqualTo(runSteps2.Length);
+        await Assert.That(runSteps2.Length).IsEqualTo(runSteps2.Length)
+            .ConfigureAwait(false);
 
         for (var i = 0; i < runSteps1.Length; i++)
         {
@@ -93,7 +96,8 @@ private static async Task AssertStepsEqual(
             var outputs1 = runStep1.Outputs.Select(x => x.Value);
             var outputs2 = runStep2.Outputs.Select(x => x.Value);
 
-            await Assert.That(outputs2).IsEquivalentTo(outputs1);
+            await Assert.That(outputs2).IsEquivalentTo(outputs1)
+                .ConfigureAwait(false);
 
             // Therefore, on the second run the results should always be cached or unchanged!
             // - Unchanged is when the _input_ has changed, but the output hasn't
@@ -102,7 +106,8 @@ private static async Task AssertStepsEqual(
                 [IncrementalStepRunReason.Cached, IncrementalStepRunReason.Unchanged];
             foreach (var (_, reason) in runStep2.Outputs)
             {
-                await Assert.That(reason).IsIn(acceptableReasons);
+                await Assert.That(reason).IsIn(acceptableReasons)
+                    .ConfigureAwait(false);
             }
         }
     }

src/AutoCtor.Tests/Utilities/TestHelper.cs

@@ -1,5 +1,13 @@
-public static class TestHelper
+using System.Runtime.CompilerServices;
+using TUnit.Assertions.Core;
+
+public static class TestHelper
 {
     public static CancellationToken CancellationToken =>
         TestContext.Current?.Execution?.CancellationToken ?? CancellationToken.None;
+
+    public static ConfiguredTaskAwaitable<TValue?> ConfigureAwait<TValue>(this Assertion<TValue> assertion, bool continueOnCapturedContext)
+    {
+        return assertion.AssertAsync().ConfigureAwait(continueOnCapturedContext);
+    }
 }