Add preliminary support for building an IL test runner using the sour…

…ce generator.

src/tests/Common/ILTestRunner/ILTestRunner.csproj

@@ -0,0 +1,19 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <OutputType>Exe</OutputType>
+    <TargetFramework>net6.0</TargetFramework>
+    <ImplicitUsings>enable</ImplicitUsings>
+    <Nullable>enable</Nullable>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="$(RepoRoot)/src/tests/Common/XUnitWrapperGenerator/XUnitWrapperGenerator.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
+  </ItemGroup>
+
+  <Import Project="$(RepoRoot)/src/tests/Common/XUnitWrapperGenerator/XUnitWrapperGenerator.props" />
+
+  <ItemGroup>
+    <ProjectReference Include="$(RepoRoot)\src\tests\JIT\IL_Conformance\Old\directed\AutoInit.ilproj" />
+  </ItemGroup>
+</Project>

src/tests/Common/XUnitWrapperGenerator/XUnitWrapperGenerator.cs

@@ -18,13 +18,23 @@ public sealed class XUnitWrapperGenerator : IIncrementalGenerator
 {
     public void Initialize(IncrementalGeneratorInitializationContext context)
     {
-        context.RegisterImplementationSourceOutput(
-            context.SyntaxProvider.CreateSyntaxProvider(
+        var methodsInSource = context.SyntaxProvider.CreateSyntaxProvider(
                 static (node, ct) =>
                     node.IsKind(SyntaxKind.MethodDeclaration)
                         && node is MethodDeclarationSyntax method
                         && method.AttributeLists.Count > 0,
-                static (context, ct) => (IMethodSymbol)context.SemanticModel.GetDeclaredSymbol(context.Node)!)
+                static (context, ct) => (IMethodSymbol)context.SemanticModel.GetDeclaredSymbol(context.Node)!);
+
+        var methodsInReferencedAssemblies = context.MetadataReferencesProvider.Combine(context.CompilationProvider).SelectMany((data, ct) =>
+        {
+            ExternallyReferencedTestMethodsVisitor visitor = new();
+            return visitor.Visit(data.Right.GetAssemblyOrModuleSymbol(data.Left))!;
+        });
+
+        var allMethods = methodsInSource.Collect().Combine(methodsInReferencedAssemblies.Collect()).SelectMany((methods, ct) => methods.Left.AddRange(methods.Right));
+
+        context.RegisterImplementationSourceOutput(
+            allMethods
             .Combine(context.AnalyzerConfigOptionsProvider)
             .SelectMany((data, ct) => ImmutableArray.CreateRange(GetTestMethodInfosForMethod(data.Left, data.Right)))
             .Collect(),
@@ -40,6 +50,58 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
             });
     }
 
+    private class ExternallyReferencedTestMethodsVisitor : SymbolVisitor<IEnumerable<IMethodSymbol>>
+    {
+        public override IEnumerable<IMethodSymbol>? VisitAssembly(IAssemblySymbol symbol)
+        {
+            if (symbol.Name.StartsWith("autoinit.exe"))
+            {
+
+            }
+            return Visit(symbol.GlobalNamespace);
+        }
+
+        public override IEnumerable<IMethodSymbol>? VisitModule(IModuleSymbol symbol)
+        {
+            return Visit(symbol.GlobalNamespace);
+        }
+
+        public override IEnumerable<IMethodSymbol>? VisitNamespace(INamespaceSymbol symbol)
+        {
+            foreach (var type in symbol.GetMembers())
+            {
+                foreach (var result in Visit(type) ?? Array.Empty<IMethodSymbol>())
+                {
+                    yield return result;
+                }
+            }
+        }
+
+        public override IEnumerable<IMethodSymbol>? VisitNamedType(INamedTypeSymbol symbol)
+        {
+            if (symbol.DeclaredAccessibility == Accessibility.Public)
+            {
+                foreach (var type in symbol.GetMembers())
+                {
+                    foreach (var result in Visit(type) ?? Array.Empty<IMethodSymbol>())
+                    {
+                        yield return result;
+                    }
+                }
+            }
+        }
+
+        public override IEnumerable<IMethodSymbol>? VisitMethod(IMethodSymbol symbol)
+        {
+            if (symbol.DeclaredAccessibility == Accessibility.Public
+                && symbol.GetAttributes().Any(attr => attr.AttributeClass?.ContainingNamespace.Name == "Xunit"))
+            {
+                return new[] { symbol };
+            }
+            return Array.Empty<IMethodSymbol>();
+        }
+    }
+
     private static IEnumerable<ITestInfo> GetTestMethodInfosForMethod(IMethodSymbol method, AnalyzerConfigOptionsProvider options)
     {
         bool factAttribute = false;

src/tests/Common/XUnitWrapperGenerator/XUnitWrapperGenerator.csproj

@@ -4,6 +4,7 @@
     <TargetFramework>netstandard2.0</TargetFramework>
     <Nullable>enable</Nullable>
     <EnableDefaultItems>true</EnableDefaultItems>
+    <IsRoslynComponent>true</IsRoslynComponent>
   </PropertyGroup>
 
   <ItemGroup>

src/tests/JIT/IL_Conformance/Old/directed/AutoInit.il

@@ -3,11 +3,21 @@
 
 
 .assembly extern legacy library mscorlib {}
-.assembly autoinit.exe{}
+.assembly autoinit.dll{}
+.assembly extern xunit.core
+{
+	.publickeytoken = (
+		8d 05 b1 bb 7a 6f db 6c
+	)
+	.ver 2:4:2:0
+}
 
 .class public auto_init {
 
-.method public static int32 main(class [mscorlib]System.String[]) {
+.method public static int32 main() {
+    .custom instance void [xunit.core]Xunit.FactAttribute::.ctor() = (
+            01 00 00 00
+        )
 .zeroinit
 .entrypoint
 .locals	(int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32,int32)