完善实现的单元测试

dotnet-campus · Sep 18, 2024 · f8cc53a · f8cc53a
1 parent 3bf1973
commit f8cc53a
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 2 deletions.
diff --git a/tests/dotnetCampus.Logger.Analyzer.Tests/Generators/LoggerBridgeGeneratorTest.cs b/tests/dotnetCampus.Logger.Analyzer.Tests/Generators/LoggerBridgeGeneratorTest.cs
@@ -43,11 +43,18 @@ namespace LoggerSample.MainApp;
                     internal partial class LoggerBridgeLinker;
                     """
                 )
-            ]
+            ],
+            references:
+            new[]
+            {
+                MetadataReference.CreateFromFile(typeof(LoggerSample.LoggerIndependentProject.SourceReferenceTarget).Assembly.Location),
+                MetadataReference.CreateFromFile(typeof(LoggerSample.LoggerIndependentLibrary.SourceReferenceTarget).Assembly.Location)
+            } // 加上整个 dotnet 的基础库
+            .Concat(MetadataReferenceProvider.GetDotNetMetadataReferenceList())
         );
 
         driver = driver.RunGenerators(compilation);
         var result = driver.GetRunResult();
         Assert.IsNotNull(result);
     }
-}
+}
diff --git a/tests/dotnetCampus.Logger.Analyzer.Tests/Generators/MetadataReferenceProvider.cs b/tests/dotnetCampus.Logger.Analyzer.Tests/Generators/MetadataReferenceProvider.cs
@@ -0,0 +1,34 @@
+using System.Reflection;
+using Microsoft.CodeAnalysis;
+
+namespace dotnetCampus.Logger.Analyzer.Tests.Generators;
+
+internal static class MetadataReferenceProvider
+{
+    public static IReadOnlyList<MetadataReference> GetDotNetMetadataReferenceList()
+    {
+        if (_cacheList is not null)
+        {
+            return _cacheList;
+        }
+
+        var metadataReferenceList = new List<MetadataReference>();
+        var assembly = Assembly.Load("System.Runtime");
+        foreach (var file in Directory.GetFiles(Path.GetDirectoryName(assembly.Location)!, "*.dll"))
+        {
+            try
+            {
+                metadataReferenceList.Add(MetadataReference.CreateFromFile(file));
+            }
+            catch
+            {
+                // 忽略
+            }
+        }
+
+        _cacheList = metadataReferenceList;
+        return _cacheList;
+    }
+
+    private static IReadOnlyList<MetadataReference>? _cacheList;
+}
diff --git a/tests/dotnetCampus.Logger.Analyzer.Tests/dotnetCampus.Logger.Analyzer.Tests.csproj b/tests/dotnetCampus.Logger.Analyzer.Tests/dotnetCampus.Logger.Analyzer.Tests.csproj
@@ -26,6 +26,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\..\samples\LoggerSample.MainApp\LoggerSample.MainApp.csproj" />
     <ProjectReference Include="..\..\src\dotnetCampus.Logger.Analyzer\dotnetCampus.Logger.Analyzer.csproj" />
   </ItemGroup>