Add example of complex join test using SqlExpression

Author: mythz

tests/ServiceStack.OrmLite.Tests/Expression/ComplexJoinTests.cs

@@ -0,0 +1,185 @@
+using System;
+using System.Data;
+using System.Linq;
+using NUnit.Framework;
+using ServiceStack.DataAnnotations;
+using ServiceStack.Model;
+using ServiceStack.Text;
+
+namespace ServiceStack.OrmLite.Tests.Expression
+{
+    public class Bar : IHasGuidId
+    {
+        [PrimaryKey]
+        public Guid Id { get; set; }
+
+        [Required]
+        public string Name { get; set; }
+    }
+
+    public class FooBar : IHasIntId
+    {
+        [AutoIncrement]
+        [PrimaryKey]
+        public int Id { get; set; }
+
+        // And a foreign key to the foo table as well, but that is not necessary to show the problem.
+
+        [Alias("fkBarId")]
+        [ForeignKey(typeof(Bar), ForeignKeyName = "fk_FooBar_Bar")]
+        public Guid BarId { get; set; }
+    }
+
+    public class Baz : IHasIntId
+    {
+        [PrimaryKey]
+        [AutoIncrement]
+        public int Id { get; set; }
+
+        [Required]
+        public string Name { get; set; }
+    }
+
+    public class FooBarBaz : IHasIntId
+    {
+        [AutoIncrement]
+        [PrimaryKey]
+        public int Id { get; set; }
+
+        [Alias("fkFooBarId")]
+        [ForeignKey(typeof(FooBar), ForeignKeyName = "fk_FooBarbaz_FooBar", OnDelete = "CASCADE")]
+        public int FooBarId { get; set; }
+
+        [Alias("fkBazId")]
+        [ForeignKey(typeof(Baz), ForeignKeyName = "fk_FooBarBaz_Baz", OnDelete = "CASCADE")]
+        public int BazId { get; set; }
+
+        [Required]
+        public decimal Amount { get; set; }
+    }
+
+    internal class JoinResult
+    {
+        [BelongTo(typeof(FooBar))]
+        public int Id { get; set; }
+
+        [Alias("fkBazId")]
+        [BelongTo(typeof(FooBarBaz))]
+        public int FooBarBazId { get; set; }
+
+        [BelongTo(typeof(FooBarBaz))]
+        public decimal Amount { get; set; }
+
+        [Alias("fkBarId")]
+        [BelongTo(typeof(Bar))]
+        public Guid BarId { get; set; }
+
+        [BelongTo(typeof(Bar))]
+        public string BarName { get; set; }
+
+        [Alias("fkBazId")]
+        [BelongTo(typeof(Baz))]
+        public int BazId { get; set; }
+
+        [BelongTo(typeof(Baz))]
+        public string BazName { get; set; }
+    }
+
+
+    [TestFixture]
+    public class ComplexJoinTests : OrmLiteTestBase
+    {
+        private static void InitTables(IDbConnection db)
+        {
+            db.DropTable<FooBarBaz>();
+            db.DropTable<FooBar>();
+            db.DropTable<Bar>();
+            db.DropTable<Baz>();
+
+            db.CreateTable<Baz>();
+            db.CreateTable<Bar>();
+            db.CreateTable<FooBar>();
+            db.CreateTable<FooBarBaz>();
+
+            var bar1Id = new Guid("5bd67b84-bfdb-4057-9799-5e7a72a6eaa9");
+            var bar2Id = new Guid("a8061d08-6816-4e1e-b3d7-1178abcefa0d");
+
+            db.Insert(new Bar { Id = bar1Id, Name = "Banana", });
+            db.Insert(new Bar { Id = bar2Id, Name = "Orange", });
+
+            db.Insert(new Baz { Id = 1, Name = "Large" });
+            db.Insert(new Baz { Id = 2, Name = "Huge" });
+
+            db.Insert(new FooBar { Id = 1, BarId = bar1Id, });
+            db.Insert(new FooBar { Id = 2, BarId = bar2Id, });
+
+            db.Insert(new FooBarBaz { Amount = 42, FooBarId = 1, BazId = 2 });
+            db.Insert(new FooBarBaz { Amount = 50, FooBarId = 1, BazId = 1 });
+            db.Insert(new FooBarBaz { Amount = 2, FooBarId = 2, BazId = 1 });
+        }
+
+        [Test]
+        public void ComplexJoin_with_JoinSqlBuilder()
+        {
+            using (var db = OpenDbConnection())
+            {
+                InitTables(db);
+
+                /* This gives the expected values for BazId */
+                var jn = new JoinSqlBuilder<JoinResult, FooBar>()
+                    .Join<FooBar, Bar>(
+                        sourceColumn: dp => dp.BarId,
+                        destinationColumn: p => p.Id,
+                        destinationTableColumnSelection: p => new { BarName = p.Name, BarId = p.Id })
+                    .Join<FooBar, FooBarBaz>(
+                        sourceColumn: dp => dp.Id,
+                        destinationColumn: dpss => dpss.FooBarId,
+                        destinationTableColumnSelection: dpss => new { dpss.Amount, FooBarBazId = dpss.Id });
+                jn.Join<FooBarBaz, Baz>(
+                    sourceColumn: dpss => dpss.BazId,
+                    destinationColumn: ss => ss.Id,
+                    destinationTableColumnSelection: ss => new { BazId = ss.Id, BazName = ss.Name });
+                jn.Select<FooBar>(dp => new { Id = dp.Id, });
+
+                var results = db.Select<JoinResult>(jn.ToSql());
+                db.GetLastSql().Print();
+
+                results.PrintDump();
+
+                var fooBarBaz = results.First(x => x.FooBarBazId == 1);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(2));
+                fooBarBaz = results.First(x => x.FooBarBazId == 2);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(1));
+                fooBarBaz = results.First(x => x.FooBarBazId == 2);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(1));
+            }
+        }
+
+        [Test]
+        public void ComplexJoin_with_SqlExpression()
+        {
+            using (var db = OpenDbConnection())
+            {
+                InitTables(db);
+
+                var q = db.From<FooBar>()
+                    .Join<Bar>((dp, p) => dp.BarId == p.Id)
+                    .Join<FooBarBaz>((dp, dpss) => dp.Id == dpss.FooBarId)
+                    .Join<FooBarBaz, Baz>((dpss, ss) => dpss.BazId == ss.Id);
+                
+                var results = db.Select<JoinResult>(q);
+
+                db.GetLastSql().Replace("INNER","\n INNER").Print();
+
+                results.PrintDump();
+
+                var fooBarBaz = results.First(x => x.FooBarBazId == 1);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(2));
+                fooBarBaz = results.First(x => x.FooBarBazId == 2);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(1));
+                fooBarBaz = results.First(x => x.FooBarBazId == 2);
+                Assert.That(fooBarBaz.BazId, Is.EqualTo(1));
+            }
+        }
+    }
+}

tests/ServiceStack.OrmLite.Tests/ServiceStack.OrmLite.Tests.csproj

@@ -117,6 +117,7 @@
     <Compile Include="CompositeKeyTests.cs" />
     <Compile Include="CustomSqlTests.cs" />
     <Compile Include="DateTimeOffsetTests.cs" />
+    <Compile Include="Expression\ComplexJoinTests.cs" />
     <Compile Include="Expression\ExpressionUsingCustomSerializedEnumTests.cs" />
     <Compile Include="Expression\FromExpressionTests.cs" />
     <Compile Include="Expression\GenericTableExpressions.cs" />