Implement IncludingInherited for Controller attributes

Author: melikpehlivanov

src/MyTested.AspNetCore.Mvc.Abstractions/Internal/TestContexts/ComponentTestContext.cs

@@ -22,8 +22,11 @@ public abstract class ComponentTestContext : HttpTestContext
 
         public object Component => this.component ??= this.ComponentConstructionDelegate();
 
-        public IEnumerable<object> ComponentAttributes 
-            => this.componentAttributes ??= Reflection.GetCustomAttributes(this.Component);
+        public IEnumerable<object> ComponentAttributes
+        {
+            get => this.componentAttributes ??= Reflection.GetCustomAttributes(this.Component);
+            private set => this.componentAttributes = value;
+        }
 
         public IDictionary<Type, object> AggregatedDependencies 
             => this.aggregatedDependencies ??= new Dictionary<Type, object>();
@@ -116,6 +119,9 @@ public void Apply<TMethodResult>(InvocationTestContext<TMethodResult> invocation
             this.CaughtException = invocationTestContext.CaughtException;
         }
 
+        public void IncludeInheritedComponentAttributes()
+            => this.ComponentAttributes = Reflection.GetCustomAttributesIncludingInherited(this.Component);
+
         protected virtual object ConvertMethodResult(object convertibleMethodResult) => convertibleMethodResult;
     }
 }

src/MyTested.AspNetCore.Mvc.Abstractions/Utilities/Reflection.cs

@@ -19,6 +19,7 @@ public static class Reflection
     {
         private static readonly ConcurrentDictionary<Type, ConstructorInfo> TypesWithOneConstructorCache = new ConcurrentDictionary<Type, ConstructorInfo>();
         private static readonly ConcurrentDictionary<Type, IEnumerable<object>> TypeAttributesCache = new ConcurrentDictionary<Type, IEnumerable<object>>();
+        private static readonly ConcurrentDictionary<Type, IEnumerable<object>> TypeInheritedAttributesCache = new ConcurrentDictionary<Type, IEnumerable<object>>();
         private static readonly ConcurrentDictionary<MethodInfo, IEnumerable<object>> MethodAttributesCache = new ConcurrentDictionary<MethodInfo, IEnumerable<object>>();
         private static readonly ConcurrentDictionary<Type, string> FriendlyTypeNames = new ConcurrentDictionary<Type, string>();
         private static readonly ConcurrentDictionary<Type, string> FullFriendlyTypeNames = new ConcurrentDictionary<Type, string>();
@@ -291,6 +292,18 @@ public static IEnumerable<object> GetCustomAttributes(object obj)
                 .GetOrAdd(type, _ => type.GetTypeInfo().GetCustomAttributes(false));
         }
 
+        /// <summary>
+        /// Gets custom attributes including inherited ones on the provided object.
+        /// </summary>
+        /// <param name="obj">Object decorated with custom attribute.</param>
+        /// <returns>IEnumerable of objects representing the custom attributes.</returns>
+        public static IEnumerable<object> GetCustomAttributesIncludingInherited(object obj)
+        {
+            var type = obj.GetType();
+            return TypeInheritedAttributesCache
+                .GetOrAdd(type, _ => type.GetTypeInfo().GetCustomAttributes(true));
+        }
+
         public static IEnumerable<object> GetCustomAttributes(MethodInfo method)
             => MethodAttributesCache
                 .GetOrAdd(method, _ => method.GetCustomAttributes(false));

src/MyTested.AspNetCore.Mvc.Controllers/Builders/Attributes/ControllerAttributesTestBuilder.cs

@@ -27,7 +27,11 @@ public ControllerAttributesTestBuilder(ControllerTestContext testContext)
         /// <inheritdoc />
         public IControllerAttributesTestBuilder AndAlso() => this;
 
+        /// <inheritdoc />
         public IAndControllerAttributesTestBuilder IncludingInherited()
-            => throw new System.NotImplementedException();
+        {
+            this.TestContext.IncludeInheritedComponentAttributes();
+            return this;
+        }
     }
 }