using NUnit.Framework.Interfaces; using NUnit.Framework.Internal; using System; using System.Collections; using System.Reflection; namespace NUnit.Framework { [AttributeUsage(AttributeTargets.Parameter, AllowMultiple = true, Inherited = false)] public class ValueSourceAttribute : DataAttribute, IParameterDataSource { private readonly string sourceName; private readonly Type sourceType; public string SourceName => sourceName; public Type SourceType => sourceType; public ValueSourceAttribute(string sourceName) { this.sourceName = sourceName; } public ValueSourceAttribute(Type sourceType, string sourceName) { this.sourceType = sourceType; this.sourceName = sourceName; } public IEnumerable GetData(ParameterInfo parameter) { return GetDataSource(parameter); } private IEnumerable GetDataSource(ParameterInfo parameter) { Type reflectedType = sourceType; if ((object)reflectedType == null) reflectedType = parameter.Member.ReflectedType; if (sourceName == null) return Reflect.Construct(reflectedType) as IEnumerable; MemberInfo[] member = reflectedType.GetMember(sourceName, BindingFlags.Instance | BindingFlags.Static | BindingFlags.Public | BindingFlags.NonPublic); if (member.Length == 1) { MemberInfo memberInfo = member[0]; object obj = Reflect.Construct(reflectedType); FieldInfo fieldInfo = memberInfo as FieldInfo; if ((object)fieldInfo != null) return (IEnumerable)fieldInfo.GetValue(obj); PropertyInfo propertyInfo = memberInfo as PropertyInfo; if ((object)propertyInfo != null) return (IEnumerable)propertyInfo.GetValue(obj, null); MethodInfo methodInfo = memberInfo as MethodInfo; if ((object)methodInfo != null) return (IEnumerable)methodInfo.Invoke(obj, null); } return null; } } }