diff --git a/src/test/kotlin/kestrel/store/sql/DiskExecutorTest.kt b/src/test/kotlin/kestrel/store/sql/DiskExecutorTest.kt index ab53f08..233c173 100644 --- a/src/test/kotlin/kestrel/store/sql/DiskExecutorTest.kt +++ b/src/test/kotlin/kestrel/store/sql/DiskExecutorTest.kt @@ -2534,3 +2534,165 @@ class DiskExecutorTest { assertEquals(listOf(2L, 2L, 2L, 2L, 1L), result.rows.map { (it[1] as Value.IntegerValue).value }) } + @Test + fun rangeFrameEntirelyPastThePartitionEdgeAggregatesToNullAndCountsZero() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount RANGE BETWEEN 100 FOLLOWING AND 200 FOLLOWING) AS s, " + + "COUNT(*) OVER (ORDER BY amount RANGE BETWEEN 100 FOLLOWING AND 200 FOLLOWING) AS c " + + "FROM sales ORDER BY id ASC" + ) + assertEquals((1L..5L).map { listOf(int(it), Value.NullValue, int(0)) }, result.rows) + } + + @Test + fun rangeFrameWithAPrecedingBoundRequiresExactlyOneOrderByColumn() { + val db = newDatabase() + db.seedWindowSales() + val error = assertFailsWith { + db.run("SELECT SUM(amount) OVER (ORDER BY amount, id RANGE BETWEEN 10 PRECEDING AND 10 FOLLOWING) FROM sales") + } + assertTrue(error.message!!.contains("exactly one ORDER BY column"), error.message!!) + } + + @Test + fun rangeFrameWithAPrecedingBoundRequiresANumericOrDateOrderColumn() { + val db = newDatabase() + db.seedWindowEmployees() + val error = assertFailsWith { + db.run("SELECT SUM(salary) OVER (ORDER BY name RANGE BETWEEN 1 PRECEDING AND CURRENT ROW) FROM employees") + } + assertTrue(error.message!!.contains("numeric or DATE"), error.message!!) + } + + @Test + fun rangeFrameOffsetMustBeNumeric() { + val db = newDatabase() + db.seedWindowSales() + val error = assertFailsWith { + db.run("SELECT SUM(amount) OVER (ORDER BY amount RANGE BETWEEN 'x' PRECEDING AND CURRENT ROW) FROM sales") + } + assertTrue(error.message!!.contains("numeric"), error.message!!) + } + + @Test + fun rangeFrameOffsetMustNotBeNegative() { + val db = newDatabase() + db.seedWindowSales() + val error = assertFailsWith { + db.run("SELECT SUM(amount) OVER (ORDER BY amount RANGE BETWEEN -1 PRECEDING AND CURRENT ROW) FROM sales") + } + assertTrue(error.message!!.contains("negative"), error.message!!) + } + + @Test + fun groupsBetweenPrecedingAndFollowingCountsPeerGroupsNotPhysicalRows() { + val db = newDatabase() + db.seedWindowSales() + val groups = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount GROUPS BETWEEN 1 PRECEDING AND 1 FOLLOWING) AS s FROM sales ORDER BY id ASC" + ) + val rows = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount ROWS BETWEEN 1 PRECEDING AND 1 FOLLOWING) AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(50L, 80L, 80L, 120L, 80L), groups.rows.map { (it[1] as Value.IntegerValue).value }) + assertEquals(listOf(30L, 50L, 70L, 100L, 80L), rows.rows.map { (it[1] as Value.IntegerValue).value }) + } + + @Test + fun groupsCurrentRowExpandsToTheWholePeerGroupJustLikeRangeCurrentRow() { + val db = newDatabase() + db.seedWindowSales() + val groups = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount GROUPS BETWEEN UNBOUNDED PRECEDING AND CURRENT ROW) AS s FROM sales ORDER BY id ASC" + ) + val range = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount RANGE BETWEEN UNBOUNDED PRECEDING AND CURRENT ROW) AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(10L, 50L, 50L, 80L, 130L), groups.rows.map { (it[1] as Value.IntegerValue).value }) + assertEquals(groups.rows, range.rows) + } + + @Test + fun groupsFrameWorksWithAMultiColumnNonNumericOrderByUnlikeRange() { + val db = newDatabase() + db.seedWindowEmployees() + val result = db.select( + "SELECT id, COUNT(*) OVER (ORDER BY deptId, name GROUPS BETWEEN 1 PRECEDING AND 1 FOLLOWING) AS c " + + "FROM employees ORDER BY deptId ASC, name ASC" + ) + assertEquals(listOf(int(1), int(3), int(2), int(5), int(6), int(4)), result.rows.map { it[0] }) + assertEquals(listOf(2L, 3L, 3L, 3L, 3L, 2L), result.rows.map { (it[1] as Value.IntegerValue).value }) + } + + @Test + fun groupsFrameEntirelyPastThePartitionEdgeAggregatesToNullAndCountsZero() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (ORDER BY amount GROUPS BETWEEN 100 FOLLOWING AND 200 FOLLOWING) AS s, " + + "COUNT(*) OVER (ORDER BY amount GROUPS BETWEEN 100 FOLLOWING AND 200 FOLLOWING) AS c " + + "FROM sales ORDER BY id ASC" + ) + assertEquals((1L..5L).map { listOf(int(it), Value.NullValue, int(0)) }, result.rows) + } + + @Test + fun groupsFrameIsRejectedOnRankingAndOffsetWindowFunctions() { + val db = newDatabase() + db.seedWindowEmployees() + val error = assertFailsWith { + db.run("SELECT ROW_NUMBER() OVER (ORDER BY id GROUPS BETWEEN 1 PRECEDING AND CURRENT ROW) FROM employees") + } + assertTrue(error.message!!.contains("GROUPS"), error.message!!) + } + + @Test + fun excludeCurrentRowRemovesOnlyTheCurrentRowFromTheWholePartitionFrame() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (" + + "ORDER BY amount ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE CURRENT ROW" + + ") AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(120L, 110L, 110L, 100L, 80L), result.rows.map { (it[1] as Value.IntegerValue).value }) + } + + @Test + fun excludeGroupRemovesTheWholePeerGroupIncludingTheCurrentRow() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (" + + "ORDER BY amount RANGE BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE GROUP" + + ") AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(120L, 90L, 90L, 100L, 80L), result.rows.map { (it[1] as Value.IntegerValue).value }) + } + + @Test + fun excludeTiesRemovesOnlyTheOtherTiedRowsAndKeepsTheCurrentRow() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (" + + "ORDER BY amount RANGE BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING EXCLUDE TIES" + + ") AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(130L, 110L, 110L, 130L, 130L), result.rows.map { (it[1] as Value.IntegerValue).value }) + } + + @Test + fun excludeCurrentRowInteractsWithABoundedFrameInsteadOfOnlyTheWholePartition() { + val db = newDatabase() + db.seedWindowSales() + val result = db.select( + "SELECT id, SUM(amount) OVER (" + + "ORDER BY amount ROWS BETWEEN 1 PRECEDING AND 1 FOLLOWING EXCLUDE CURRENT ROW" + + ") AS s FROM sales ORDER BY id ASC" + ) + assertEquals(listOf(20L, 30L, 50L, 70L, 30L), result.rows.map { (it[1] as Value.IntegerValue).value }) + } +