Add db replace partition test (#174)

regression-test/suites/db-sync-replace-partition/test_db_sync_replace_partition.groovy

@@ -0,0 +1,155 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+suite("test_db_sync_replace_partition") {
+    def helper = new GroovyShell(new Binding(['suite': delegate]))
+            .evaluate(new File("${context.config.suitePath}/../common", "helper.groovy"))
+
+    def baseTableName = "test_replace_partition_" + UUID.randomUUID().toString().replace("-", "")
+    def test_num = 0
+    def insert_num = 5
+    def opPartitonName = "less0"
+
+    def exist = { res -> Boolean
+        return res.size() != 0
+    }
+    def notExist = { res -> Boolean
+        return res.size() == 0
+    }
+
+    logger.info("=== Create table ===")
+    tableName = "${baseTableName}"
+    sql """
+        CREATE TABLE if NOT EXISTS ${tableName}
+        (
+            `test` INT,
+            `id` INT
+        )
+        ENGINE=OLAP
+        UNIQUE KEY(`test`, `id`)
+        PARTITION BY RANGE(`id`)
+        (
+            PARTITION `p1` VALUES LESS THAN ("0"),
+            PARTITION `p2` VALUES LESS THAN ("100"),
+            PARTITION `p3` VALUES LESS THAN ("200"),
+            PARTITION `p4` VALUES LESS THAN ("300")
+        )
+        DISTRIBUTED BY HASH(id) BUCKETS AUTO
+        PROPERTIES (
+            "replication_allocation" = "tag.location.default: 1",
+            "binlog.enable" = "true"
+        )
+    """
+    sql """
+        CREATE TABLE if NOT EXISTS ${tableName}_1
+        (
+            `test` INT,
+            `id` INT
+        )
+        ENGINE=OLAP
+        UNIQUE KEY(`test`, `id`)
+        PARTITION BY RANGE(`id`)
+        (
+            PARTITION `p1` VALUES LESS THAN ("0"),
+            PARTITION `p2` VALUES LESS THAN ("100"),
+            PARTITION `p3` VALUES LESS THAN ("200"),
+            PARTITION `p4` VALUES LESS THAN ("300")
+        )
+        DISTRIBUTED BY HASH(id) BUCKETS AUTO
+        PROPERTIES (
+            "replication_allocation" = "tag.location.default: 1",
+            "binlog.enable" = "true"
+        )
+    """
+
+    helper.enableDbBinlog()
+    helper.ccrJobDelete()
+    helper.ccrJobCreate()
+
+    assertTrue(helper.checkRestoreFinishTimesOf("${tableName}", 60))
+
+    logger.info("=== Add temp partition p5 ===")
+
+    sql """
+        ALTER TABLE ${tableName} ADD TEMPORARY PARTITION p5 VALUES [("0"), ("100"))
+        """
+
+    assertTrue(helper.checkShowTimesOf("""
+                                SHOW TEMPORARY PARTITIONS
+                                FROM ${tableName}
+                                WHERE PartitionName = "p5"
+                                """,
+                                exist, 60, "sql"))
+
+    sql "INSERT INTO ${tableName} TEMPORARY PARTITION (p5) VALUES (1, 50)"
+
+    assertTrue(helper.checkShowTimesOf("""
+                                SELECT *
+                                FROM ${tableName}
+                                TEMPORARY PARTITION (p5)
+                                WHERE id = 50
+                                """,
+                                exist, 60, "sql"))
+
+    logger.info("=== Replace partition p2 by p5 ===")
+
+    assertTrue(helper.checkShowTimesOf("""
+                                SELECT *
+                                FROM ${tableName}
+                                WHERE id = 50
+                                """,
+                                notExist, 60, "target"))
+
+    sql "ALTER TABLE ${tableName} REPLACE PARTITION (p2) WITH TEMPORARY PARTITION (p5)"
+
+    assertTrue(helper.checkShowTimesOf("""
+                                SELECT *
+                                FROM ${tableName}
+                                WHERE id = 50
+                                """,
+                                exist, 60, "target"))
+
+    // The last restore should contains only partition p2
+    def show_restore_result = target_sql "SHOW RESTORE FROM TEST_${context.dbName}"
+    def restore_num = show_restore_result.size()
+    def last_restore_result = show_restore_result[restore_num-1]
+    def restore_objects = last_restore_result[10]   // RestoreObjs
+    logger.info("The restore result: ${last_restore_result}")
+    logger.info("The restore objects: ${restore_objects}")
+
+    // {
+    //  "name": "ccrp_regression_test_table_sync_test_replace_partial_p_02f747eda70e4f768afd613e074e790d_1722983645",
+    //  "database": "regression_test_table_sync",
+    //  "backup_time": 1722983645667,
+    //  "content": "ALL",
+    //  "olap_table_list": [
+    //    {
+    //      "name": "test_replace_partial_p_02f747eda70e4f768afd613e074e790d",
+    //      "partition_names": [
+    //        "p2"
+    //      ]
+    //    }
+    //  ],
+    //  "view_list": [],
+    //  "odbc_table_list": [],
+    //  "odbc_resource_list": []
+    // }
+    def jsonSlurper = new groovy.json.JsonSlurper()
+    def object = jsonSlurper.parseText "${restore_objects}"
+    assertTrue(object.olap_table_list[0].partition_names.size() == 1)
+    assertTrue(object.olap_table_list[0].partition_names[0] == "p2");
+}

regression-test/suites/db-sync-schema-change/test_db_sync_schema_change.groovy

@@ -93,6 +93,7 @@ suite("test_db_sync_schema_change") {
         """
     sql "sync"
 
+    helper.ccrJobDelete()
     def bodyJson = get_ccr_body ""
     ccr_name = get_ccr_name(bodyJson)
     helper.ccrJobCreate()

regression-test/suites/table-sync/test_replace_partial_partition.groovy

@@ -157,7 +157,8 @@ suite("test_replace_partial_partition") {
     // }
     def jsonSlurper = new groovy.json.JsonSlurper()
     def object = jsonSlurper.parseText "${restore_objects}"
-    assertTrue(object.olap_table_list[0].partition_names[0] == "p2");
+    assertTrue(object.olap_table_list[0].partition_names.size() == 1)
+    assertTrue(object.olap_table_list[0].partition_names[0] == "p2")
 }