From 47e6ad2e072bed9f381c947b565b2f818c649121 Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Thu, 26 May 2022 17:30:53 -0500
Subject: Fix DncHostGeneratorTests.

---
 .../Bal/test/WixToolsetTest.Dnc.HostGenerator/DncHostGeneratorTests.cs | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src')

diff --git a/src/ext/Bal/test/WixToolsetTest.Dnc.HostGenerator/DncHostGeneratorTests.cs b/src/ext/Bal/test/WixToolsetTest.Dnc.HostGenerator/DncHostGeneratorTests.cs
index 6ede1089..ae55b2b5 100644
--- a/src/ext/Bal/test/WixToolsetTest.Dnc.HostGenerator/DncHostGeneratorTests.cs
+++ b/src/ext/Bal/test/WixToolsetTest.Dnc.HostGenerator/DncHostGeneratorTests.cs
@@ -18,6 +18,7 @@ namespace WixToolsetTest.Dnc.HostGenerator
     {
         static readonly MetadataReference MbaCoreAssembly = MetadataReference.CreateFromFile(typeof(BootstrapperApplicationFactoryAttribute).Assembly.Location);
 
+        // https://github.com/dotnet/roslyn/blob/main/docs/features/source-generators.cookbook.md#unit-testing-of-generators
         [Fact]
         public async Task FailsBuildWhenMissingAttribute()
         {
@@ -42,6 +43,7 @@ namespace Test
                 TestState = 
                 {
                     Sources = { code },
+                    ReferenceAssemblies = ReferenceAssemblies.Net.Net60,
                     AdditionalReferences = { MbaCoreAssembly },
                     ExpectedDiagnostics =
                     {
@@ -80,6 +82,7 @@ namespace Test
                     {
                         (typeof(DncHostGenerator), "WixToolset.Dnc.Host.g.cs", SourceText.From(generated, Encoding.UTF8, SourceHashAlgorithm.Sha256)),
                     },
+                    ReferenceAssemblies = ReferenceAssemblies.Net.Net60,
                     AdditionalReferences = { MbaCoreAssembly },
                 },
             }.RunAsync();
-- 
cgit v1.2.3-55-g6feb