Unit Test for FetchMany and ThenFetchMany in Stateless Session

Signed-off-by: CSharper2010 <[email protected]>

Author: csharper2010

src/NHibernate.Test/NHibernate.Test.csproj

@@ -1233,13 +1233,15 @@
     <Compile Include="RecordingInterceptor.cs" />
     <Compile Include="Stateless\Contact.cs" />
     <Compile Include="Stateless\Country.cs" />
+    <Compile Include="Stateless\FetchingLazyCollections\TreeFetchTests.cs" />
     <Compile Include="Stateless\FetchingLazyCollections\LazyCollectionFetchTests.cs" />
     <Compile Include="Stateless\Fetching\Resource.cs" />
     <Compile Include="Stateless\Fetching\StatelessSessionFetchingTest.cs" />
     <Compile Include="Stateless\Fetching\Task.cs" />
     <Compile Include="Stateless\Fetching\User.cs" />
     <Compile Include="Stateless\Org.cs" />
     <Compile Include="Stateless\StatelessSessionQueryFixture.cs" />
+    <Compile Include="Stateless\TreeNode.cs" />
     <Compile Include="Subselect\ClassSubselectFixture.cs" />
     <Compile Include="Subselect\Domain.cs" />
     <Compile Include="TestCaseMappingByCode.cs" />

src/NHibernate.Test/Stateless/FetchingLazyCollections/TreeFetchTests.cs

@@ -0,0 +1,70 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NHibernate.Cfg.MappingSchema;
+using NHibernate.Mapping.ByCode;
+using NHibernate.Mapping.ByCode.Conformist;
+using NHibernate.Linq;
+using NUnit.Framework;
+using SharpTestsEx;
+
+namespace NHibernate.Test.Stateless.FetchingLazyCollections
+{
+    public class TreeFetchTests : TestCaseMappingByCode
+    {
+        protected override HbmMapping GetMappings()
+        {
+            var mapper = new ModelMapper();
+            mapper.BeforeMapClass += (mi, t, cm) => cm.Id(im => im.Generator(Generators.HighLow));
+            mapper.Class<TreeNode>(
+                mc =>
+                {
+                    mc.Id(x => x.Id);
+                    mc.Property(x => x.Content);
+                    mc.Set(x => x.Children, cam =>
+                    {
+                        cam.Key(km => km.Column("parentId"));
+                        cam.Cascade(Mapping.ByCode.Cascade.All);
+                    }, rel => rel.OneToMany());
+                });
+            var mappings = mapper.CompileMappingForAllExplicitlyAddedEntities();
+            return mappings;
+        }
+
+        [Test]
+        public void FetchMultipleHierarchies()
+        {
+            using (ISession s = sessions.OpenSession())
+            using (ITransaction tx = s.BeginTransaction())
+            {
+                var root = new TreeNode { Content = "Root" };
+                var child1 = new TreeNode { Content = "Child1" };
+                root.Children.Add(child1);
+                root.Children.Add(new TreeNode { Content = "Child2" });
+                child1.Children.Add(new TreeNode { Content = "Child1Child1" });
+                child1.Children.Add(new TreeNode { Content = "Child1Child2" });
+                s.Save(root);
+                tx.Commit();
+            }
+
+            using (IStatelessSession s = sessions.OpenStatelessSession())
+            using (ITransaction tx = s.BeginTransaction())
+            {
+                IList<TreeNode> rootNodes = s.Query<TreeNode>().Where(t => t.Content == "Root")
+                    .FetchMany(f => f.Children)
+                    .ThenFetchMany(f => f.Children).ToList();
+                Assert.That(rootNodes.Count, Is.EqualTo(1));
+                Assert.That(rootNodes.First().Children.Count, Is.EqualTo(2));
+
+                tx.Commit();
+            }
+
+            using (ISession s = sessions.OpenSession())
+            using (ITransaction tx = s.BeginTransaction())
+            {
+                s.Delete("from TreeNode");
+                tx.Commit();
+            }
+        }
+    }
+}

src/NHibernate.Test/Stateless/TreeNode.cs

@@ -0,0 +1,20 @@
+using System.Collections.Generic;
+
+
+namespace NHibernate.Test.Stateless
+{
+    public class TreeNode
+    {
+        private ISet<TreeNode> children = new HashSet<TreeNode>();
+
+        public virtual int Id { get; protected set; }
+
+        public virtual string Content { get; set; }
+
+        public virtual ISet<TreeNode> Children
+        {
+            get { return children; }
+            protected set { children = value; }
+        }
+    }
+}