iss-507: add more tests

Author: ahmad-moussawi

QueryBuilder.Tests/MySqlExecutionTest.cs

@@ -200,6 +200,55 @@ public void ExistsShouldReturnTrueForNonEmptyTable()
             db.Drop("Transaction");
         }
 
+        [Fact]
+        public void BasicSelectFilter()
+        {
+            var db = DB().Create("Transaction", new[] {
+                    "Id INT PRIMARY KEY AUTO_INCREMENT",
+                    "Date DATE NOT NULL",
+                    "Amount int NOT NULL",
+            });
+
+            var data = new Dictionary<string, int> {
+                // 2020
+                {"2020-01-01", 10},
+                {"2020-05-01", 20},
+                
+                // 2021
+                {"2021-01-01", 40},
+                {"2021-02-01", 10},
+                {"2021-04-01", -10},
+                
+                // 2022
+                {"2022-01-01", 80},
+                {"2022-02-01", -30},
+                {"2022-05-01", 50},
+            };
+
+            foreach (var row in data)
+            {
+                db.Query("Transaction").Insert(new
+                {
+                    Date = row.Key,
+                    Amount = row.Value
+                });
+            }
+
+            var query = db.Query("Transaction")
+                .SelectSum("Amount as Total_2020", q => q.WhereDatePart("year", "date", 2020))
+                .SelectSum("Amount as Total_2021", q => q.WhereDatePart("year", "date", 2021))
+                .SelectSum("Amount as Total_2022", q => q.WhereDatePart("year", "date", 2022))
+                ;
+
+            var results = query.Get().ToList();
+            Assert.Single(results);
+            Assert.Equal(30, results[0].Total_2020);
+            Assert.Equal(40, results[0].Total_2021);
+            Assert.Equal(100, results[0].Total_2022);
+
+            db.Drop("Transaction");
+        }
+
         QueryFactory DB()
         {
             var host = System.Environment.GetEnvironmentVariable("SQLKATA_MYSQL_HOST");

QueryBuilder.Tests/SQLiteExecutionTest.cs

@@ -207,6 +207,57 @@ public void InlineTable()
             db.Drop("Transaction");
         }
 
+
+
+        [Fact]
+        public void BasicSelectFilter()
+        {
+            var db = DB().Create("Transaction", new[] {
+                    "Id INTEGER PRIMARY KEY AUTOINCREMENT",
+                    "Date DATE NOT NULL",
+                    "Amount int NOT NULL",
+            });
+
+            var data = new Dictionary<string, int> {
+                // 2020
+                {"2020-01-01", 10},
+                {"2020-05-01", 20},
+                
+                // 2021
+                {"2021-01-01", 40},
+                {"2021-02-01", 10},
+                {"2021-04-01", -10},
+                
+                // 2022
+                {"2022-01-01", 80},
+                {"2022-02-01", -30},
+                {"2022-05-01", 50},
+            };
+
+            foreach (var row in data)
+            {
+                db.Query("Transaction").Insert(new
+                {
+                    Date = row.Key,
+                    Amount = row.Value
+                });
+            }
+
+            var query = db.Query("Transaction")
+                .SelectSum("Amount as Total_2020", q => q.WhereDatePart("year", "date", 2020))
+                .SelectSum("Amount as Total_2021", q => q.WhereDatePart("year", "date", 2021))
+                .SelectSum("Amount as Total_2022", q => q.WhereDatePart("year", "date", 2022))
+                ;
+
+            var results = query.Get().ToList();
+            Assert.Single(results);
+            Assert.Equal(30, results[0].Total_2020);
+            Assert.Equal(40, results[0].Total_2021);
+            Assert.Equal(100, results[0].Total_2022);
+
+            db.Drop("Transaction");
+        }
+
         QueryFactory DB()
         {
             var cs = $"Data Source=file::memory:;Cache=Shared";

QueryBuilder.Tests/SelectTests.cs

@@ -819,7 +819,7 @@ public void EscapeClauseThrowsForMultipleCharacters()
         [Fact]
         public void BasicSelectRaw_WithNoTable()
         {
-            var q = new Query().SelectRaw("somefunction() as c1");                
+            var q = new Query().SelectRaw("somefunction() as c1");
 
             var c = Compilers.CompileFor(EngineCodes.SqlServer, q);
             Assert.Equal("SELECT somefunction() as c1", c.ToString());
@@ -848,6 +848,60 @@ public void BasicSelect_WithNoTableWhereRawClause()
             var c = Compilers.CompileFor(EngineCodes.SqlServer, q);
             Assert.Equal("SELECT [c1] WHERE 1 = 1", c.ToString());
         }
-        
+
+        [Fact]
+        public void BasicSelectAggregate()
+        {
+            var q = new Query("Posts").Select("Title")
+                .SelectAggregate("sum", "ViewCount");
+
+            var sqlServer = Compilers.CompileFor(EngineCodes.SqlServer, q);
+            Assert.Equal("SELECT [Title], SUM([ViewCount]) FROM [Posts]", sqlServer.ToString());
+        }
+
+        [Fact]
+        public void SelectAggregateShouldIgnoreEmptyFilter()
+        {
+            var q = new Query("Posts").Select("Title")
+                .SelectAggregate("sum", "ViewCount", q => q);
+
+            var sqlServer = Compilers.CompileFor(EngineCodes.SqlServer, q);
+            Assert.Equal("SELECT [Title], SUM([ViewCount]) FROM [Posts]", sqlServer.ToString());
+        }
+
+        [Fact]
+        public void SelectAggregateShouldIgnoreEmptyQueryFilter()
+        {
+            var q = new Query("Posts").Select("Title")
+                .SelectAggregate("sum", "ViewCount", new Query());
+
+            var sqlServer = Compilers.CompileFor(EngineCodes.SqlServer, q);
+            Assert.Equal("SELECT [Title], SUM([ViewCount]) FROM [Posts]", sqlServer.ToString());
+        }
+
+        [Fact]
+        public void BasicSelectAggregateWithAlias()
+        {
+            var q = new Query("Posts").Select("Title")
+                .SelectAggregate("sum", "ViewCount as TotalViews");
+
+            var sqlServer = Compilers.CompileFor(EngineCodes.SqlServer, q);
+            Assert.Equal("SELECT [Title], SUM([ViewCount]) AS [TotalViews] FROM [Posts]", sqlServer.ToString());
+        }
+
+        [Fact]
+        public void SelectWithFilter()
+        {
+            var q = new Query("Posts").Select("Title")
+                .SelectAggregate("sum", "ViewCount as Published_Jan", q => q.Where("Published_Month", "Jan"))
+                .SelectAggregate("sum", "ViewCount as Published_Feb", q => q.Where("Published_Month", "Feb"));
+
+            var pgSql = Compilers.CompileFor(EngineCodes.PostgreSql, q);
+            Assert.Equal("SELECT \"Title\", SUM(\"ViewCount\") FILTER (WHERE \"Published_Month\" = 'Jan') AS \"Published_Jan\", SUM(\"ViewCount\") FILTER (WHERE \"Published_Month\" = 'Feb') AS \"Published_Feb\" FROM \"Posts\"", pgSql.ToString());
+
+            var sqlServer = Compilers.CompileFor(EngineCodes.SqlServer, q);
+            Assert.Equal("SELECT [Title], SUM(CASE WHEN [Published_Month] = 'Jan' THEN [ViewCount] END) AS [Published_Jan], SUM(CASE WHEN [Published_Month] = 'Feb' THEN [ViewCount] END) AS [Published_Feb] FROM [Posts]", sqlServer.ToString());
+        }
+
     }
 }