Related entity not found with RetrieveRequest #271

using FakeXrmEasy.Abstractions;
using FakeXrmEasy.Abstractions.Enums;
using FakeXrmEasy.Abstractions.Integrity;
using FakeXrmEasy.Integrity;
using FakeXrmEasy.Middleware;
using FakeXrmEasy.Middleware.Crud;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Microsoft.Xrm.Sdk;
using Microsoft.Xrm.Sdk.Messages;
using Microsoft.Xrm.Sdk.Query;
using StrategicIT.CRM.Tools.Entities;
using System;

namespace Plugins.TestsFakeXrmEasy2.Tests
{
    [TestClass]
    public class TestRelatedAccounts
    {
        [TestMethod]
        public void Test()
        {
            // Arrange
            var context = MiddlewareBuilder
                .New()
                .SetLicense(FakeXrmEasyLicense.Commercial)
                .AddCrud()
                .UseCrud()
                .Build();
            context.EnableProxyTypes(typeof(Account).Assembly);
            context.SetProperty<IIntegrityOptions>(new IntegrityOptions() { ValidateEntityReferences = true });
            context.AddRelationship(Account.Fields.Referencingaccount_parent_account, new XrmFakedRelationship()
            {
                RelationshipType = XrmFakedRelationship.FakeRelationshipType.OneToMany,
                Entity1LogicalName = Account.EntityLogicalName,
                Entity1Attribute = Account.Fields.ParentAccountId,
                Entity2LogicalName = Account.EntityLogicalName,
                Entity2Attribute = Account.Fields.Id,
            });

            var service = context.GetOrganizationService();
            var parentAccountId = service.Create(new Account
            {
                Name = "Parent Account"
            });
            var subAccountId = service.Create(new Account
            {
                Name = "Sub Account",
                ParentAccountId = new EntityReference(Account.EntityLogicalName, parentAccountId)
            });
            Console.WriteLine($"Created parent account: {parentAccountId}");
            Console.WriteLine($"Created sub account: {subAccountId}");

            // Act
            var retrieveRequest = new RetrieveRequest
            {
                Target = new EntityReference(Account.EntityLogicalName, subAccountId),
                ColumnSet = new ColumnSet(true),
                RelatedEntitiesQuery = new RelationshipQueryCollection
                {
                    {
                        new Relationship()
                        {
                            SchemaName = Account.Fields.Referencingaccount_parent_account,
                            PrimaryEntityRole = EntityRole.Referencing
                        },
                        new QueryExpression()
                        {
                            EntityName = Account.EntityLogicalName,
                            ColumnSet = new ColumnSet(true),
                        }
                    }
                }
            };
            var retrieveResponse = (RetrieveResponse)service.Execute(retrieveRequest);
            var retrievedAccount = retrieveResponse.Entity.ToEntity<Account>();

            // Assert
            var retrievedAccountId = retrievedAccount.Id;
            Console.WriteLine($"Retrieved sub account: {retrievedAccountId}");
            var retrievedParentAccountId = retrievedAccount.Referencingaccount_parent_account?.Id;
            Console.WriteLine($"Retrieved parent account: {retrievedParentAccountId}");

            Assert.AreEqual(subAccountId, retrievedAccountId);
            Assert.AreEqual(parentAccountId, retrievedParentAccountId);
        }
    }
}