From c7400a4182ddcb25fed819b81b12c9250c4e74b4 Mon Sep 17 00:00:00 2001
From: WenyXu <wenymedia@gmail.com>
Date: Thu, 18 Apr 2024 13:59:20 +0000
Subject: [PATCH] chore: reduce sleep time

---
 tests-chaos/src/main.rs  |  6 +++---
 tests-chaos/src/utils.rs | 10 ++++++++--
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/tests-chaos/src/main.rs b/tests-chaos/src/main.rs
index e02dc35ec6..b94d905723 100644
--- a/tests-chaos/src/main.rs
+++ b/tests-chaos/src/main.rs
@@ -54,9 +54,9 @@ async fn main() {
         loop {
             warn!("Staring");
             let pid = start_database().await.expect("Failed to start database");
-            let secs = rng.gen_range(1..2);
+            let secs = rng.gen_range(100..300);
             moved_state.killed.store(false, Ordering::Relaxed);
-            tokio::time::sleep(Duration::from_secs(secs)).await;
+            tokio::time::sleep(Duration::from_millis(secs)).await;
             warn!("After {secs}s, Killing pid: {pid}");
             moved_state.killed.store(true, Ordering::Relaxed);
             ProcessManager::kill(pid, Signal::SIGKILL).expect("Failed to kill");
@@ -155,7 +155,7 @@ async fn start_database() -> Result<Pid> {
     let pid = start_process(&process_manager, binary_path, test_dir, template_filename)
         .await
         .unwrap();
-    tokio::time::timeout(Duration::from_secs(10), health_check(health_url))
+    tokio::time::timeout(Duration::from_secs(100), health_check(health_url))
         .await
         .expect("Failed to start GreptimeDB process");
     info!("GreptimeDB started, pid: {pid}");
diff --git a/tests-chaos/src/utils.rs b/tests-chaos/src/utils.rs
index 62948f26ea..48e30aef93 100644
--- a/tests-chaos/src/utils.rs
+++ b/tests-chaos/src/utils.rs
@@ -18,7 +18,10 @@ use rand::Rng;
 use serde::Serialize;
 use snafu::ResultExt;
 use tests_fuzz::context::TableContextRef;
-use tests_fuzz::fake::WordGenerator;
+use tests_fuzz::fake::{
+    merge_two_word_map_fn, random_capitalize_map, uppercase_and_keyword_backtick_map,
+    MappedGenerator, WordGenerator,
+};
 use tests_fuzz::generator::alter_expr::{
     AlterExprAddColumnGeneratorBuilder, AlterExprDropColumnGeneratorBuilder,
     AlterExprRenameGeneratorBuilder,
@@ -63,7 +66,10 @@ pub(crate) fn render_config_file<C: Serialize>(template_path: &str, context: &C)
 pub(crate) fn generate_create_table_expr<R: Rng + 'static>(rng: &mut R) -> CreateTableExpr {
     let columns = rng.gen_range(2..30);
     let create_table_generator = CreateTableExprGeneratorBuilder::default()
-        .name_generator(Box::new(WordGenerator))
+        .name_generator(Box::new(MappedGenerator::new(
+            WordGenerator,
+            merge_two_word_map_fn(random_capitalize_map, uppercase_and_keyword_backtick_map),
+        )))
         .columns(columns)
         .engine("mito")
         .build()