|
5 | 5 |
|
6 | 6 | using System.Reflection; |
7 | 7 |
|
| 8 | +using Microsoft.Extensions.DependencyInjection; |
| 9 | + |
8 | 10 | namespace Bss.Testing.Xunit.Sdk; |
9 | 11 |
|
10 | 12 | [DataDiscoverer("Bss.Testing.Xunit.Sdk.ServiceProviderMemberDataDiscoverer", "Bss.Testing.Xunit")] |
11 | 13 | [AttributeUsage(AttributeTargets.Method, AllowMultiple = true)] |
12 | | -public class ServiceProviderMemberDataAttribute(string methodName) : DataAttribute |
| 14 | +public class ServiceProviderMemberDataAttribute(string methodOrPropertyName) : DataAttribute |
13 | 15 | { |
14 | | - public Type? MemberType { get; set; } |
| 16 | + public Type MemberType { get; set; } |
15 | 17 |
|
16 | | - string MemberName { get; set; } = methodName; |
| 18 | + string MemberName { get; set; } = methodOrPropertyName; |
17 | 19 |
|
18 | | - public override IEnumerable<object[]>? GetData(MethodInfo testMethod) => null; |
| 20 | + public override IEnumerable<object[]> GetData(MethodInfo testMethod) => null; |
19 | 21 |
|
20 | | - public IEnumerable<object[]>? GetData(MethodInfo testMethod, IServiceProvider? serviceProvider) |
| 22 | + public IEnumerable<object[]> GetData(MethodInfo testMethod, IServiceProvider serviceProvider) |
21 | 23 | { |
22 | 24 | var type = this.MemberType ?? testMethod.DeclaringType; |
23 | | - if (type == null) |
24 | | - { |
25 | | - throw new ArgumentException( |
26 | | - string.Format( |
27 | | - CultureInfo.CurrentCulture, |
28 | | - "Could not find type {0}", |
29 | | - type?.FullName) |
30 | | - ); |
31 | | - } |
| 25 | + var accessor = this.GetMethodAccessor(type, serviceProvider) |
| 26 | + ?? this.GetPropertyAccessor(type, serviceProvider); |
32 | 27 |
|
33 | | - var accessor = this.GetMethodAccessor(type, serviceProvider); |
34 | 28 | if (accessor == null) |
35 | 29 | { |
36 | 30 | throw new ArgumentException( |
37 | 31 | string.Format( |
38 | 32 | CultureInfo.CurrentCulture, |
39 | | - "Could not find public static method named '{0}' on {1}{2}", |
| 33 | + "Could not find parameterless method or property named '{0}' on {1} provided in ServiceProviderMemberDataAttribute", |
40 | 34 | this.MemberName, |
41 | | - type?.FullName, |
42 | | - " with parameter types: IServiceProvider") |
| 35 | + type?.FullName) |
43 | 36 | ); |
44 | 37 | } |
45 | 38 |
|
46 | 39 | var obj = accessor(); |
47 | 40 | if (obj == null) |
48 | 41 | { |
49 | | - return (IEnumerable<object[]>) Array.Empty<object>(); |
| 42 | + return null; |
50 | 43 | } |
51 | 44 |
|
52 | 45 | if (obj is not IEnumerable dataItems) |
53 | 46 | { |
54 | | - throw new ArgumentException(string.Format(CultureInfo.CurrentCulture, "Method {0} on {1} did not return IEnumerable", this.MemberName, type?.FullName)); |
| 47 | + throw new ArgumentException(string.Format(CultureInfo.CurrentCulture, "Method/property {0} on {1} did not return IEnumerable", this.MemberName, type?.FullName)); |
55 | 48 | } |
56 | 49 |
|
57 | | - return dataItems.Cast<object>().Select(item => this.ConvertDataItem(testMethod, item))!; |
| 50 | + return dataItems.Cast<object>().Select(item => this.ConvertDataItem(testMethod, item)); |
58 | 51 | } |
59 | 52 |
|
60 | | - protected Func<object>? GetMethodAccessor(Type type, IServiceProvider? serviceProvider) |
| 53 | + private Func<object?>? GetMethodAccessor(Type type, IServiceProvider serviceProvider) |
61 | 54 | { |
62 | 55 | MethodInfo? methodInfo = null; |
63 | 56 | for (var reflectionType = type; reflectionType != null; reflectionType = reflectionType.GetTypeInfo().BaseType) |
64 | 57 | { |
65 | | - var runtimeMethodsWithGivenName = reflectionType.GetRuntimeMethods() |
66 | | - .Where(m => m.Name == this.MemberName) |
67 | | - .ToArray(); |
| 58 | + methodInfo = reflectionType |
| 59 | + .GetRuntimeMethods() |
| 60 | + .FirstOrDefault(m => m.Name == this.MemberName); |
| 61 | + if (methodInfo != null) |
| 62 | + { |
| 63 | + break; |
| 64 | + } |
| 65 | + } |
| 66 | + |
| 67 | + if (methodInfo == null) |
| 68 | + { |
| 69 | + return null; |
| 70 | + } |
68 | 71 |
|
69 | | - methodInfo = runtimeMethodsWithGivenName |
70 | | - .FirstOrDefault(m => m.GetParameters() |
71 | | - .Count(x => x.ParameterType.IsAssignableTo(typeof(IServiceProvider))) == 1); |
| 72 | + var @object = ActivatorUtilities.CreateInstance(serviceProvider, type); |
72 | 73 |
|
73 | | - if (methodInfo != null) |
| 74 | + return () => methodInfo.Invoke(@object, null); |
| 75 | + } |
| 76 | + |
| 77 | + private Func<object?>? GetPropertyAccessor(Type type, IServiceProvider serviceProvider) |
| 78 | + { |
| 79 | + PropertyInfo? propertyInfo = null; |
| 80 | + for (var reflectionType = type; reflectionType != null; reflectionType = reflectionType.GetTypeInfo().BaseType) |
| 81 | + { |
| 82 | + propertyInfo = reflectionType |
| 83 | + .GetProperties() |
| 84 | + .FirstOrDefault(m => m.Name == this.MemberName); |
| 85 | + |
| 86 | + if (propertyInfo != null) |
74 | 87 | { |
75 | 88 | break; |
76 | 89 | } |
77 | 90 | } |
78 | 91 |
|
79 | | - if (methodInfo == null || !methodInfo.IsStatic) |
| 92 | + if (propertyInfo == null) |
80 | 93 | { |
81 | 94 | return null; |
82 | 95 | } |
83 | 96 |
|
84 | | - return () => methodInfo.Invoke(null, [serviceProvider])!; |
| 97 | + var @object = ActivatorUtilities.CreateInstance(serviceProvider, type); |
| 98 | + |
| 99 | + return () => propertyInfo.GetValue(@object); |
85 | 100 | } |
86 | 101 |
|
87 | | - protected object[]? ConvertDataItem(MethodInfo testMethod, object? item) |
| 102 | + private object[]? ConvertDataItem(MethodInfo testMethod, object? item) |
88 | 103 | { |
89 | 104 | if (item == null) |
90 | 105 | { |
|
0 commit comments