summaryrefslogtreecommitdiffstats
path: root/tests/dialects/test_risingwave.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/dialects/test_risingwave.py')
-rw-r--r--tests/dialects/test_risingwave.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/dialects/test_risingwave.py b/tests/dialects/test_risingwave.py
index 7d6d50c..80c5265 100644
--- a/tests/dialects/test_risingwave.py
+++ b/tests/dialects/test_risingwave.py
@@ -12,3 +12,12 @@ class TestRisingWave(Validator):
"": "SELECT a FROM tbl FOR UPDATE",
},
)
+ self.validate_identity(
+ "CREATE SOURCE from_kafka (*, gen_i32_field INT AS int32_field + 2, gen_i64_field INT AS int64_field + 2, WATERMARK FOR time_col AS time_col - INTERVAL '5 SECOND') INCLUDE header foo VARCHAR AS myheader INCLUDE key AS mykey WITH (connector='kafka', topic='my_topic') FORMAT PLAIN ENCODE PROTOBUF (A=1, B=2) KEY ENCODE PROTOBUF (A=3, B=4)"
+ )
+ self.validate_identity(
+ "CREATE SINK my_sink AS SELECT * FROM A WITH (connector='kafka', topic='my_topic') FORMAT PLAIN ENCODE PROTOBUF (A=1, B=2) KEY ENCODE PROTOBUF (A=3, B=4)"
+ )
+ self.validate_identity(
+ "WITH t1 AS MATERIALIZED (SELECT 1), t2 AS NOT MATERIALIZED (SELECT 2) SELECT * FROM t1, t2"
+ )