zjffdu commented on a change in pull request #7121: [FLINK-9555]Support table
api in scala shell
URL: https://github.com/apache/flink/pull/7121#discussion_r237736038
##########
File path:
flink-scala-shell/src/test/scala/org/apache/flink/api/scala/ScalaShellITCase.scala
##########
@@ -168,6 +168,56 @@ class ScalaShellITCase extends TestLogger {
Assert.assertTrue(output.contains("WC(world,10)"))
}
+ @Test
+ def testSimpleSelectWithFilterBatchTableAPIQuery: Unit = {
+ val input =
+ """
+ |import _root_.scala.collection.mutable
+ |val data = new mutable.MutableList[(Int, Long, String)]
+ | data.+=((1, 1L, "Hi"))
+ | data.+=((2, 2L, "Hello"))
+ | data.+=((3, 2L, "Hello world"))
+ |val t = benv.fromCollection(data).toTable(btenv, 'a, 'b,
'c).select("*").where('a % 2 === 1 )
+ |val results = t.toDataSet[Row].collect()
+ |results.foreach(println)
+ |:q
+ """.stripMargin
+ val output = processInShell(input)
+ Assert.assertFalse(output.contains("failed"))
+ Assert.assertFalse(output.contains("error"))
+ Assert.assertFalse(output.contains("Exception"))
+ }
+
+ @Test
+ def testGroupedAggregationStreamTableAPIQuery: Unit = {
+ val input =
+ """
+ | val data = List(
+ | ("Hello", 1),
+ | ("word", 1),
+ | ("Hello", 1),
+ | ("bark", 1),
+ | ("bark", 1),
+ | ("bark", 1),
+ | ("bark", 1),
+ | ("bark", 1),
+ | ("bark", 1),
+ | ("flink", 1)
+ | )
+ | val stream = senv.fromCollection(data)
+ | val table = stream.toTable(stenv, 'word, 'num)
+ | val resultTable = table.groupBy('word).select('num.sum as
'count).groupBy('count).select(
+ | 'count,'count.count as 'frequency)
+ | val results = resultTable.toRetractStream[Row]
+ | results.print
+ | senv.execute
+ """.stripMargin
+ val output = processInShell(input)
+ Assert.assertFalse(output.contains("failed"))
+ Assert.assertFalse(output.contains("error"))
+ Assert.assertFalse(output.contains("Exception"))
Review comment:
We also need to verify the output contains the correct word count result.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
