Skip to content

Commit b33d913

Browse files
committed
Add output compilation verification
1 parent eba15cc commit b33d913

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

src/Compilers/CSharp/Test/Semantic/SourceGeneration/GeneratorDriverTests.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2464,7 +2464,7 @@ class C { }
24642464
}));
24652465

24662466
GeneratorDriver driver = CSharpGeneratorDriver.Create([generator], parseOptions: parseOptions, driverOptions: TestOptions.GeneratorDriverOptions);
2467-
driver = driver.RunGenerators(compilation);
2467+
driver = driver.RunGeneratorsAndUpdateCompilation(compilation, out var outputCompilation, out _);
24682468
var runResult = driver.GetRunResult().Results[0];
24692469

24702470
Assert.Single(runResult.GeneratedSources);
@@ -2480,6 +2480,8 @@ internal sealed partial class EmbeddedAttribute : global::System.Attribute
24802480
}
24812481
""", generatedSource.SourceText.ToString());
24822482
Assert.Equal("Microsoft.CodeAnalysis.EmbeddedAttribute.cs", generatedSource.HintName);
2483+
2484+
outputCompilation.VerifyDiagnostics();
24832485
}
24842486

24852487
[Fact]

src/Compilers/VisualBasic/Test/Semantic/SourceGeneration/GeneratorDriverTests.vb

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -646,7 +646,9 @@ End Class
646646
End Sub))
647647

648648
Dim driver As GeneratorDriver = VisualBasicGeneratorDriver.Create(ImmutableArray.Create(Of ISourceGenerator)(generator), parseOptions:=parseOptions, driverOptions:=TestOptions.GeneratorDriverOptions)
649-
driver = driver.RunGenerators(compilation)
649+
Dim outputCompilation As Compilation = Nothing
650+
Dim diagnostics As ImmutableArray(Of Diagnostic) = Nothing
651+
driver = driver.RunGeneratorsAndUpdateCompilation(compilation, outputCompilation, diagnostics)
650652
Dim runResult = driver.GetRunResult().Results(0)
651653

652654
Assert.Single(runResult.GeneratedSources)
@@ -659,6 +661,8 @@ End Class
659661
End Class
660662
End Namespace", generatedSource.SourceText.ToString())
661663
Assert.Equal("Microsoft.CodeAnalysis.EmbeddedAttribute.vb", generatedSource.HintName)
664+
665+
outputCompilation.VerifyDiagnostics()
662666
End Sub
663667

664668
<Fact>

0 commit comments

Comments
 (0)