/
IncludeMemberByPathSelectionRule.cs
41 lines (35 loc) · 1.38 KB
/
IncludeMemberByPathSelectionRule.cs
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
using System;
using System.Collections.Generic;
using System.Linq;
using FluentAssertions.Common;
namespace FluentAssertions.Equivalency.Selection
{
/// <summary>
/// Selection rule that includes a particular property in the structural comparison.
/// </summary>
internal class IncludeMemberByPathSelectionRule : SelectMemberByPathSelectionRule
{
private readonly MemberPath memberToInclude;
public IncludeMemberByPathSelectionRule(MemberPath pathToInclude)
: base(pathToInclude.ToString())
{
memberToInclude = pathToInclude;
}
public override bool IncludesMembers => true;
protected override IEnumerable<SelectedMemberInfo> OnSelectMembers(IEnumerable<SelectedMemberInfo> selectedMembers,
string currentPath, IMemberInfo context)
{
var matchingMembers =
from member in context.RuntimeType.GetNonPrivateMembers()
let memberPath = currentPath.Combine(member.Name)
where memberToInclude.IsSameAs(memberPath, member.DeclaringType) ||
memberToInclude.IsParentOrChildOf(memberPath)
select member;
return selectedMembers.Concat(matchingMembers).ToArray();
}
public override string ToString()
{
return "Include member root." + memberToInclude;
}
}
}