test: fix nightly tests broken by MaterializeInc#14467

wangandi · Aug 31, 2022 · eceee6f · eceee6f
1 parent b11a185
commit eceee6f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/misc/python/materialize/checks/sink.py b/misc/python/materialize/checks/sink.py
@@ -89,6 +89,8 @@ def manipulate(self) -> List[Testdrive]:
 
                 > CREATE CONNECTION IF NOT EXISTS kafka_conn FOR KAFKA BROKER '${testdrive.kafka-addr}';
 
+                > CREATE CONNECTION IF NOT EXISTS csr_conn FOR CONFLUENT SCHEMA REGISTRY URL '${testdrive.schema-registry-url}';
+
                 > CREATE SINK sink_sink2 FROM sink_source_view
                   INTO KAFKA CONNECTION kafka_conn
                   TOPIC 'sink-sink2'
@@ -122,6 +124,8 @@ def validate(self) -> Testdrive:
 
                 > CREATE CONNECTION IF NOT EXISTS kafka_conn FOR KAFKA BROKER '${testdrive.kafka-addr}';
 
+                > CREATE CONNECTION IF NOT EXISTS csr_conn FOR CONFLUENT SCHEMA REGISTRY URL '${testdrive.schema-registry-url}';
+
                 > CREATE SOURCE sink_view1
                   FROM KAFKA CONNECTION kafka_conn
                   TOPIC 'sink-sink1'
@@ -204,10 +208,12 @@ def initialize(self) -> Testdrive:
 
                 > CREATE CONNECTION IF NOT EXISTS kafka_conn FOR KAFKA BROKER '${testdrive.kafka-addr}';
 
+                > CREATE CONNECTION IF NOT EXISTS csr_conn FOR CONFLUENT SCHEMA REGISTRY URL '${testdrive.schema-registry-url}';
+
                 > CREATE SINK sink_large_transaction_sink1 FROM sink_large_transaction_view
                   INTO KAFKA CONNECTION kafka_conn
                   TOPIC 'testdrive-sink-large-transaction-sink-${testdrive.seed}'
-                  FORMAT AVRO USING CONFLUENT SCHEMA REGISTRY CONNECTION csr_conn
+                  FORMAT AVRO USING CONFLUENT SCHEMA REGISTRY CONNECTION csr_conn;
                 """
             )
         )
@@ -232,6 +238,8 @@ def validate(self) -> Testdrive:
                 # We check the contents of the sink topics by re-ingesting them.
                 > CREATE CONNECTION IF NOT EXISTS kafka_conn FOR KAFKA BROKER '${testdrive.kafka-addr}';
 
+                > CREATE CONNECTION IF NOT EXISTS csr_conn FOR CONFLUENT SCHEMA REGISTRY URL '${testdrive.schema-registry-url}';
+
                 > CREATE SOURCE sink_large_transaction_source
                   FROM KAFKA CONNECTION kafka_conn
                   TOPIC 'testdrive-sink-large-transaction-sink-${testdrive.seed}'

diff --git a/test/limits/mzcompose.py b/test/limits/mzcompose.py
@@ -189,7 +189,7 @@ def body(cls) -> None:
         print('"123" {"f1": "123"}')
 
         print(
-            """> CREATE CONNECTION IF NOT EXISTS csr_conn
+            f"""> CREATE CONNECTION IF NOT EXISTS csr_conn
             FOR CONFLUENT SCHEMA REGISTRY
             URL '${{testdrive.schema-registry-url}}';
             """