test: add tests for replacement materialized views

Author: teskje

test/race-condition/mzcompose.py

@@ -665,6 +665,40 @@ def verify(self) -> str:
         raise NotImplementedError
 
 
+class ReplacementMaterializedView(Object):
+    def create(self) -> str:
+        select = (
+            "* FROM " + self.references.name
+            if self.references
+            else "'foo' AS a, 'bar' AS b"
+        )
+        return f"> CREATE MATERIALIZED VIEW {self.name} AS SELECT {select}"
+
+    def destroy(self) -> str:
+        return f"> DROP MATERIALIZED VIEW {self.name} CASCADE"
+
+    def manipulate(self, kind: int) -> str:
+        select = (
+            "* FROM " + self.references.name
+            if self.references
+            else "'foo' AS a, 'bar' AS b"
+        )
+        manipulations = [
+            lambda: "",
+            lambda: dedent(
+                f"""
+                > DROP MATERIALIZED VIEW IF EXISTS {self.name}_replacement
+                > CREATE MATERIALIZED VIEW {self.name}_replacement REPLACING {self.name} AS SELECT {select}
+                > ALTER MATERIALIZED VIEW {self.name} APPLY REPLACEMENT {self.name}_replacement
+                """
+            ),
+        ]
+        return manipulations[kind % len(manipulations)]()
+
+    def verify(self) -> str:
+        raise NotImplementedError
+
+
 class DefaultIndex(Object):
     can_refer: bool = False
 

test/sqllogictest/replacement-materialized-views.slt

@@ -0,0 +1,149 @@
+# Copyright Materialize, Inc. and contributors. All rights reserved.
+#
+# Use of this software is governed by the Business Source License
+# included in the LICENSE file at the root of this repository.
+#
+# As of the Change Date specified in that file, in accordance with
+# the Business Source License, use of this software will be governed
+# by the Apache License, Version 2.0.
+
+mode cockroach
+
+# Setup
+
+statement ok
+CREATE TABLE t (a int, b int)
+
+statement ok
+INSERT INTO t VALUES (1, 2), (3, 4), (5, 6)
+
+statement ok
+CREATE CLUSTER other REPLICAS (r1 (SIZE 'scale=1,workers=1'), r2 (SIZE 'scale=2,workers=2'))
+
+
+# Test: basic replacement workflow
+
+statement ok
+CREATE MATERIALIZED VIEW mv AS SELECT a, b FROM t
+
+query II
+SELECT * FROM mv
+----
+1 2
+3 4
+5 6
+
+statement ok
+CREATE MATERIALIZED VIEW rp REPLACING mv AS SELECT a + b as a, b FROM t
+
+query II
+SELECT * FROM mv
+----
+1 2
+3 4
+5 6
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster    comment
+mv   quickstart (empty)
+rp   quickstart (empty)
+
+statement ok
+ALTER MATERIALIZED VIEW mv APPLY REPLACEMENT rp
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster    comment
+mv   quickstart (empty)
+
+query II
+SELECT * FROM mv
+----
+3  2
+7  4
+11 6
+
+
+# Test: aborted replacement workflow
+
+statement ok
+CREATE MATERIALIZED VIEW rp REPLACING mv AS SELECT -a as a, b FROM t
+
+query II
+SELECT * FROM mv
+----
+3  2
+7  4
+11 6
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster    comment
+mv   quickstart (empty)
+rp   quickstart (empty)
+
+statement ok
+DROP MATERIALIZED VIEW rp
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster    comment
+mv   quickstart (empty)
+
+query II
+SELECT * FROM mv
+----
+3  2
+7  4
+11 6
+
+
+# Test: schemas must match
+
+statement error db error: ERROR: incompatible schemas
+CREATE MATERIALIZED VIEW rp REPLACING mv AS SELECT a, b, 1 FROM t
+
+# Test: replacement can be created in another cluster
+
+statement ok
+CREATE MATERIALIZED VIEW rp REPLACING mv IN CLUSTER other AS SELECT a, b FROM t;
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster    comment
+mv   quickstart (empty)
+rp   other      (empty)
+
+statement ok
+ALTER MATERIALIZED VIEW mv APPLY REPLACEMENT rp
+
+query TTT colnames,rowsort
+SHOW MATERIALIZED VIEWS
+----
+name cluster comment
+mv   other   (empty)
+
+# Test: replacement depends on target materialized view
+
+statement ok
+CREATE MATERIALIZED VIEW rp REPLACING mv AS SELECT a, b FROM t
+
+statement error db error: ERROR: cannot drop materialized view "mv": still depended upon by materialized view "rp"
+DROP MATERIALIZED VIEW mv;
+
+statement ok
+DROP MATERIALIZED VIEW mv CASCADE;
+
+# TODO: Test: replacement cannot be selected from
+# TODO: Test: not more than one replacement allowed per MV
+# TODO: Test: cannot create an object depending on a replacement
+# TODO: Test: replacing with the same SQL
+# TODO: Test: audit log
+# TODO: Test: error on wrong replacement type in ALTER MV
+# TODO: Test: error on wrong target type in CREATE MV