From 9cf07dcc948b60f5b14357f58ff64d18ca68615c Mon Sep 17 00:00:00 2001
From: ZhangCheng <chengzhang@apache.org>
Date: Mon, 6 Jan 2025 13:11:19 +0800
Subject: [PATCH] Support insert statement rewrite use quote

---
 .../dml/insert/insert-column.xml              | 30 ++++---------------
 1 file changed, 6 insertions(+), 24 deletions(-)

diff --git a/test/it/rewriter/src/test/resources/scenario/mix/case/query-with-cipher/dml/insert/insert-column.xml b/test/it/rewriter/src/test/resources/scenario/mix/case/query-with-cipher/dml/insert/insert-column.xml
index d3ac167011ddb..c26fba7e7b20a 100644
--- a/test/it/rewriter/src/test/resources/scenario/mix/case/query-with-cipher/dml/insert/insert-column.xml
+++ b/test/it/rewriter/src/test/resources/scenario/mix/case/query-with-cipher/dml/insert/insert-column.xml
@@ -79,8 +79,8 @@
     
     <rewrite-assertion id="insert_values_without_columns_with_id_for_literals" db-types="MySQL">
         <input sql="INSERT INTO t_account VALUES (1, 'aaa', 1000), (2, 'bbb', 2000), (3, 'ccc', 3000), (4, 'ddd', 4000)" />
-        <output sql="INSERT INTO t_account_1(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (1, 'encrypt_aaa', 'assisted_query_aaa', 'encrypt_1000'), (3, 'encrypt_ccc', 'assisted_query_ccc', 'encrypt_3000')" />
-        <output sql="INSERT INTO t_account_0(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
+        <output sql="INSERT INTO t_account_1(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (1, 'encrypt_aaa', 'assisted_query_aaa', 'encrypt_1000'), (3, 'encrypt_ccc', 'assisted_query_ccc', 'encrypt_3000')" />
+        <output sql="INSERT INTO t_account_0(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
     </rewrite-assertion>
     
     <rewrite-assertion id="insert_values_without_columns_without_id_for_parameters" db-types="MySQL">
@@ -95,14 +95,14 @@
     
     <rewrite-assertion id="insert_values_without_columns_with_plain_with_id_for_parameters" db-types="MySQL">
         <input sql="INSERT INTO t_account_bak VALUES (?, ?, ?), (2, 'bbb', 2000), (?, ?, ?), (4, 'ddd', 4000)" parameters="1, aaa, 1000, 3, ccc, 3000" />
-        <output sql="INSERT INTO t_account_bak_1(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (?, ?, ?, ?), (?, ?, ?, ?)" parameters="1, encrypt_aaa, assisted_query_aaa, encrypt_1000, 3, encrypt_ccc, assisted_query_ccc, encrypt_3000" />
-        <output sql="INSERT INTO t_account_bak_0(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
+        <output sql="INSERT INTO t_account_bak_1(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (?, ?, ?, ?), (?, ?, ?, ?)" parameters="1, encrypt_aaa, assisted_query_aaa, encrypt_1000, 3, encrypt_ccc, assisted_query_ccc, encrypt_3000" />
+        <output sql="INSERT INTO t_account_bak_0(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
     </rewrite-assertion>
     
     <rewrite-assertion id="insert_values_without_columns_with_plain_with_id_for_literals" db-types="MySQL">
         <input sql="INSERT INTO t_account_bak VALUES (1, 'aaa', 1000), (2, 'bbb', 2000), (3, 'ccc', 3000), (4, 'ddd', 4000)" />
