Handle UnavailableType when creating an Attribute

Author: alexanderlinne

ArchUnit.sln

@@ -44,6 +44,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "ArchUnitNET.TUnit", "ArchUn
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "ArchUnitNET.TUnitTests", "ArchUnitNET.TUnitTests\ArchUnitNET.TUnitTests.csproj", "{C54143CE-3256-4313-B991-0706705EEA1D}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "FilteredDirectoryAssemblyAttributeAssembly", "FilteredDirectoryAssemblyAttributeAssembly\FilteredDirectoryAssemblyAttributeAssembly.csproj", "{AA695589-8CCC-4474-9A7F-01A6376C375A}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -130,6 +132,10 @@ Global
 		{C54143CE-3256-4313-B991-0706705EEA1D}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{C54143CE-3256-4313-B991-0706705EEA1D}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{C54143CE-3256-4313-B991-0706705EEA1D}.Release|Any CPU.Build.0 = Release|Any CPU
+		{AA695589-8CCC-4474-9A7F-01A6376C375A}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{AA695589-8CCC-4474-9A7F-01A6376C375A}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{AA695589-8CCC-4474-9A7F-01A6376C375A}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{AA695589-8CCC-4474-9A7F-01A6376C375A}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
@@ -141,5 +147,6 @@ Global
 		{0243F2D4-AC89-4561-A936-D647B6BB821F} = {B1191F18-91CB-4387-B775-A5EB64D3AC30}
 		{5A24529B-1794-4080-ADCC-77440BA0A0B3} = {B1191F18-91CB-4387-B775-A5EB64D3AC30}
 		{E6CB8C69-25F5-4C94-8EA3-D56E444EB46B} = {B1191F18-91CB-4387-B775-A5EB64D3AC30}
+		{AA695589-8CCC-4474-9A7F-01A6376C375A} = {B1191F18-91CB-4387-B775-A5EB64D3AC30}
 	EndGlobalSection
 EndGlobal

ArchUnitNET/Loader/MonoCecilAttributeExtensions.cs

@@ -20,7 +20,12 @@ TypeFactory typeFactory
             var attributeType = typeFactory.GetOrCreateStubTypeInstanceFromTypeReference(
                 attributeTypeReference
             );
-            if (!(attributeType.Type is Attribute attribute))
+            var attribute = attributeType.Type as Attribute;
+            if (attributeType.Type is UnavailableType unavailableType)
+            {
+                attribute = new Attribute(unavailableType, null, null);
+            }
+            if (attribute == null)
             {
                 throw new ArgumentException(
                     $"Attribute type {attributeType.Type.FullName} is not an attribute."

ArchUnitNETTests/Loader/ArchLoaderTests.cs

@@ -2,6 +2,7 @@
 extern alias OtherLoaderTestAssemblyAlias;
 
 using System;
+using System.IO;
 using System.Linq;
 using ArchUnitNET.Domain;
 using ArchUnitNET.Domain.Extensions;
@@ -132,15 +133,18 @@ public void UnavailableTypeTest()
             // When loading an assembly from a file, there are situations where the assemblies dependencies are not
             // available in the current AppDomain. This test checks that the loader does not throw an exception in this
             // case.
-            var assemblyPath = AppDomain.CurrentDomain.BaseDirectory[
+            var currentAssemblyPath = AppDomain.CurrentDomain.BaseDirectory[
                 ..AppDomain.CurrentDomain.BaseDirectory.IndexOf(
                     @"ArchUnitNETTests",
                     StringComparison.InvariantCulture
                 )
             ];
+            var assemblySearchPath = Path.Combine(
+                [currentAssemblyPath, "TestAssemblies", "FilteredDirectoryLoaderTestAssembly"]
+            );
             var architecture = new ArchLoader()
                 .LoadFilteredDirectory(
-                    assemblyPath,
+                    assemblySearchPath,
                     "FilteredDirectoryLoaderTestAssembly.dll",
                     System.IO.SearchOption.AllDirectories
                 )

FilteredDirectoryAssemblyAttributeAssembly/AssemblyAttribute.cs

@@ -0,0 +1,6 @@
+namespace FilteredDirectoryAssemblyAttributeAssembly;
+
+[AttributeUsage(AttributeTargets.Assembly)]
+public class AssemblyAttribute : Attribute
+{
+}

FilteredDirectoryAssemblyAttributeAssembly/FilteredDirectoryAssemblyAttributeAssembly.csproj

@@ -0,0 +1,9 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFramework>net9.0</TargetFramework>
+        <ImplicitUsings>enable</ImplicitUsings>
+        <Nullable>enable</Nullable>
+    </PropertyGroup>
+
+</Project>

TestAssemblies/FilteredDirectoryLoaderTestAssembly/Class1.cs

@@ -1,5 +1,7 @@
 using Serilog;
 
+[assembly:FilteredDirectoryAssemblyAttributeAssembly.AssemblyAttribute]
+
 namespace FilteredDirectoryLoaderTestAssembly;
 
 public class Class1

TestAssemblies/FilteredDirectoryLoaderTestAssembly/FilteredDirectoryLoaderTestAssembly.csproj

@@ -10,4 +10,11 @@
     <PackageReference Include="Serilog" Version="4.2.0" />
     <PackageReference Include="Serilog.Sinks.Console" Version="6.0.0" />
   </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\..\FilteredDirectoryAssemblyAttributeAssembly\FilteredDirectoryAssemblyAttributeAssembly.csproj">
+        <!-- Do not copy the referenced assembly to the output directory -->
+        <!-- in order to test handling of unavailable types as attribute assemblies -->
+        <Private>False</Private>
+    </ProjectReference>
+  </ItemGroup>
 </Project>