From 90301a625030115281e46b32e524996cf0be2775 Mon Sep 17 00:00:00 2001
From: Weny Xu <wenymedia@gmail.com>
Date: Wed, 31 Jul 2024 20:19:43 +0800
Subject: [PATCH] fix: generate unique timestamp for inserting tests (#4472)

---
 tests-fuzz/targets/fuzz_insert.rs               | 5 +++--
 tests-fuzz/targets/fuzz_insert_logical_table.rs | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/tests-fuzz/targets/fuzz_insert.rs b/tests-fuzz/targets/fuzz_insert.rs
index 45eec19e1b..a5db2bfffe 100644
--- a/tests-fuzz/targets/fuzz_insert.rs
+++ b/tests-fuzz/targets/fuzz_insert.rs
@@ -17,6 +17,7 @@
 use std::sync::Arc;
 
 use common_telemetry::info;
+use common_time::util::current_time_millis;
 use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured};
 use libfuzzer_sys::fuzz_target;
 use rand::{Rng, SeedableRng};
@@ -33,7 +34,7 @@ use tests_fuzz::generator::create_expr::CreateTableExprGeneratorBuilder;
 use tests_fuzz::generator::insert_expr::InsertExprGeneratorBuilder;
 use tests_fuzz::generator::Generator;
 use tests_fuzz::ir::{
-    generate_random_timestamp_for_mysql, generate_random_value, replace_default,
+    generate_random_value, generate_unique_timestamp_for_mysql, replace_default,
     sort_by_primary_keys, CreateTableExpr, InsertIntoExpr, MySQLTsColumnTypeGenerator,
 };
 use tests_fuzz::translator::mysql::create_expr::CreateTableExprTranslator;
@@ -107,7 +108,7 @@ fn generate_insert_expr<R: Rng + 'static>(
         .omit_column_list(omit_column_list)
         .rows(input.rows)
         .value_generator(Box::new(generate_random_value))
-        .ts_value_generator(Box::new(generate_random_timestamp_for_mysql))
+        .ts_value_generator(generate_unique_timestamp_for_mysql(current_time_millis()))
         .build()
         .unwrap();
     insert_generator.generate(rng)
diff --git a/tests-fuzz/targets/fuzz_insert_logical_table.rs b/tests-fuzz/targets/fuzz_insert_logical_table.rs
index cde0a87fe2..6efb6ea509 100644
--- a/tests-fuzz/targets/fuzz_insert_logical_table.rs
+++ b/tests-fuzz/targets/fuzz_insert_logical_table.rs
@@ -18,6 +18,7 @@ use std::collections::HashMap;
 use std::sync::Arc;
 
 use common_telemetry::info;
+use common_time::util::current_time_millis;
 use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured};
 use libfuzzer_sys::fuzz_target;
 use rand::{Rng, SeedableRng};
@@ -36,7 +37,7 @@ use tests_fuzz::generator::create_expr::{
 use tests_fuzz::generator::insert_expr::InsertExprGeneratorBuilder;
 use tests_fuzz::generator::Generator;
 use tests_fuzz::ir::{
-    generate_random_timestamp_for_mysql, generate_random_value, replace_default,
+    generate_random_value, generate_unique_timestamp_for_mysql, replace_default,
     sort_by_primary_keys, CreateTableExpr, InsertIntoExpr,
 };
 use tests_fuzz::translator::mysql::create_expr::CreateTableExprTranslator;
@@ -119,7 +120,7 @@ fn generate_insert_expr<R: Rng + 'static>(
         .table_ctx(table_ctx)
         .rows(rows)
         .value_generator(Box::new(generate_random_value))
-        .ts_value_generator(Box::new(generate_random_timestamp_for_mysql))
+        .ts_value_generator(generate_unique_timestamp_for_mysql(current_time_millis()))
         .build()
         .unwrap();
     insert_generator.generate(rng)