-        <output sql="INSERT INTO t_account_bak_1(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (1, 'encrypt_aaa', 'assisted_query_aaa', 'encrypt_1000'), (3, 'encrypt_ccc', 'assisted_query_ccc', 'encrypt_3000')" />
-        <output sql="INSERT INTO t_account_bak_0(account_id, cipher_password, assisted_query_password, cipher_amount) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
+        <output sql="INSERT INTO t_account_bak_1(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (1, 'encrypt_aaa', 'assisted_query_aaa', 'encrypt_1000'), (3, 'encrypt_ccc', 'assisted_query_ccc', 'encrypt_3000')" />
+        <output sql="INSERT INTO t_account_bak_0(`account_id`, `cipher_password`, `assisted_query_password`, `cipher_amount`) VALUES (2, 'encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000'), (4, 'encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000')" />
     </rewrite-assertion>
     
     <rewrite-assertion id="insert_values_without_columns_with_plain_without_id_for_parameters" db-types="MySQL">
@@ -114,22 +114,4 @@
         <input sql="INSERT INTO t_account_bak VALUES ('aaa', 1000), ('bbb', 2000), ('ccc', 3000), ('ddd', 4000)" />
         <output sql="INSERT INTO t_account_bak_1(cipher_password, assisted_query_password, cipher_amount, account_id) VALUES ('encrypt_aaa', 'assisted_query_aaa', 'encrypt_1000', 1), ('encrypt_bbb', 'assisted_query_bbb', 'encrypt_2000', 1), ('encrypt_ccc', 'assisted_query_ccc', 'encrypt_3000', 1), ('encrypt_ddd', 'assisted_query_ddd', 'encrypt_4000', 1)" />
     </rewrite-assertion>
-
-    <rewrite-assertion id="insert_single_values_with_columns_and_quote" db-types="MySQL">
-        <input sql="INSERT INTO t_account_bak(account_id, certificate_number, password, amount, status) VALUES (?, ?, ?, ?, ?)" parameters="1, cert, pwd, 1000, OK" />
-        <output sql="INSERT INTO `encrypt_ds`.t_account_bak(account_id, cipher_certificate_number, digest_certificate_amount, cipher_password, plain_password, digest_password, cipher_amount, status) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"
-                parameters="1, encrypt_cert, digest_cert_1000, encrypt_pwd, pwd, digest_pwd, encrypt_1000, OK" />
-    </rewrite-assertion>
-
-    <rewrite-assertion id="insert_values_with_columns_and_quote" db-types="MySQL">
-        <input sql="INSERT INTO `t_account`(`account_id`, `certificate_number`, `password`, `amount`, `status`) VALUES (?, ?, ?, ?, ?)" parameters="1, cert1, pwd1, 1000, OK1" />
-        <output sql="INSERT INTO `encrypt_ds`.`t_account`(`account_id`, `cipher_certificate_number`, `assisted_query_certificate_number`, `like_query_certificate_number`, `digest_certificate_number`, `digest_certificate_password`, `digest_certificate_amount`, `digest_certificate_password_amount`, `cipher_password`, `assisted_query_password`, `like_query_password`, `digest_password`, `cipher_amount`, `digest_amount`, `status`) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
-                parameters="1, encrypt_cert1, assisted_query_cert1, like_query_cert1, digest_cert1, digest_cert1_pwd1, digest_cert1_1000, digest_cert1_pwd1_1000, encrypt_pwd1, assisted_query_pwd1, like_query_pwd1, digest_pwd1, encrypt_1000, digest_1000, OK1" />
-    </rewrite-assertion>
-
-    <rewrite-assertion id="insert_values_without_columns_and_quote" db-types="MySQL">
-        <input sql="INSERT INTO t_account_bak VALUES (?, ?, ?, ?)" parameters="1, cert1, pwd1, 1000" />
-        <output sql="INSERT INTO `encrypt_ds`.t_account_bak(`account_id`, `cipher_certificate_number`, `digest_certificate_amount`, `cipher_password`, `plain_password`, `digest_password`, `cipher_amount`) VALUES (?, ?, ?, ?, ?, ?, ?)"
-                parameters="1, encrypt_cert1, digest_cert1_1000, encrypt_pwd1, pwd1, digest_pwd1, encrypt_1000" />
-    </rewrite-assertion>
 </rewrite-assertions>