test(sql): cover rangeFrameEntirelyPastThePartitionEdgeAggregatesToNu...
This commit is contained in:
dualfroz committed
1 parent
613cb1ecef
commit
5d49c4c96d
1 file changed
+162
@@ -2534,3 +2534,165 @@ class DiskExecutorTest {
|
|||||||
assertEquals(listOf(2L, 2L, 2L, 2L, 1L), result.rows.map { (it[1] as Value.IntegerValue).value })
|
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<DiskExecutorException> {
|
||||||
|
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<DiskExecutorException> {
|
||||||
|
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<DiskExecutorException> {
|
||||||
|
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<DiskExecutorException> {
|
||||||
|
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<DiskExecutorException> {
|
||||||
|
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 })
|
||||||
|
}
|
||||||
|
|
||||||
Reference in new issue
Block a user