aboutsummaryrefslogtreecommitdiff
path: root/BackEnd/Timeline.Tests/Helpers/ParameterInfoAssertions.cs
blob: d3e5a41e55893b54e9aca9fac29891a6c5bdd118 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
using FluentAssertions;
using FluentAssertions.Execution;
using FluentAssertions.Formatting;
using FluentAssertions.Primitives;
using System;
using System.Reflection;

namespace Timeline.Tests.Helpers
{
    public class ParameterInfoValueFormatter : IValueFormatter
    {
        public bool CanHandle(object value)
        {
            return value is ParameterInfo;
        }

        public string Format(object value, FormattingContext context, FormatChild formatChild)
        {
            var param = (ParameterInfo)value;
            return $"{param.Member.DeclaringType.FullName}.{param.Member.Name}#{param.Name}";
        }
    }

    public class ParameterInfoAssertions : ReferenceTypeAssertions<ParameterInfo, ParameterInfoAssertions>
    {
        static ParameterInfoAssertions()
        {
            Formatter.AddFormatter(new ParameterInfoValueFormatter());
        }

        public ParameterInfoAssertions(ParameterInfo parameterInfo)
        {
            Subject = parameterInfo;
        }

        protected override string Identifier => "parameter";

        public AndWhichConstraint<ParameterInfoAssertions, TAttribute> BeDecoratedWith<TAttribute>(string because = "", params object[] becauseArgs)
            where TAttribute : Attribute
        {
            var attribute = Subject.GetCustomAttribute<TAttribute>(false);

            Execute.Assertion
                .BecauseOf(because, becauseArgs)
                .ForCondition(attribute != null)
                .FailWith("Expected {0} {1} to be decorated with {2}{reason}, but that attribute was not found.",
                Identifier, Subject, typeof(TAttribute).FullName);

            return new AndWhichConstraint<ParameterInfoAssertions, TAttribute>(this, attribute);
        }
    }

    public static class ParameterInfoAssertionExtensions
    {
        public static ParameterInfoAssertions Should(this ParameterInfo parameterInfo)
        {
            return new ParameterInfoAssertions(parameterInfo);
        }
    }
}