Merge pull request #319 from made-by-karl/referencedAssemblyNames

Fill the ReferencedAssemblyNames Property for all loaded Assemblies

Author: alexanderlinne

ArchUnitNET/Domain/Assembly.cs

@@ -1,6 +1,5 @@
 using System.Collections.Generic;
 using System.Linq;
-using JetBrains.Annotations;
 
 namespace ArchUnitNET.Domain
 {
@@ -10,7 +9,7 @@ public Assembly(
             string name,
             string fullName,
             bool isOnlyReferenced,
-            [CanBeNull] List<string> referencedAssemblyNames
+            List<string> referencedAssemblyNames
         )
         {
             Name = name;
@@ -23,7 +22,6 @@ [CanBeNull] List<string> referencedAssemblyNames
 
         public string Name { get; }
 
-        [CanBeNull]
         public List<string> ReferencedAssemblyNames { get; }
         public string FullName { get; }
 

ArchUnitNET/Loader/ArchBuilder.cs

@@ -38,13 +38,11 @@ public ArchBuilder()
         public IEnumerable<Assembly> Assemblies => _assemblyRegistry.Assemblies;
         public IEnumerable<Namespace> Namespaces => _namespaceRegistry.Namespaces;
 
-        public void AddAssembly(
-            [NotNull] AssemblyDefinition moduleAssembly,
-            bool isOnlyReferenced,
-            [CanBeNull] IEnumerable<AssemblyNameReference> moduleReferences
-        )
+        public void AddAssembly([NotNull] AssemblyDefinition moduleAssembly, bool isOnlyReferenced)
         {
-            var references = moduleReferences?.Select(reference => reference.Name).ToList();
+            var references = moduleAssembly
+                .MainModule.AssemblyReferences.Select(reference => reference.Name)
+                .ToList();
 
             if (!_assemblyRegistry.ContainsAssembly(moduleAssembly.Name.FullName))
             {

ArchUnitNET/Loader/ArchLoader.cs

@@ -128,7 +128,7 @@ private void LoadModule(
                 var processedAssemblies = new List<AssemblyNameReference> { module.Assembly.Name };
                 var resolvedModules = new List<ModuleDefinition>();
                 _assemblyResolver.AddLib(module.Assembly);
-                _archBuilder.AddAssembly(module.Assembly, false, module.AssemblyReferences);
+                _archBuilder.AddAssembly(module.Assembly, false);
                 foreach (var assemblyReference in module.AssemblyReferences)
                 {
                     if (includeDependencies && recursive)
@@ -151,7 +151,7 @@ private void LoadModule(
                                 var assemblyDefinition =
                                     _assemblyResolver.Resolve(assemblyReference)
                                     ?? throw new AssemblyResolutionException(assemblyReference);
-                                _archBuilder.AddAssembly(assemblyDefinition, false, null);
+                                _archBuilder.AddAssembly(assemblyDefinition, false);
                                 resolvedModules.AddRange(assemblyDefinition.Modules);
                             }
                         }
@@ -197,7 +197,7 @@ FilterFunc filterFunc
                 var filterResult = filterFunc?.Invoke(assemblyDefinition);
                 if (filterResult?.LoadThisAssembly != false)
                 {
-                    _archBuilder.AddAssembly(assemblyDefinition, false, null);
+                    _archBuilder.AddAssembly(assemblyDefinition, false);
                     resolvedModules.AddRange(assemblyDefinition.Modules);
                 }
 

ArchUnitNETTests/Loader/ArchLoaderTests.cs

@@ -63,12 +63,16 @@ public void SameFullNameInMultipleAssemblies()
         [Fact]
         public void LoadAssembliesIncludingRecursiveDependencies()
         {
-            var archUnitNetTestArchitectureWithRecursiveDependencies = new ArchLoader()
+            var architecture = new ArchLoader()
                 .LoadAssembliesIncludingDependencies(new[] { typeof(BaseClass).Assembly }, true)
                 .Build();
 
-            Assert.True(
-                archUnitNetTestArchitectureWithRecursiveDependencies.Assemblies.Count() > 100
+            Assert.True(architecture.Assemblies.Count() > 100);
+
+            // Check for well-known assemblies, that ReferencedAssemblyNames is not empty
+            Assert.All(
+                architecture.Assemblies.Where(x => x.Name.StartsWith("ArchUnit")),
+                x => Assert.NotEmpty(x.ReferencedAssemblyNames)
             );
         }
 
@@ -85,6 +89,9 @@ public void LoadAssembliesRecursivelyWithCustomFilter()
                 .Build();
 
             Assert.Equal(3, architecture.Assemblies.Count());
+
+            // Check that ReferencedAssemblyNames is not empty
+            Assert.All(architecture.Assemblies, x => Assert.NotEmpty(x.ReferencedAssemblyNames));
         }
 
         [Fact]