Skip to content

Commit

Permalink
add test for char(2)
Browse files Browse the repository at this point in the history
  • Loading branch information
xuyangzhong committed Dec 21, 2023
1 parent 0044ad4 commit 9b78b3e
Showing 1 changed file with 24 additions and 22 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1335,16 +1335,17 @@ class AggregateITCase(aggMode: AggMode, miniBatch: MiniBatchMode, backend: State
def testMinMaxWithChar(): Unit = {
val data =
List(
rowOf(1, "a"),
rowOf(1, "b"),
rowOf(2, "d"),
rowOf(2, "c")
rowOf(1, "a", "gg"),
rowOf(1, "b", "hh"),
rowOf(2, "d", "j"),
rowOf(2, "c", "i")
)
val dataId = TestValuesTableFactory.registerData(data)
tEnv.executeSql(s"""
|CREATE TABLE src(
| `id` INT,
| `char` CHAR(1)
| `char1` CHAR(1),
| `char2` CHAR(2)
|) WITH (
| 'connector' = 'values',
| 'data-id' = '$dataId'
Expand All @@ -1353,39 +1354,40 @@ class AggregateITCase(aggMode: AggMode, miniBatch: MiniBatchMode, backend: State

val sql =
"""
|select `id`, count(*), min(`char`), max(`char`) from src group by `id`
|select `id`, count(*), min(`char1`), max(`char1`), min(`char2`), max(`char2`) from src group by `id`
""".stripMargin

val sink = new TestingRetractSink()
tEnv.sqlQuery(sql).toRetractStream[Row].addSink(sink)
env.execute()

val expected = List("1,2,a,b", "2,2,c,d")
val expected = List("1,2,a,b,gg,hh", "2,2,c,d,i,j")
assertThat(sink.getRetractResults.sorted).isEqualTo(expected.sorted)
}

@TestTemplate
def testRetractMinMaxWithChar(): Unit = {
val data =
List(
changelogRow("+I", Int.box(1), "a"),
changelogRow("+I", Int.box(1), "b"),
changelogRow("+I", Int.box(1), "c"),
changelogRow("-D", Int.box(1), "c"),
changelogRow("-D", Int.box(1), "a"),
changelogRow("+I", Int.box(2), "a"),
changelogRow("+I", Int.box(2), "b"),
changelogRow("+I", Int.box(2), "c"),
changelogRow("-U", Int.box(2), "b"),
changelogRow("+U", Int.box(2), "d"),
changelogRow("-U", Int.box(2), "a"),
changelogRow("+U", Int.box(2), "b")
changelogRow("+I", Int.box(1), "a", "ee"),
changelogRow("+I", Int.box(1), "b", "ff"),
changelogRow("+I", Int.box(1), "c", "gg"),
changelogRow("-D", Int.box(1), "c", "gg"),
changelogRow("-D", Int.box(1), "a", "ee"),
changelogRow("+I", Int.box(2), "a", "e"),
changelogRow("+I", Int.box(2), "b", "f"),
changelogRow("+I", Int.box(2), "c", "g"),
changelogRow("-U", Int.box(2), "b", "f"),
changelogRow("+U", Int.box(2), "d", "h"),
changelogRow("-U", Int.box(2), "a", "e"),
changelogRow("+U", Int.box(2), "b", "f")
)
val dataId = TestValuesTableFactory.registerData(data)
tEnv.executeSql(s"""
|CREATE TABLE src(
| `id` INT,
| `char` CHAR(1)
| `char1` CHAR(1),
| `char2` CHAR(2)
|) WITH (
| 'connector' = 'values',
| 'data-id' = '$dataId',
Expand All @@ -1395,14 +1397,14 @@ class AggregateITCase(aggMode: AggMode, miniBatch: MiniBatchMode, backend: State

val sql =
"""
|select `id`, count(*), min(`char`), max(`char`) from src group by `id`
|select `id`, count(*), min(`char1`), max(`char1`), min(`char2`), max(`char2`) from src group by `id`
""".stripMargin

val sink = new TestingRetractSink()
tEnv.sqlQuery(sql).toRetractStream[Row].addSink(sink)
env.execute()

val expected = List("1,1,b,b", "2,3,b,d")
val expected = List("1,1,b,b,ff,ff", "2,3,b,d,f,h")
assertThat(sink.getRetractResults.sorted).isEqualTo(expected.sorted)
}

Expand Down

0 comments on commit 9b78b3e

Please sign in to comment.