Added understandable exception for the case when MainConfigAssemblyAt…

…tribute is applied several times
kyrylomyr · Dec 4, 2016 · dee7a58 · dee7a58
1 parent 863e7fa
commit dee7a58
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/ConfigEx/AssemblyLocator.cs b/ConfigEx/AssemblyLocator.cs
@@ -16,12 +16,17 @@ public static Assembly GetEntryAssembly()
         private static Assembly GetAssemblyWithAttribute()
         {
             var assemblies = AppDomain.CurrentDomain.GetAssemblies();
-            var entryAssemblies = from assembly in assemblies
-                                  let attribute = assembly.GetCustomAttributes(typeof(MainConfigAssemblyAttribute)).SingleOrDefault()
-                                  where attribute != null
-                                  select assembly;
+            var entryAssemblies = (from assembly in assemblies
+                                   let attribute = assembly.GetCustomAttributes(typeof (MainConfigAssemblyAttribute)).FirstOrDefault()
+                                   where attribute != null
+                                   select assembly).ToList();
 
-            return entryAssemblies.SingleOrDefault();
+            if (entryAssemblies.Count > 1)
+            {
+                throw new Exception("Only one assembly should be marked as main using the MainConfigAssemblyAttribute");
+            }
+
+            return entryAssemblies.FirstOrDefault();
         }
     }
 }