From f295110b7e603dbe99a876964ac9941316b2a390 Mon Sep 17 00:00:00 2001
From: wyxxxcat <1520358997@qq.com>
Date: Mon, 4 Nov 2024 17:42:05 +0800
Subject: [PATCH] 1

---
 .../test_db_sync_tanle_store.groovy           | 120 ++++++++++++++++++
 1 file changed, 120 insertions(+)
 create mode 100644 regression-test/suites/db-sync-table-restore/test_db_sync_tanle_store.groovy

diff --git a/regression-test/suites/db-sync-table-restore/test_db_sync_tanle_store.groovy b/regression-test/suites/db-sync-table-restore/test_db_sync_tanle_store.groovy
new file mode 100644
index 00000000..53d89b60
--- /dev/null
+++ b/regression-test/suites/db-sync-table-restore/test_db_sync_tanle_store.groovy
@@ -0,0 +1,120 @@
+// 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_tanle_store") {
+    def helper = new GroovyShell(new Binding(['suite': delegate]))
+            .evaluate(new File("${context.config.suitePath}/../common", "helper.groovy"))
+
+    def tableName = "test_db_sync_backup_restore_table_1"
+    def newtableName = "test_db_sync_backup_restore_table_2"
+    def snapshotName = "test_db_sync_backup_restore_table_snapshot"
+    def repoName = "repo_" + UUID.randomUUID().toString().replace("-", "")
+    def test_num = 0
+    def insert_num = 10
+    def syncer = getSyncer()
+    def dbNameOrigin = context.dbName
+    def dbNameTarget = "TEST_" + context.dbName
+    syncer.createS3Repository(repoName)
+
+    def exist = { res -> Boolean
+        return res.size() != 0
+    }
+    def notExist = { res -> Boolean
+        return res.size() == 0
+    }
+
+    helper.enableDbBinlog()
+
+    sql "DROP TABLE IF EXISTS ${dbNameOrigin}.${tableName}"
+    sql "DROP TABLE IF EXISTS ${dbNameOrigin}.${newtableName}"
+    sql "DROP TABLE IF EXISTS ${dbNameTarget}.${tableName}"
+    sql "DROP TABLE IF EXISTS ${dbNameTarget}.${newtableName}"
+
+    sql """
+        CREATE TABLE if NOT EXISTS ${dbNameOrigin}.${tableName}
+        (
+            `test` INT,
+            `id` INT
+        )
+        UNIQUE KEY(`test`, `id`)
+        DISTRIBUTED BY HASH(id) BUCKETS 1
+        PROPERTIES (
+            "replication_allocation" = "tag.location.default: 1",
+            "binlog.enable" = "true"
+        )
+    """
+
+    helper.ccrJobDelete()
+    helper.ccrJobCreate()
+
+    assertTrue(helper.checkRestoreFinishTimesOf("${tableName}", 60))
+
+    for (int index = 0; index < insert_num; index++) {
+        sql """
+            INSERT INTO ${dbNameOrigin}.${tableName} VALUES (${test_num}, ${index})
+            """
+    }
+
+    logger.info("=== Test 1: Check table not exist ===")
+
+    assertTrue(helper.checkShowTimesOf(""" SHOW TABLES LIKE "${newtableName}" """, notExist, 60, "sql"))
+
+
+    logger.info("=== Test 2: Backup table===")
+    
+    assertTrue(helper.checkShowTimesOf(""" select * from ${dbNameOrigin}.${tableName} """, exist, 60, "sql"))
+
+    sql """ 
+            BACKUP SNAPSHOT ${snapshotName} 
+            TO `${repoName}` 
+            ON ( ${tableName} )
+            PROPERTIES ("type" = "full")
+        """
+
+    syncer.waitSnapshotFinish()
+    def snapshot = syncer.getSnapshotTimestamp(repoName, snapshotName)
+    assertTrue(snapshot != null)
+    syncer.waitTargetRestoreFinish()
+    
+    logger.info("=== Test 3: Restore new table ===")
+
+    sql """
+        RESTORE SNAPSHOT ${snapshotName}
+        FROM `${repoName}`
+        ON (${tableName} as ${newtableName})
+        PROPERTIES
+        (
+            "backup_timestamp" = "${snapshot}",
+            "replication_num" = "1"
+        )
+    """
+
+    syncer.waitAllRestoreFinish()
+
+    logger.info("=== Test 4: Check table ===")
+
+    assertTrue(helper.checkShowTimesOf(""" SHOW TABLES LIKE "${tableName}" """, exist, 60, "sql"))
+
+    assertTrue(helper.checkShowTimesOf(""" SHOW TABLES LIKE "${newtableName}" """, exist, 60, "sql"))
+
+    sql "USE ${dbNameTarget}"
+
+    assertTrue(helper.checkShowTimesOf(""" SHOW TABLES LIKE "${tableName}" """, exist, 60, "sql"))
+
+    assertTrue(helper.checkShowTimesOf(""" SHOW TABLES LIKE "${newtableName}" """, notExist, 60, "sql"))
+}
+