rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2025-12-26 16:10:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: generate unique timestamp for inserting tests (#4472)

Browse Source
This commit is contained in:
Weny Xu
2024-07-31 20:19:43 +08:00
committed by GitHub
parent c66d3090b6
commit 90301a6250
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests-fuzz/targets/fuzz_insert.rs
View File

@@ -17,6 +17,7 @@
use std::sync::Arc; use std::sync::Arc;
use common_telemetry::info; use common_telemetry::info;
use common_time::util::current_time_millis;
use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured}; use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured};
use libfuzzer_sys::fuzz_target; use libfuzzer_sys::fuzz_target;
use rand::{Rng, SeedableRng}; 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::insert_expr::InsertExprGeneratorBuilder;
use tests_fuzz::generator::Generator; use tests_fuzz::generator::Generator;
use tests_fuzz::ir::{ 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, sort_by_primary_keys, CreateTableExpr, InsertIntoExpr, MySQLTsColumnTypeGenerator,
}; };
use tests_fuzz::translator::mysql::create_expr::CreateTableExprTranslator; 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) .omit_column_list(omit_column_list)
.rows(input.rows) .rows(input.rows)
.value_generator(Box::new(generate_random_value)) .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() .build()
.unwrap(); .unwrap();
insert_generator.generate(rng) insert_generator.generate(rng)

5
tests-fuzz/targets/fuzz_insert_logical_table.rs
View File

@@ -18,6 +18,7 @@ use std::collections::HashMap;
use std::sync::Arc; use std::sync::Arc;
use common_telemetry::info; use common_telemetry::info;
use common_time::util::current_time_millis;
use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured}; use libfuzzer_sys::arbitrary::{Arbitrary, Unstructured};
use libfuzzer_sys::fuzz_target; use libfuzzer_sys::fuzz_target;
use rand::{Rng, SeedableRng}; 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::insert_expr::InsertExprGeneratorBuilder;
use tests_fuzz::generator::Generator; use tests_fuzz::generator::Generator;
use tests_fuzz::ir::{ 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, sort_by_primary_keys, CreateTableExpr, InsertIntoExpr,
}; };
use tests_fuzz::translator::mysql::create_expr::CreateTableExprTranslator; use tests_fuzz::translator::mysql::create_expr::CreateTableExprTranslator;
@@ -119,7 +120,7 @@ fn generate_insert_expr<R: Rng + 'static>(
.table_ctx(table_ctx) .table_ctx(table_ctx)
.rows(rows) .rows(rows)
.value_generator(Box::new(generate_random_value)) .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() .build()
.unwrap(); .unwrap();
insert_generator.generate(rng) insert_generator.generate(rng)